fleet/server/datastore/datastore_options_test.go

package datastore

import (
	"reflect"
	"sort"
	"testing"

	"github.com/kolide/fleet/server/datastore/internal/appstate"
	"github.com/kolide/fleet/server/kolide"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func testOptions(t *testing.T, ds kolide.Datastore) {
	if ds.Name() == "inmem" {
		t.Skip("inmem is being depracated")
	}
	require.Nil(t, ds.MigrateData())
	// were options pre-loaded?
	opts, err := ds.ListOptions()
	require.Nil(t, err)
	assert.Len(t, opts, len(appstate.Options()))

	opt, err := ds.OptionByName("aws_profile_name")
	require.Nil(t, err)
	assert.False(t, opt.OptionSet())

	// try to save non-readonly list of options with same values, it should not err out
	var writableOpts []kolide.Option
	for _, o := range opts {
		if !o.ReadOnly {
			writableOpts = append(writableOpts, o)
		}
	}
	err = ds.SaveOptions(writableOpts)
	assert.Nil(t, err)

	opt, err = ds.OptionByName("aws_access_key_id")
	require.Nil(t, err)
	require.NotNil(t, opt)
	opt2, err := ds.Option(opt.ID)
	require.Nil(t, err)
	require.NotNil(t, opt2)
	assert.True(t, reflect.DeepEqual(opt, opt2))

	opt.SetValue("somekey")
	err = ds.SaveOptions([]kolide.Option{*opt})
	require.Nil(t, err)
	opt, err = ds.Option(opt.ID)
	require.Nil(t, err)
	assert.Equal(t, "somekey", opt.GetValue())

	// can't change a read only option
	opt, err = ds.OptionByName("disable_distributed")
	require.Nil(t, err)
	opt.SetValue(true)
	err = ds.SaveOptions([]kolide.Option{*opt})
	assert.NotNil(t, err)
	// check that it didn't change
	opt, err = ds.OptionByName("disable_distributed")
	require.Nil(t, err)
	require.False(t, opt.GetValue().(bool))

	opt, _ = ds.OptionByName("aws_profile_name")
	assert.False(t, opt.OptionSet())
	opt.SetValue("zip")
	opt2, _ = ds.OptionByName("disable_distributed")
	assert.Equal(t, false, opt2.GetValue())
	opt2.SetValue(true)
	modList := []kolide.Option{*opt, *opt2}
	// The aws access key option can be saved but because the disable_events can't
	// be we want to verify that the whole transaction is rolled back
	tx, err := ds.Begin()
	require.Nil(t, err)
	err = ds.SaveOptions(modList, kolide.HasTransaction(tx))
	assert.NotNil(t, err)
	err = tx.Rollback()
	require.Nil(t, err)

	opt2, err = ds.OptionByName("disable_distributed")
	require.Nil(t, err)
	assert.Equal(t, false, opt2.GetValue())
	opt, err = ds.OptionByName("aws_profile_name")
	require.Nil(t, err)
	assert.False(t, opt.OptionSet())

}

func testOptionsToConfig(t *testing.T, ds kolide.Datastore) {
	require.Nil(t, ds.MigrateData())
	resp, err := ds.GetOsqueryConfigOptions()
	require.Nil(t, err)
	assert.Len(t, resp, 10)
	assert.Equal(t, "/api/v1/osquery/distributed/read", resp["distributed_tls_read_endpoint"])
	opt, _ := ds.OptionByName("aws_profile_name")
	assert.False(t, opt.OptionSet())
	opt.SetValue("zip")
	err = ds.SaveOptions([]kolide.Option{*opt})
	require.Nil(t, err)
	resp, err = ds.GetOsqueryConfigOptions()
	require.Nil(t, err)
	assert.Len(t, resp, 11)
	assert.Equal(t, "zip", resp["aws_profile_name"])
}

func testResetOptions(t *testing.T, ds kolide.Datastore) {
	if ds.Name() == "inmem" {
		t.Skip("inmem is being deprecated, test skipped")
	}
	require.Nil(t, ds.MigrateData())
	// get originals
	originals, err := ds.ListOptions()
	require.Nil(t, err)
	sort.SliceStable(originals, func(i, j int) bool { return originals[i].ID < originals[j].ID })

	// grab and options, change it, save it, verify that saved
	opt, err := ds.OptionByName("aws_profile_name")
	require.Nil(t, err)
	assert.False(t, opt.OptionSet())
	opt.SetValue("zip")
	err = ds.SaveOptions([]kolide.Option{*opt})
	require.Nil(t, err)
	opt, _ = ds.OptionByName("aws_profile_name")
	assert.Equal(t, "zip", opt.GetValue())

	resetOptions, err := ds.ResetOptions()
	require.Nil(t, err)
	sort.SliceStable(resetOptions, func(i, j int) bool { return resetOptions[i].ID < resetOptions[j].ID })
	require.Equal(t, len(originals), len(resetOptions))

	for i, _ := range originals {
		require.Equal(t, originals[i].ID, resetOptions[i].ID)
		require.Equal(t, originals[i].GetValue(), resetOptions[i].GetValue())
		require.Equal(t, originals[i].Name, resetOptions[i].Name)
	}
}