docdb/internal/util/telemetry/telemetry_test.go

// Copyright 2021 DocDB Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package telemetry

import (
	"testing"

	"github.com/AlekSi/pointer"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/hanzoai/docdb/internal/util/testutil"
)

func TestState(t *testing.T) {
	t.Parallel()

	for name, tc := range map[string]struct {
		flag     string
		dnt      string
		execName string
		prev     *bool
		state    *bool
		locked   bool
		err      string
	}{
		"default": {},
		"prev": {
			prev:  pointer.ToBool(false),
			state: pointer.ToBool(false),
		},
		"flag": {
			flag:   "disable",
			prev:   pointer.ToBool(true),
			state:  pointer.ToBool(false),
			locked: true,
		},
		"dnt": {
			dnt:    "1",
			state:  pointer.ToBool(false),
			locked: true,
		},
		"invalidDnt": {
			dnt: "foo",
			err: "failed to parse foo",
		},
		"conflict": {
			flag:     "enable",
			execName: "DoNotTrack",
			err:      "telemetry can't be enabled",
		},
	} {
		t.Run(name, func(t *testing.T) {
			t.Parallel()

			var f Flag
			err := f.UnmarshalText([]byte(tc.flag))
			require.NoError(t, err)

			logger := testutil.Logger(t)

			state, locked, err := initialState(&f, tc.dnt, tc.execName, tc.prev, logger)
			if tc.err != "" {
				assert.EqualError(t, err, tc.err)
				return
			}
			assert.NoError(t, err)
			assert.Equal(t, tc.state, state)
			assert.Equal(t, tc.locked, locked)
		})
	}
}