40 lines
741 B
Go
40 lines
741 B
Go
package kv_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/influxdb"
|
|
"github.com/influxdata/influxdb/kv"
|
|
)
|
|
|
|
type mockStore struct {
|
|
}
|
|
|
|
func (s mockStore) View(context.Context, func(kv.Tx) error) error {
|
|
return nil
|
|
}
|
|
|
|
func (s mockStore) Update(context.Context, func(kv.Tx) error) error {
|
|
return nil
|
|
}
|
|
|
|
func TestNewService(t *testing.T) {
|
|
s := kv.NewService(mockStore{})
|
|
|
|
if s.Config.SessionLength != influxdb.DefaultSessionLength {
|
|
t.Errorf("Service session length should use default length when not set")
|
|
}
|
|
|
|
config := kv.ServiceConfig{
|
|
SessionLength: time.Duration(time.Hour * 4),
|
|
}
|
|
|
|
s = kv.NewService(mockStore{}, config)
|
|
|
|
if s.Config != config {
|
|
t.Errorf("Service config not set by constructor")
|
|
}
|
|
}
|