Use mock repositories for testing and fix initialization issues

Author: Jetlum

main.go

@@ -5,9 +5,11 @@ import (
 	"fmt"
 	"log"
 	"math/big"
+	"os"
 
 	"github.com/Jetlum/WalletAlertService/config"
 	"github.com/Jetlum/WalletAlertService/database"
+	"github.com/Jetlum/WalletAlertService/mock"
 	"github.com/Jetlum/WalletAlertService/models"
 	nfts "github.com/Jetlum/WalletAlertService/nft"
 	"github.com/Jetlum/WalletAlertService/repository"
@@ -17,6 +19,9 @@ import (
 )
 
 func init() {
+	if os.Getenv("GO_ENV") == "test" {
+		return
+	}
 	cfg, err := config.LoadConfig()
 	if err != nil {
 		log.Fatal("Failed to load config:", err)
@@ -34,8 +39,17 @@ func main() {
 		log.Fatal("Failed to load config:", err)
 	}
 
-	eventRepo := repository.NewEventRepository(database.DB)
-	userPrefRepo := repository.NewUserPreferenceRepository(database.DB)
+	var eventRepo repository.EventRepositoryInterface
+	var userPrefRepo repository.UserPreferenceRepositoryInterface
+
+	if os.Getenv("GO_ENV") == "test" {
+		eventRepo = mock.NewMockEventRepository()
+		userPrefRepo = mock.NewMockUserPreferenceRepository()
+	} else {
+		eventRepo = repository.NewEventRepository(database.DB)
+		userPrefRepo = repository.NewUserPreferenceRepository(database.DB)
+	}
+
 	emailNotification := services.NewEmailNotification(cfg.SendGridAPIKey)
 	nftDetector := nfts.NewNFTDetector()
 

main_test.go

@@ -1,13 +1,19 @@
 package main
 
 import (
+	"os"
 	"testing"
 
 	"github.com/Jetlum/WalletAlertService/mock"
 	"github.com/Jetlum/WalletAlertService/models"
 	"github.com/stretchr/testify/assert"
 )
 
+func TestMain(m *testing.M) {
+	os.Setenv("GO_ENV", "test")
+	os.Exit(m.Run())
+}
+
 func TestNotifyUsers(t *testing.T) {
 	// Create a sample event to pass to notifyUsers
 	event := &models.Event{

mock/database_mock.go

@@ -0,0 +1,38 @@
+package mock
+
+import (
+	"github.com/Jetlum/WalletAlertService/models"
+	"gorm.io/gorm"
+)
+
+type MockDB struct {
+	Events          []models.Event
+	UserPreferences []models.UserPreference
+}
+
+func NewMockDB() *MockDB {
+	return &MockDB{
+		Events:          []models.Event{},
+		UserPreferences: []models.UserPreference{},
+	}
+}
+
+func (db *MockDB) Create(value interface{}) *gorm.DB {
+	switch v := value.(type) {
+	case *models.Event:
+		db.Events = append(db.Events, *v)
+	case *models.UserPreference:
+		db.UserPreferences = append(db.UserPreferences, *v)
+	}
+	return &gorm.DB{}
+}
+
+func (db *MockDB) Find(out interface{}, where ...interface{}) *gorm.DB {
+	switch out := out.(type) {
+	case *[]models.Event:
+		*out = db.Events
+	case *[]models.UserPreference:
+		*out = db.UserPreferences
+	}
+	return &gorm.DB{}
+}

mock/repository_mock.go

@@ -23,3 +23,13 @@ func (m *MockEventRepository) Create(event *models.Event) error {
 	}
 	return nil
 }
+
+// Constructor for MockUserPreferenceRepository
+func NewMockUserPreferenceRepository() *MockUserPreferenceRepository {
+	return &MockUserPreferenceRepository{}
+}
+
+// Constructor for MockEventRepository
+func NewMockEventRepository() *MockEventRepository {
+	return &MockEventRepository{}
+}

repository/event_repository.go

@@ -9,20 +9,10 @@ type EventRepository struct {
 	db *gorm.DB
 }
 
-var _ EventRepositoryInterface = (*EventRepository)(nil)
-
 func NewEventRepository(db *gorm.DB) *EventRepository {
 	return &EventRepository{db: db}
 }
 
-func (er *EventRepository) Create(event *models.Event) error {
-	return er.db.Create(event).Error
-}
-
-func (er *EventRepository) GetUnnotifiedEvents() ([]models.Event, error) {
-	return nil, nil
-}
-
-func (er *EventRepository) MarkAsNotified(eventID uint) error {
-	return nil
+func (r *EventRepository) Create(event *models.Event) error {
+	return r.db.Create(event).Error
 }

repository/repository.go

@@ -2,12 +2,10 @@ package repository
 
 import "github.com/Jetlum/WalletAlertService/models"
 
-type UserPreferenceRepositoryInterface interface {
-	GetMatchingPreferences(event *models.Event) ([]models.UserPreference, error)
-}
-
 type EventRepositoryInterface interface {
 	Create(event *models.Event) error
-	GetUnnotifiedEvents() ([]models.Event, error)
-	MarkAsNotified(eventID uint) error
+}
+
+type UserPreferenceRepositoryInterface interface {
+	GetMatchingPreferences(event *models.Event) ([]models.UserPreference, error)
 }

repository/user_preference.go

@@ -9,12 +9,12 @@ type UserPreferenceRepository struct {
 	db *gorm.DB
 }
 
-var _ UserPreferenceRepositoryInterface = (*UserPreferenceRepository)(nil)
-
 func NewUserPreferenceRepository(db *gorm.DB) *UserPreferenceRepository {
 	return &UserPreferenceRepository{db: db}
 }
 
-func (upr *UserPreferenceRepository) GetMatchingPreferences(event *models.Event) ([]models.UserPreference, error) {
-	return nil, nil
+func (r *UserPreferenceRepository) GetMatchingPreferences(event *models.Event) ([]models.UserPreference, error) {
+	var preferences []models.UserPreference
+	err := r.db.Where("wallet_address = ?", event.ToAddress).Find(&preferences).Error
+	return preferences, err
 }