90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package meta
|
|
|
|
import (
|
|
"reflect"
|
|
"sort"
|
|
"time"
|
|
|
|
"testing"
|
|
)
|
|
|
|
func Test_newShardOwner(t *testing.T) {
|
|
// An error is returned if there are no data nodes available.
|
|
_, err := NewShardOwner(ShardInfo{}, map[int]int{})
|
|
if err == nil {
|
|
t.Error("got no error, but expected one")
|
|
}
|
|
|
|
ownerFreqs := map[int]int{1: 15, 2: 11, 3: 12}
|
|
id, err := NewShardOwner(ShardInfo{ID: 4}, ownerFreqs)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// The ID that owns the fewest shards is returned.
|
|
if got, exp := id, uint64(2); got != exp {
|
|
t.Errorf("got id %d, expected id %d", got, exp)
|
|
}
|
|
|
|
// The ownership frequencies are updated.
|
|
if got, exp := ownerFreqs, map[int]int{1: 15, 2: 12, 3: 12}; !reflect.DeepEqual(got, exp) {
|
|
t.Errorf("got owner frequencies %v, expected %v", got, exp)
|
|
}
|
|
}
|
|
|
|
func TestShardGroupSort(t *testing.T) {
|
|
sg1 := ShardGroupInfo{
|
|
ID: 1,
|
|
StartTime: time.Unix(1000, 0),
|
|
EndTime: time.Unix(1100, 0),
|
|
TruncatedAt: time.Unix(1050, 0),
|
|
}
|
|
|
|
sg2 := ShardGroupInfo{
|
|
ID: 2,
|
|
StartTime: time.Unix(1000, 0),
|
|
EndTime: time.Unix(1100, 0),
|
|
}
|
|
|
|
sgs := ShardGroupInfos{sg2, sg1}
|
|
|
|
sort.Sort(sgs)
|
|
|
|
if sgs[len(sgs)-1].ID != 2 {
|
|
t.Fatal("unstable sort for ShardGroupInfos")
|
|
}
|
|
}
|
|
|
|
func Test_Data_RetentionPolicy_MarshalBinary(t *testing.T) {
|
|
zeroTime := time.Time{}
|
|
epoch := time.Unix(0, 0).UTC()
|
|
|
|
startTime := zeroTime
|
|
sgi := &ShardGroupInfo{
|
|
StartTime: startTime,
|
|
}
|
|
isgi := sgi.marshal()
|
|
sgi.unmarshal(isgi)
|
|
if got, exp := sgi.StartTime.UTC(), epoch.UTC(); got != exp {
|
|
t.Errorf("unexpected start time. got: %s, exp: %s", got, exp)
|
|
}
|
|
|
|
startTime = time.Unix(0, 0)
|
|
endTime := startTime.Add(time.Hour * 24)
|
|
sgi = &ShardGroupInfo{
|
|
StartTime: startTime,
|
|
EndTime: endTime,
|
|
}
|
|
isgi = sgi.marshal()
|
|
sgi.unmarshal(isgi)
|
|
if got, exp := sgi.StartTime.UTC(), startTime.UTC(); got != exp {
|
|
t.Errorf("unexpected start time. got: %s, exp: %s", got, exp)
|
|
}
|
|
if got, exp := sgi.EndTime.UTC(), endTime.UTC(); got != exp {
|
|
t.Errorf("unexpected end time. got: %s, exp: %s", got, exp)
|
|
}
|
|
if got, exp := sgi.DeletedAt.UTC(), zeroTime.UTC(); got != exp {
|
|
t.Errorf("unexpected DeletedAt time. got: %s, exp: %s", got, exp)
|
|
}
|
|
}
|