Refactor Chats Service (#48)

chats/handler/chats_test.go

@@ -0,0 +1,83 @@
+package handler_test
+
+import (
+	"testing"
+	"time"
+
+	"github.com/micro/services/chats/handler"
+	pb "github.com/micro/services/chats/proto"
+	"github.com/stretchr/testify/assert"
+
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func testHandler(t *testing.T) *handler.Chats {
+	// connect to the database
+	db, err := gorm.Open(postgres.Open("postgresql://postgres@localhost:5432/chats?sslmode=disable"), &gorm.Config{})
+	if err != nil {
+		t.Fatalf("Error connecting to database: %v", err)
+	}
+
+	// migrate the database
+	if err := db.AutoMigrate(&handler.Chat{}, &handler.Message{}); err != nil {
+		t.Fatalf("Error migrating database: %v", err)
+	}
+
+	// clean any data from a previous run
+	if err := db.Exec("TRUNCATE TABLE chats, messages CASCADE").Error; err != nil {
+		t.Fatalf("Error cleaning database: %v", err)
+	}
+
+	return &handler.Chats{DB: db, Time: func() time.Time { return time.Unix(1611327673, 0) }}
+}
+
+func assertChatsMatch(t *testing.T, exp, act *pb.Chat) {
+	if act == nil {
+		t.Errorf("Chat not returned")
+		return
+	}
+
+	// adapt this check so we can reuse the func in testing create, where we don't know the exact id
+	// which will be generated
+	if len(exp.Id) > 0 {
+		assert.Equal(t, exp.Id, act.Id)
+	} else {
+		assert.NotEmpty(t, act.Id)
+	}
+
+	assert.Equal(t, exp.UserIds, act.UserIds)
+
+	if act.CreatedAt == nil {
+		t.Errorf("CreatedAt not set")
+		return
+	}
+
+	assert.True(t, exp.CreatedAt.AsTime().Equal(act.CreatedAt.AsTime()))
+}
+
+func assertMessagesMatch(t *testing.T, exp, act *pb.Message) {
+	if act == nil {
+		t.Errorf("Message not returned")
+		return
+	}
+
+	// adapt this check so we can reuse the func in testing create, where we don't know the exact id
+	// which will be generated
+	if len(exp.Id) > 0 {
+		assert.Equal(t, exp.Id, act.Id)
+	} else {
+		assert.NotEmpty(t, act.Id)
+	}
+
+	assert.Equal(t, exp.Text, act.Text)
+	assert.Equal(t, exp.AuthorId, act.AuthorId)
+	assert.Equal(t, exp.ChatId, act.ChatId)
+
+	if act.SentAt == nil {
+		t.Errorf("SentAt not set")
+		return
+	}
+
+	assert.True(t, exp.SentAt.AsTime().Equal(act.SentAt.AsTime()))
+}