add tests for various packages (#365)

Co-authored-by: mehrotra234 <[email protected]>

Author: Umang01-hash

examples/using-migrations/migrations/all_test.go

@@ -0,0 +1,23 @@
+package migrations
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"gofr.dev/pkg/gofr/migration"
+)
+
+func TestAll(t *testing.T) {
+	// Get the map of migrations
+	allMigrations := All()
+
+	expected := map[int64]migration.Migrate{
+		1708322067: createTableEmployee(),
+		1708322089: addEmployeeInRedis(),
+		1708322090: createTopicsForStore(),
+	}
+
+	// Check if the length of the maps match
+	assert.Equal(t, len(expected), len(allMigrations), "TestAll Failed!")
+}

examples/using-publisher/main.go

@@ -2,16 +2,12 @@ package main
 
 import (
 	"encoding/json"
-	"fmt"
-
 	"gofr.dev/pkg/gofr"
 )
 
 func main() {
 	app := gofr.New()
 
-	fmt.Println(app.Config.Get("PUBSUB_BROKER"))
-
 	app.POST("/publish-order", order)
 
 	app.POST("/publish-product", product)

pkg/gofr/cmd/request.go

@@ -80,13 +80,10 @@ func (r *Request) Context() context.Context {
 	return context.Background()
 }
 
-func (r *Request) HostName() string {
-	h, err := os.Hostname()
-	if err != nil {
-		return ""
-	}
+func (r *Request) HostName() (hostname string) {
+	hostname, _ = os.Hostname()
 
-	return h
+	return hostname
 }
 
 func (r *Request) Bind(i interface{}) error {

pkg/gofr/cmd/request_test.go

@@ -54,3 +54,18 @@ func TestRequest_WithOneArg(t *testing.T) {
 
 	assert.Equal(t, req, r, "TEST Failed.\n Hostname did not match.")
 }
+
+func TestHostName(t *testing.T) {
+	r := &Request{}
+
+	// Get the hostname using os.Hostname()
+	hostname, err := os.Hostname()
+	if err != nil {
+		t.Fatalf("Error getting hostname: %v", err)
+	}
+
+	// Get the hostname from the mock request
+	result := r.HostName()
+
+	assert.Equal(t, hostname, result, "TestHostName Failed!")
+}

pkg/gofr/container/container_test.go

@@ -130,6 +130,16 @@ func TestContainer_GetSubscriber(t *testing.T) {
 	assert.Equal(t, subscriber, out)
 }
 
+func TestContainer_NewEmptyContainer(t *testing.T) {
+	container := NewEmptyContainer()
+
+	assert.Nil(t, container.Redis, "TestContainer_NewEmptyContainer Failed!")
+	assert.Nil(t, container.SQL, "TestContainer_NewEmptyContainer Failed")
+	assert.Nil(t, container.Services, "TestContainer_NewEmptyContainer Failed")
+	assert.Nil(t, container.PubSub, "TestContainer_NewEmptyContainer Failed")
+	assert.Nil(t, container.Logger, "TestContainer_NewEmptyContainer Failed")
+}
+
 type mockPubSub struct {
 }
 

pkg/gofr/logging/dynamicLevelLogger_test.go

@@ -7,6 +7,9 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
+
+	"gofr.dev/pkg/gofr/service"
 	"gofr.dev/pkg/gofr/testutil"
 )
 
@@ -47,3 +50,42 @@ func TestDynamicLoggerSuccess(t *testing.T) {
 		t.Errorf("TestDynamicLoggerSuccess failed! missing debug log")
 	}
 }
+
+func Test_fetchAndUpdateLogLevel_ErrorCases(t *testing.T) {
+	logger := testutil.NewMockLogger(testutil.INFOLOG)
+
+	remoteService := service.NewHTTPService("http://", logger, nil)
+
+	mockServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		body := `{
+			"data": [
+				{
+					"invalid"
+					}
+				}
+			]
+		}`
+		_, _ = w.Write([]byte(body))
+	}))
+	defer mockServer.Close()
+
+	remoteService2 := service.NewHTTPService(mockServer.URL, logger, nil)
+
+	tests := []struct {
+		desc            string
+		remoteService   service.HTTP
+		currentLogLevel Level
+	}{
+		{"invalid URL for remote service", remoteService, testutil.INFOLOG},
+		{"invalid response from remote service", remoteService2, testutil.DEBUGLOG},
+	}
+
+	for i, tc := range tests {
+		level, err := fetchAndUpdateLogLevel(tc.remoteService, tc.currentLogLevel)
+
+		assert.Equal(t, tc.currentLogLevel, level, "TEST[%d], Failed.\n%s", i, tc.desc)
+
+		assert.NotNil(t, err)
+	}
+}

pkg/gofr/subscriber_test.go

@@ -46,12 +46,6 @@ func (s mockSubscriber) Publish(_ context.Context, _ string, _ []byte) error {
 	return nil
 }
 
-type Message struct {
-	Topic    string
-	Value    []byte
-	MetaData interface{}
-}
-
 func (mockSubscriber) Subscribe(_ context.Context, topic string) (*pubsub.Message, error) {
 	if topic == "test-topic" {
 		return &pubsub.Message{