File size: 2,572 Bytes
d6f631f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | package streaming
import (
"fmt"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/openmeterio/openmeter/openmeter/meter"
"github.com/openmeterio/openmeter/pkg/models"
)
type testCustomer struct {
ID string
Key *string
SubjectKeys []string
}
func (c testCustomer) GetUsageAttribution() CustomerUsageAttribution {
return NewCustomerUsageAttribution(c.ID, c.Key, c.SubjectKeys)
}
func TestQueryParamsValidate(t *testing.T) {
queryWindowSizeMinute := meter.WindowSizeMinute
tests := []struct {
name string
paramFrom string
paramTo string
paramWindowTimeZone string
paramWindowSize *meter.WindowSize
paramFilterCustomer []Customer
want error
}{
{
name: "should fail when from and to are equal",
paramFrom: "2023-01-01T00:00:00Z",
paramTo: "2023-01-01T00:00:00Z",
paramWindowSize: &queryWindowSizeMinute,
want: models.NewGenericValidationError(fmt.Errorf("from and to cannot be equal")),
},
{
name: "should fail when from is before to",
paramFrom: "2023-01-02T00:00:00Z",
paramTo: "2023-01-01T00:00:00Z",
paramWindowSize: &queryWindowSizeMinute,
want: models.NewGenericValidationError(fmt.Errorf("from must be before to")),
},
{
name: "should fail when filter customer has no id",
paramFrom: "2023-01-01T00:00:00Z",
paramTo: "2023-01-02T00:00:00Z",
paramWindowSize: &queryWindowSizeMinute,
paramFilterCustomer: []Customer{
testCustomer{
ID: "",
},
},
want: models.NewGenericValidationError(fmt.Errorf("usage attribution must have an id")),
},
}
for _, tt := range tests {
tt := tt
paramWindowSize := "none"
if tt.paramWindowSize != nil {
paramWindowSize = string(*tt.paramWindowSize)
}
name := fmt.Sprintf("%s/%s", paramWindowSize, tt.name)
t.Run(name, func(t *testing.T) {
from, err := time.Parse(time.RFC3339, tt.paramFrom)
if err != nil {
t.Fatal(fmt.Errorf("failed to parse from: %w", err))
return
}
to, err := time.Parse(time.RFC3339, tt.paramTo)
if err != nil {
t.Fatal(fmt.Errorf("failed to parse to: %w", err))
return
}
p := QueryParams{
From: &from,
To: &to,
WindowSize: tt.paramWindowSize,
FilterCustomer: tt.paramFilterCustomer,
}
got := p.Validate()
if tt.want == nil {
assert.NoError(t, got)
} else {
assert.ErrorAs(t, got, &tt.want)
}
})
}
}
|