103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
package coordinator_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/influxdb/coordinator"
|
|
"github.com/influxdata/influxdb/influxql"
|
|
"github.com/influxdata/influxdb/services/meta"
|
|
"github.com/influxdata/influxdb/tsdb"
|
|
)
|
|
|
|
func TestLocalShardMapper(t *testing.T) {
|
|
var metaClient MetaClient
|
|
metaClient.ShardGroupsByTimeRangeFn = func(database, policy string, min, max time.Time) ([]meta.ShardGroupInfo, error) {
|
|
if database != "db0" {
|
|
t.Errorf("unexpected database: %s", database)
|
|
}
|
|
if policy != "rp0" {
|
|
t.Errorf("unexpected retention policy: %s", policy)
|
|
}
|
|
return []meta.ShardGroupInfo{
|
|
{ID: 1, Shards: []meta.ShardInfo{
|
|
{ID: 1, Owners: []meta.ShardOwner{{NodeID: 0}}},
|
|
{ID: 2, Owners: []meta.ShardOwner{{NodeID: 0}}},
|
|
}},
|
|
{ID: 2, Shards: []meta.ShardInfo{
|
|
{ID: 3, Owners: []meta.ShardOwner{{NodeID: 0}}},
|
|
{ID: 4, Owners: []meta.ShardOwner{{NodeID: 0}}},
|
|
}},
|
|
}, nil
|
|
}
|
|
|
|
var tsdbStore TSDBStore
|
|
tsdbStore.ShardGroupFn = func(ids []uint64) tsdb.ShardGroup {
|
|
if !reflect.DeepEqual(ids, []uint64{1, 2, 3, 4}) {
|
|
t.Errorf("unexpected shard ids: %#v", ids)
|
|
}
|
|
|
|
var sh MockShard
|
|
sh.CreateIteratorFn = func(measurement string, opt influxql.IteratorOptions) (influxql.Iterator, error) {
|
|
if measurement != "cpu" {
|
|
t.Errorf("unexpected measurement: %s", measurement)
|
|
}
|
|
return &FloatIterator{}, nil
|
|
}
|
|
return &sh
|
|
}
|
|
|
|
// Initialize the shard mapper.
|
|
shardMapper := &coordinator.LocalShardMapper{
|
|
MetaClient: &metaClient,
|
|
TSDBStore: &tsdbStore,
|
|
}
|
|
|
|
// Normal measurement.
|
|
measurement := &influxql.Measurement{
|
|
Database: "db0",
|
|
RetentionPolicy: "rp0",
|
|
Name: "cpu",
|
|
}
|
|
ic, err := shardMapper.MapShards([]influxql.Source{measurement}, &influxql.SelectOptions{})
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
|
|
// This should be a LocalShardMapping.
|
|
m, ok := ic.(*coordinator.LocalShardMapping)
|
|
if !ok {
|
|
t.Fatalf("unexpected mapping type: %T", ic)
|
|
} else if len(m.ShardMap) != 1 {
|
|
t.Fatalf("unexpected number of shard mappings: %d", len(m.ShardMap))
|
|
}
|
|
|
|
if _, err := ic.CreateIterator(measurement, influxql.IteratorOptions{}); err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
|
|
// Subquery.
|
|
subquery := &influxql.SubQuery{
|
|
Statement: &influxql.SelectStatement{
|
|
Sources: []influxql.Source{measurement},
|
|
},
|
|
}
|
|
ic, err = shardMapper.MapShards([]influxql.Source{subquery}, &influxql.SelectOptions{})
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
|
|
// This should be a LocalShardMapping.
|
|
m, ok = ic.(*coordinator.LocalShardMapping)
|
|
if !ok {
|
|
t.Fatalf("unexpected mapping type: %T", ic)
|
|
} else if len(m.ShardMap) != 1 {
|
|
t.Fatalf("unexpected number of shard mappings: %d", len(m.ShardMap))
|
|
}
|
|
|
|
if _, err := ic.CreateIterator(measurement, influxql.IteratorOptions{}); err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
}
|