41 lines
937 B
Go
41 lines
937 B
Go
package models
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/v2/bson"
|
|
)
|
|
|
|
func TestInstanceLockFieldsBSON(t *testing.T) {
|
|
at := time.Date(2026, 9, 10, 10, 0, 0, 0, time.UTC)
|
|
|
|
raw, err := bson.Marshal(Instance{InstanceID: "i1", LockedAt: &at, PurgeAfter: &at})
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
var set bson.M
|
|
if err := bson.Unmarshal(raw, &set); err != nil {
|
|
t.Fatalf("unmarshal: %v", err)
|
|
}
|
|
for _, k := range []string{"locked_at", "purge_after"} {
|
|
if _, ok := set[k]; !ok {
|
|
t.Errorf("%s missing from %v", k, set)
|
|
}
|
|
}
|
|
|
|
raw, err = bson.Marshal(Instance{InstanceID: "i1"})
|
|
if err != nil {
|
|
t.Fatalf("marshal: %v", err)
|
|
}
|
|
var unset bson.M
|
|
if err := bson.Unmarshal(raw, &unset); err != nil {
|
|
t.Fatalf("unmarshal: %v", err)
|
|
}
|
|
for _, k := range []string{"locked_at", "purge_after"} {
|
|
if _, ok := unset[k]; ok {
|
|
t.Errorf("%s written for an unlocked instance: %v", k, unset)
|
|
}
|
|
}
|
|
}
|