MiFaZhan
112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/os/gcfg"
|
|
)
|
|
|
|
func TestAppDiagnosisTrafficDatabaseFor(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
config appDiagnosisTrafficConfig
|
|
stable string
|
|
want string
|
|
}{
|
|
{
|
|
name: "daily usage uses daily database",
|
|
config: appDiagnosisTrafficConfig{DbName: "iotlcm", DailyDbName: "iotlcm_daily", MonthlyDbName: "iotlcm_monthly"},
|
|
stable: appDiagnosisDailyStable,
|
|
want: "iotlcm_daily",
|
|
},
|
|
{
|
|
name: "monthly usage uses monthly database",
|
|
config: appDiagnosisTrafficConfig{DbName: "iotlcm", DailyDbName: "iotlcm_daily", MonthlyDbName: "iotlcm_monthly"},
|
|
stable: appDiagnosisMonthlyStable,
|
|
want: "iotlcm_monthly",
|
|
},
|
|
{
|
|
name: "daily usage falls back to legacy database",
|
|
config: appDiagnosisTrafficConfig{DbName: "iotlcm"},
|
|
stable: appDiagnosisDailyStable,
|
|
want: "iotlcm",
|
|
},
|
|
{
|
|
name: "monthly usage falls back to legacy database",
|
|
config: appDiagnosisTrafficConfig{DbName: "iotlcm"},
|
|
stable: appDiagnosisMonthlyStable,
|
|
want: "iotlcm",
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
if got := test.config.databaseFor(test.stable); got != test.want {
|
|
t.Fatalf("databaseFor(%q) = %q, want %q", test.stable, got, test.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAppDiagnosisGetTrafficConfig(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
content string
|
|
wantDaily string
|
|
wantMonthly string
|
|
}{
|
|
{
|
|
name: "explicit split databases",
|
|
content: `{"tsdb":{"tdengine":{"dbName":"iotlcm","dailyDbName":"iotlcm_daily","monthlyDbName":"iotlcm_monthly"}}}`,
|
|
wantDaily: "iotlcm_daily",
|
|
wantMonthly: "iotlcm_monthly",
|
|
},
|
|
{
|
|
name: "custom database names are trimmed",
|
|
content: `{"tsdb":{"tdengine":{"dbName":" legacy ","dailyDbName":" custom_daily ","monthlyDbName":" custom_monthly "}}}`,
|
|
wantDaily: "custom_daily",
|
|
wantMonthly: "custom_monthly",
|
|
},
|
|
{
|
|
name: "missing split settings use configured database",
|
|
content: `{"tsdb":{"tdengine":{"dbName":"iotlcm"}}}`,
|
|
wantDaily: "iotlcm",
|
|
wantMonthly: "iotlcm",
|
|
},
|
|
{
|
|
name: "blank split settings use configured database",
|
|
content: `{"tsdb":{"tdengine":{"dbName":"iotlcm","dailyDbName":" ","monthlyDbName":""}}}`,
|
|
wantDaily: "iotlcm",
|
|
wantMonthly: "iotlcm",
|
|
},
|
|
{
|
|
name: "missing settings match console default database",
|
|
content: `{"tsdb":{"tdengine":{}}}`,
|
|
wantDaily: "iotlcm_daily",
|
|
wantMonthly: "iotlcm_daily",
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
adapter, err := gcfg.NewAdapterContent(test.content)
|
|
if err != nil {
|
|
t.Fatalf("configure traffic database: %v", err)
|
|
}
|
|
previous := g.Cfg().GetAdapter()
|
|
g.Cfg().SetAdapter(adapter)
|
|
t.Cleanup(func() { g.Cfg().SetAdapter(previous) })
|
|
|
|
config := appDiagnosisGetTrafficConfig(context.Background())
|
|
if got := config.databaseFor(appDiagnosisDailyStable); got != test.wantDaily {
|
|
t.Fatalf("daily database = %q, want %q", got, test.wantDaily)
|
|
}
|
|
if got := config.databaseFor(appDiagnosisMonthlyStable); got != test.wantMonthly {
|
|
t.Fatalf("monthly database = %q, want %q", got, test.wantMonthly)
|
|
}
|
|
})
|
|
}
|
|
}
|