mirror of
https://github.com/fleetdm/fleet
synced 2026-05-03 21:38:24 +00:00
46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
package ctxdb
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIsPrimaryRequired(t *testing.T) {
|
|
cases := []struct {
|
|
desc string
|
|
ctx context.Context
|
|
want bool
|
|
}{
|
|
{"not set", context.Background(), false},
|
|
{"set to true", RequirePrimary(context.Background(), true), true},
|
|
{"set to false", RequirePrimary(context.Background(), false), false},
|
|
{"set to true then false", RequirePrimary(RequirePrimary(context.Background(), true), false), false},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.desc, func(t *testing.T) {
|
|
got := IsPrimaryRequired(c.ctx)
|
|
require.Equal(t, c.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestIsCachedMysqlBypassed(t *testing.T) {
|
|
cases := []struct {
|
|
desc string
|
|
ctx context.Context
|
|
want bool
|
|
}{
|
|
{"not set", context.Background(), false},
|
|
{"set to true", BypassCachedMysql(context.Background(), true), true},
|
|
{"set to false", BypassCachedMysql(context.Background(), false), false},
|
|
{"set to true then false", BypassCachedMysql(BypassCachedMysql(context.Background(), true), false), false},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.desc, func(t *testing.T) {
|
|
got := IsCachedMysqlBypassed(c.ctx)
|
|
require.Equal(t, c.want, got)
|
|
})
|
|
}
|
|
}
|