Add unit tests for player configurations and referral system
- Implement tests for player configuration creation, update, and deletion, ensuring proper handling of free and paid user scenarios. - Add tests for referral registration, including valid and invalid referrer cases. - Create tests for referral reward flow, verifying correct reward distribution and eligibility. - Establish a test database setup with necessary schema for user, plan, payment, and notification models. - Introduce helper functions for seeding test data and loading entities from the database.
This commit is contained in:
74
cmd/server/main.go
Normal file
74
cmd/server/main.go
Normal file
@@ -0,0 +1,74 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"stream.api/internal/adapters/redis"
|
||||
"stream.api/internal/config"
|
||||
"stream.api/internal/database/query"
|
||||
"stream.api/internal/transport/grpc"
|
||||
"stream.api/pkg/database"
|
||||
"stream.api/pkg/logger"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 1. Load Config
|
||||
cfg, err := config.LoadConfig()
|
||||
if err != nil {
|
||||
// Use default if env/file issues, usually LoadConfig returns error only on serious issues
|
||||
// But here if it returns error we might want to panic
|
||||
log.Fatalf("Failed to load config: %v", err)
|
||||
}
|
||||
|
||||
// 2. Connect DB
|
||||
db, err := database.Connect(cfg.Database.DSN)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to connect to database: %v", err)
|
||||
}
|
||||
// Initialize generated query
|
||||
query.SetDefault(db)
|
||||
|
||||
// TODO: Tách database migration ra luồng riêng nếu cần.
|
||||
|
||||
// 3. Connect Redis (Cache Interface)
|
||||
rdb, err := redis.NewAdapter(cfg.Redis.Addr, cfg.Redis.Password, cfg.Redis.DB)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to connect to redis: %v", err)
|
||||
}
|
||||
defer rdb.Close() // Ensure we close cache on exit
|
||||
|
||||
// 4. Initialize Components
|
||||
appLogger := logger.NewLogger(cfg.Server.Mode)
|
||||
|
||||
module, err := grpc.NewGRPCModule(context.Background(), cfg, db, rdb, appLogger)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to setup gRPC runtime module: %v", err)
|
||||
}
|
||||
|
||||
grpcListener, err := net.Listen("tcp", ":"+cfg.Server.GRPCPort)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to listen on gRPC port %s: %v", cfg.Server.GRPCPort, err)
|
||||
}
|
||||
|
||||
go func() {
|
||||
log.Printf("Starting gRPC server on port %s", cfg.Server.GRPCPort)
|
||||
if err := module.ServeGRPC(grpcListener); err != nil {
|
||||
log.Fatalf("Failed to run gRPC server: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
log.Println("Shutting down gRPC server...")
|
||||
|
||||
module.Shutdown()
|
||||
_ = grpcListener.Close()
|
||||
|
||||
log.Println("Server exiting")
|
||||
}
|
||||
Reference in New Issue
Block a user