blocky/config/query_log_test.go

package config

import (
	"time"

	"github.com/creasty/defaults"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("QueryLogConfig", func() {
	var cfg QueryLog

	suiteBeforeEach()

	BeforeEach(func() {
		cfg = QueryLog{
			Target:           "/dev/null",
			Type:             QueryLogTypeCsvClient,
			LogRetentionDays: 0,
			CreationAttempts: 1,
			CreationCooldown: Duration(time.Millisecond),
		}
	})

	Describe("IsEnabled", func() {
		It("should be true by default", func() {
			cfg := QueryLog{}
			Expect(defaults.Set(&cfg)).Should(Succeed())

			Expect(cfg.IsEnabled()).Should(BeTrue())
		})

		When("enabled", func() {
			It("should be true", func() {
				Expect(cfg.IsEnabled()).Should(BeTrue())
			})
		})

		When("disabled", func() {
			It("should be false", func() {
				cfg := QueryLog{
					Type: QueryLogTypeNone,
				}

				Expect(cfg.IsEnabled()).Should(BeFalse())
			})
		})
	})

	Describe("LogConfig", func() {
		It("should log configuration", func() {
			cfg.LogConfig(logger)

			Expect(hook.Calls).ShouldNot(BeEmpty())
			Expect(hook.Messages).Should(ContainElement(ContainSubstring("logRetentionDays:")))
			Expect(hook.Messages).Should(ContainElement(ContainSubstring("sudn:")))
		})

		DescribeTable("secret censoring", func(target string) {
			cfg.Type = QueryLogTypeMysql
			cfg.Target = target

			cfg.LogConfig(logger)

			Expect(hook.Calls).ShouldNot(BeEmpty())
			Expect(hook.Messages).ShouldNot(ContainElement(ContainSubstring("password")))
		},
			Entry("without scheme", "user:password@localhost"),
			Entry("with scheme", "scheme://user:password@localhost"),
			Entry("no password", "localhost"),
			Entry("not a URL", "invalid!://"),
		)
	})

	Describe("SetDefaults", func() {
		It("should log configuration", func() {
			cfg := QueryLog{}
			Expect(cfg.Fields).Should(BeEmpty())

			Expect(defaults.Set(&cfg)).Should(Succeed())

			Expect(cfg.Fields).ShouldNot(BeEmpty())
		})
	})
})