MiFaZhan
55 lines
2.1 KiB
Go
55 lines
2.1 KiB
Go
package trafficusage
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
|
"testing"
|
|
)
|
|
|
|
func TestPersistenceFailureRetainsSnapshot(t *testing.T) {
|
|
upstream := &upstreamStub{dailyOutput: validDailyOutput()}
|
|
repository := &repositoryStub{dailyErr: errors.New("database offline")}
|
|
service := New(upstream, repository)
|
|
message := validDailyMessage()
|
|
var failure *Failure
|
|
if err := service.RetryDaily(context.Background(), message); !errors.As(err, &failure) || failure.UsageMb == nil || *failure.UsageMb != 1 {
|
|
t.Fatalf("snapshot missing: %v", err)
|
|
}
|
|
message.UsageMb = failure.UsageMb
|
|
repository.dailyErr = nil
|
|
if err := service.RetryDaily(context.Background(), message); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if upstream.dailyCalls != 1 || repository.dailyCalls != 2 {
|
|
t.Fatal("persistence retry queried upstream again")
|
|
}
|
|
}
|
|
|
|
func TestZeroSnapshotBypassesUpstream(t *testing.T) {
|
|
upstream, repository := &upstreamStub{}, &repositoryStub{}
|
|
service := New(upstream, repository)
|
|
zero := 0.0
|
|
message := validDailyMessage()
|
|
message.UsageMb = &zero
|
|
if err := service.RetryDaily(context.Background(), message); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
monthly := model.TrafficUsageMonthlyRetryMessage{SimKey: message.SimKey, Card: message.Card, QueryDate: "202607", UsageMb: &zero}
|
|
if err := service.RetryMonthly(context.Background(), monthly); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if upstream.dailyCalls != 0 || upstream.monthlyCalls != 0 || repository.dailyCalls != 1 || repository.monthlyCalls != 1 {
|
|
t.Fatal("zero snapshot was not persisted directly")
|
|
}
|
|
}
|
|
|
|
func TestRejectMismatchedOrMissingResult(t *testing.T) {
|
|
for _, output := range [][]model.SimBaseRes[model.SimDataUsageDailyOutputItem]{nil, {{}}, {{SimKey: &model.SimKey{ICCID: "wrong"}}}, {validDailyOutput()[0], validDailyOutput()[0]}} {
|
|
upstream, repository := &upstreamStub{dailyOutput: output}, &repositoryStub{}
|
|
if err := New(upstream, repository).RetryDaily(context.Background(), validDailyMessage()); err == nil || repository.dailyCalls != 0 {
|
|
t.Fatal("invalid result accepted")
|
|
}
|
|
}
|
|
}
|