redis-client_test.go

package redisclient

import (
	"bytes"
	"os"
	"testing"
)

// Helper function to compare byte slices
func bytesEqual(a, b []byte) bool {
	return bytes.Equal(a, b)
}

// Test the encodeCommand function
func TestEncodeCommand(t *testing.T) {
	tests := []struct {
		name     string
		args     []string
		expected []byte
	}{
		{
			name:     "Simple SET command",
			args:     []string{"SET", "key", "value"},
			expected: []byte("*3\r\n$3\r\nSET\r\n$3\r\nkey\r\n$5\r\nvalue\r\n"),
		},
		{
			name:     "GET command",
			args:     []string{"GET", "key"},
			expected: []byte("*2\r\n$3\r\nGET\r\n$3\r\nkey\r\n"),
		},
		{
			name:     "Complex SADD command",
			args:     []string{"SADD", "myset", "value1", "value2", "value3"},
			expected: []byte("*5\r\n$4\r\nSADD\r\n$5\r\nmyset\r\n$6\r\nvalue1\r\n$6\r\nvalue2\r\n$6\r\nvalue3\r\n"),
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := encodeCommand(tt.args)
			if !bytesEqual(result, tt.expected) {
				t.Errorf("encodeCommand(%v) = %v, want %v", tt.args, result, tt.expected)
			}
		})
	}
}

func TestRedisConnection(t *testing.T) {

	/* 	err := godotenv.Load(".env")

	   	if err != nil {
	   		t.Fatalf("Error loading .env file")
	   	} */

	// Fetch Redis connection details from environment variables
	redisHost := os.Getenv("REDIS_HOST")

	redisPassword := os.Getenv("REDIS_PASSWORD")

	t.Log(redisHost, redisPassword)

	// Create Redis client
	rdb, err := NewClient(redisHost, 11364, WithPassword(redisPassword))
	if err != nil {
		t.Log("error while connecting ")
	}

	// Test connection
	res, err := rdb.Ping()

	t.Log(res)

	if res.(string) != "PONG" {
		t.Fatalf("Incorrect response: %v", err)
	}

	if err != nil {
		t.Fatalf("Failed to connect to Redis: %v", err)
	}

	t.Log("Redis connection test passed successfully")
}

func TestGetCommand(t *testing.T) {

	redisHost := os.Getenv("REDIS_HOST")

	redisPassword := os.Getenv("REDIS_PASSWORD")

	// Create Redis client
	rdb, err := NewClient(redisHost, 11364, WithPassword(redisPassword))

	if err != nil {
		t.Log("error while connecting ")
	}

	res, err := rdb.Set("test1", "test1val")

	if res.(string) != "OK" {
		t.Fatalf("Incorrect response: %v", err)
	}

	res, err = rdb.Get("test1")

	if res.(string) != "test1val" {
		t.Fatalf("Incorrect response from get command: %v", err)
	}

	if err != nil {
		t.Fatalf("Failed to connect to Redis: %v", err)
	}

	t.Log("get command tested successfully")
}

func TestHSetCommand(t *testing.T) {

	redisHost := os.Getenv("REDIS_HOST")

	redisPassword := os.Getenv("REDIS_PASSWORD")

	// Create Redis client
	rdb, err := NewClient(redisHost, 11364, WithPassword(redisPassword))

	if err != nil {
		t.Log("error while connecting ")
	}

	rdb.HDel("myhash", "field1")

	hres, err := rdb.HSet("myhash", "field1", "foo")

	if hres.(int64) != 1 {
		t.Fatalf("Incorrect response: %v", hres)
	}

	dres, err := rdb.HDel("myhash", "field1")

	if dres.(int64) != 1 {
		t.Fatalf("Incorrect response from hdel command: %v", dres)
	}

	if err != nil {
		t.Fatalf("Failed to connect to Redis: %v", err)
	}

	t.Log("hdel command tested successfully")
}

func TestHExistsCommand(t *testing.T) {

	redisHost := os.Getenv("REDIS_HOST")

	redisPassword := os.Getenv("REDIS_PASSWORD")

	// Create Redis client
	rdb, err := NewClient(redisHost, 11364, WithPassword(redisPassword))

	if err != nil {
		t.Log("error while connecting ")
	}

	//in case it already exists
	rdb.HDel("myhash", "field1")

	hres, err := rdb.HSet("myhash", "field1", "foo")

	if hres.(int64) != 1 {
		t.Fatalf("Incorrect response: %v", hres)
	}

	exists_res, _ := rdb.HExists("myhash", "field1")

	if exists_res.(int64) != 1 {
		t.Fatalf("Incorrect response from hexists command: %v", exists_res)
	}

	if err != nil {
		t.Fatalf("Failed to connect to Redis: %v", err)
	}

	t.Log("hexists command tested successfully")
}

func TestHGetCommand(t *testing.T) {

	redisHost := os.Getenv("REDIS_HOST")

	redisPassword := os.Getenv("REDIS_PASSWORD")

	// Create Redis client
	rdb, err := NewClient(redisHost, 11364, WithPassword(redisPassword))

	if err != nil {
		t.Log("error while connecting ")
	}

	//in case it already exists
	rdb.HDel("myhash", "field1")

	hres, err := rdb.HSet("myhash", "field1", "foo")

	if hres.(int64) != 1 {
		t.Fatalf("Incorrect response: %v", hres)
	}

	exists_res, _ := rdb.HGet("myhash", "field1")

	if exists_res.(string) != "foo" {
		t.Fatalf("Incorrect response from hexists command: %v", exists_res)
	}

	if err != nil {
		t.Fatalf("Failed to connect to Redis: %v", err)
	}

	t.Log("hget command tested successfully")
}