httpHandler_test.go

package main

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/gorilla/mux"
	"github.com/iamNoah1/vortex/transaction"
)

type MockTransactionLogger struct{}

func (m *MockTransactionLogger) Put(key string, value string) error {
	return nil
}

func (m *MockTransactionLogger) Delete(key string) error {
	return nil
}

func (m *MockTransactionLogger) ReplayEvents() error {
	return nil
}

var mockLogger transaction.TransactionLogger = &MockTransactionLogger{}

func TestPutKeyValuePairHandler(t *testing.T) {
	req, err := http.NewRequest("PUT", "/v1/kv/test_key", bytes.NewBuffer([]byte(`{"value":"test_value"}`)))
	if err != nil {
		t.Fatal(err)
	}

	responseRecorder := httptest.NewRecorder()
	router := mux.NewRouter()
	router.HandleFunc("/v1/kv/{key}", func(w http.ResponseWriter, r *http.Request) {
		putKeyValuePairHandler(w, r, mockLogger)
	}).Methods("PUT")

	router.ServeHTTP(responseRecorder, req)

	if status := responseRecorder.Code; status != http.StatusCreated {
		t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusCreated)
	}
}

func TestGetKeyValuePairHandler(t *testing.T) {
	req, err := http.NewRequest("GET", "/v1/kv/test_key", nil)
	if err != nil {
		t.Fatal(err)
	}

	responseRecorder := httptest.NewRecorder()
	router := mux.NewRouter()
	router.HandleFunc("/v1/kv/{key}", getKeyValuePairHandler).Methods("GET")

	router.ServeHTTP(responseRecorder, req)

	if status := responseRecorder.Code; status != http.StatusOK {
		t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK)
	}
}

func TestGetKeyValuePairHandlerKeyNotFound(t *testing.T) {
	req, err := http.NewRequest("GET", "/v1/kv/non_existing_key", nil)
	if err != nil {
		t.Fatal(err)
	}

	responseRecorder := httptest.NewRecorder()
	router := mux.NewRouter()
	router.HandleFunc("/v1/kv/{key}", getKeyValuePairHandler).Methods("GET")

	router.ServeHTTP(responseRecorder, req)

	if status := responseRecorder.Code; status != http.StatusNotFound {
		t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusNotFound)
	}
}

func TestDeleteKeyValuePairHandler(t *testing.T) {
	req, err := http.NewRequest("DELETE", "/v1/kv/test_key", nil)
	if err != nil {
		t.Fatal(err)
	}

	responseRecorder := httptest.NewRecorder()
	router := mux.NewRouter()
	router.HandleFunc("/v1/kv/{key}", func(w http.ResponseWriter, r *http.Request) {
		deleteKeyValuePairHandler(w, r, mockLogger)
	}).Methods("DELETE")

	router.ServeHTTP(responseRecorder, req)

	if status := responseRecorder.Code; status != http.StatusOK {
		t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK)
	}

	// Add more assertions based on your application's logic
}