chronograf/bolt/client_test.go

package bolt_test

import (
	"errors"
	"io/ioutil"
	"os"
	"time"

	"github.com/influxdata/chronograf/bolt"
)

// TestNow is a set time for testing.
var TestNow = time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC)

// TestClient wraps *bolt.Client.
type TestClient struct {
	*bolt.Client
}

// NewTestClient creates new *bolt.Client with a set time and temp path.
func NewTestClient() (*TestClient, error) {
	f, err := ioutil.TempFile("", "chronograf-bolt-")
	if err != nil {
		return nil, errors.New("unable to open temporary boltdb file")
	}
	f.Close()

	c := &TestClient{
		Client: bolt.NewClient(),
	}
	c.Path = f.Name()
	c.Now = func() time.Time { return TestNow }

	return c, nil
}

func (c *TestClient) Close() error {
	defer os.Remove(c.Path)
	return c.Client.Close()
}