influxdb/kv/service_test.go

package kv_test

import (
	"context"
	"io"
	"testing"
	"time"

	"github.com/influxdata/influxdb/v2"
	"github.com/influxdata/influxdb/v2/kv"
	"go.uber.org/zap/zaptest"
)

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 (s mockStore) Backup(ctx context.Context, w io.Writer) error {
	return nil
}

func TestNewService(t *testing.T) {
	s := kv.NewService(zaptest.NewLogger(t), 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(zaptest.NewLogger(t), mockStore{}, config)

	if s.Config != config {
		t.Errorf("Service config not set by constructor")
	}
}