Moved tests for module commands and apis into 'test' folder

test/modules/connection/commands_test.go

@@ -0,0 +1,118 @@
+// Copyright 2024 Kelvin Clement Mwinuka
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package connection
+
+import (
+	"bytes"
+	"context"
+	"errors"
+	"github.com/echovault/echovault/internal/config"
+	"github.com/echovault/echovault/pkg/constants"
+	"github.com/echovault/echovault/pkg/echovault"
+	"github.com/echovault/echovault/pkg/modules/connection"
+	"github.com/echovault/echovault/pkg/types"
+	"github.com/tidwall/resp"
+	"net"
+	"strings"
+	"testing"
+)
+
+var mockServer *echovault.EchoVault
+
+func init() {
+	mockServer, _ = echovault.NewEchoVault(
+		echovault.WithCommands(connection.Commands()),
+		echovault.WithConfig(config.Config{
+			DataDir:        "",
+			EvictionPolicy: constants.NoEviction,
+		}),
+	)
+}
+
+func getHandler(commands ...string) types.HandlerFunc {
+	if len(commands) == 0 {
+		return nil
+	}
+	for _, c := range mockServer.GetAllCommands() {
+		if strings.EqualFold(commands[0], c.Command) && len(commands) == 1 {
+			// Get command handler
+			return c.HandlerFunc
+		}
+		if strings.EqualFold(commands[0], c.Command) {
+			// Get sub-command handler
+			for _, sc := range c.SubCommands {
+				if strings.EqualFold(commands[1], sc.Command) {
+					return sc.HandlerFunc
+				}
+			}
+		}
+	}
+	return nil
+}
+
+func getHandlerFuncParams(ctx context.Context, cmd []string, conn *net.Conn) types.HandlerFuncParams {
+	return types.HandlerFuncParams{
+		Context:    ctx,
+		Command:    cmd,
+		Connection: conn,
+	}
+}
+
+func Test_HandlePing(t *testing.T) {
+	ctx := context.Background()
+
+	tests := []struct {
+		command     []string
+		expected    string
+		expectedErr error
+	}{
+		{
+			command:     []string{"PING"},
+			expected:    "PONG",
+			expectedErr: nil,
+		},
+		{
+			command:     []string{"PING", "Hello, world!"},
+			expected:    "Hello, world!",
+			expectedErr: nil,
+		},
+		{
+			command:     []string{"PING", "Hello, world!", "Once more"},
+			expected:    "",
+			expectedErr: errors.New(constants.WrongArgsResponse),
+		},
+	}
+
+	for _, test := range tests {
+		res, err := getHandler("PING")(getHandlerFuncParams(ctx, test.command, nil))
+		if test.expectedErr != nil && err != nil {
+			if err.Error() != test.expectedErr.Error() {
+				t.Errorf("expected error %s, got: %s", test.expectedErr.Error(), err.Error())
+			}
+			continue
+		}
+		if err != nil {
+			t.Error(err)
+		}
+		rd := resp.NewReader(bytes.NewBuffer(res))
+		v, _, err := rd.ReadValue()
+		if err != nil {
+			t.Error(err)
+		}
+		if v.String() != test.expected {
+			t.Errorf("expected %s, got: %s", test.expected, v.String())
+		}
+	}
+}