| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109 |
- package main
- import (
- "testing"
- "time"
- "github.com/Wei-Shaw/sub2api/internal/config"
- "github.com/Wei-Shaw/sub2api/internal/handler"
- "github.com/Wei-Shaw/sub2api/internal/service"
- "github.com/stretchr/testify/require"
- )
- func TestProvideServiceBuildInfo(t *testing.T) {
- in := handler.BuildInfo{
- Version: "v-test",
- BuildType: "release",
- }
- out := provideServiceBuildInfo(in)
- require.Equal(t, in.Version, out.Version)
- require.Equal(t, in.BuildType, out.BuildType)
- }
- func TestProvideCleanup_WithMinimalDependencies_NoPanic(t *testing.T) {
- cfg := &config.Config{}
- oauthSvc := service.NewOAuthService(nil, nil)
- openAIOAuthSvc := service.NewOpenAIOAuthService(nil, nil)
- geminiOAuthSvc := service.NewGeminiOAuthService(nil, nil, nil, nil, cfg)
- antigravityOAuthSvc := service.NewAntigravityOAuthService(nil)
- tokenRefreshSvc := service.NewTokenRefreshService(
- nil,
- oauthSvc,
- openAIOAuthSvc,
- geminiOAuthSvc,
- antigravityOAuthSvc,
- nil,
- nil,
- cfg,
- nil,
- )
- accountExpirySvc := service.NewAccountExpiryService(nil, time.Second)
- codexVersionSyncSvc := service.NewOpenAICodexVersionSyncService(nil, nil, nil, time.Second)
- proxyExpirySvc := service.NewProxyExpiryService(nil, time.Second)
- subscriptionExpirySvc := service.NewSubscriptionExpiryService(nil, time.Second)
- pricingSvc := service.NewPricingService(cfg, nil)
- emailQueueSvc := service.NewEmailQueueService(nil, 1)
- billingCacheSvc := service.NewBillingCacheService(nil, nil, nil, nil, nil, nil, cfg, nil)
- idempotencyCleanupSvc := service.NewIdempotencyCleanupService(nil, cfg)
- schedulerSnapshotSvc := service.NewSchedulerSnapshotService(nil, nil, nil, nil, cfg)
- opsSystemLogSinkSvc := service.NewOpsSystemLogSink(nil)
- cleanup := provideCleanup(
- nil, // entClient
- nil, // redis
- &service.OpsMetricsCollector{},
- &service.OpsAggregationService{},
- &service.OpsAlertEvaluatorService{},
- &service.OpsCleanupService{},
- &service.OpsScheduledReportService{},
- opsSystemLogSinkSvc,
- nil, // opsService
- nil, // opsIngressRejectAggregator
- nil, // apiKeyService
- nil, // authCacheInvalidationWorker
- schedulerSnapshotSvc,
- tokenRefreshSvc,
- accountExpirySvc,
- nil, // cnProviderBalanceCheck
- codexVersionSyncSvc,
- proxyExpirySvc,
- nil, // accountHealth
- nil, // margin
- nil, // spendGuard
- subscriptionExpirySvc,
- &service.UsageCleanupService{},
- idempotencyCleanupSvc,
- &service.BatchImageCleanupService{},
- nil, // batchImageWorker
- pricingSvc,
- emailQueueSvc,
- billingCacheSvc,
- &service.UsageRecordWorkerPool{},
- &service.SubscriptionService{},
- oauthSvc,
- openAIOAuthSvc,
- geminiOAuthSvc,
- antigravityOAuthSvc,
- nil, // grokOAuth
- nil, // openAIGateway
- nil, // scheduledTestRunner
- nil, // intelligentTests
- nil, // backupSvc
- nil, // paymentOrderExpiry
- nil, // channelMonitorRunner
- nil, // channelMonitorV2Aggregator
- nil, // quotaFlusher
- nil, // upstreamBillingProbe
- nil, // ollamaCloudUsage
- nil, // auditLog
- nil, // openAIAutoReset
- nil, // promptAudit
- nil, // pluginManager
- )
- require.NotPanics(t, func() {
- cleanup()
- })
- }
|