NotAShelf/watchdog
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

watchdog: migrate to Cobra and Viper for config management; search /etc for configs

Browse source 
Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: I65dbf466cb030dccc7025585d6282bd26a6a6964
This commit is contained in:
raf 2026-03-01 18:09:34 +03:00
commit f988174145
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
9 changed files with 187 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

28
test/integration_test.go
View file

@ -39,14 +39,14 @@ func TestEndToEnd_BasicFlow(t *testing.T) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, *cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
// Register metrics
promRegistry := prometheus.NewRegistry()
metricsAgg.MustRegister(promRegistry)
// Create handlers
ingestionHandler := api.NewIngestionHandler(*cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
ingestionHandler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
metricsHandler := promhttp.HandlerFor(promRegistry, promhttp.HandlerOpts{})
// Test pageview ingestion
@ -152,12 +152,12 @@ func TestEndToEnd_CardinalityLimits(t *testing.T) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, &cfg)
promRegistry := prometheus.NewRegistry()
metricsAgg.MustRegister(promRegistry)
handler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
handler := api.NewIngestionHandler(&cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
// Send more paths than limit
t.Run("PathOverflow", func(t *testing.T) {
@ -200,11 +200,11 @@ func TestEndToEnd_GracefulShutdown(t *testing.T) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, &cfg)
// Send some events to populate HLL
pathNorm := normalize.NewPathNormalizer(cfg.Site.Path)
handler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
handler := api.NewIngestionHandler(&cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
for i := 0; i < 10; i++ {
event := `{"d":"test.example.com","p":"/","r":"","w":1920}`
@ -247,9 +247,9 @@ func TestEndToEnd_InvalidRequests(t *testing.T) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, &cfg)
handler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
handler := api.NewIngestionHandler(&cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
tests := []struct {
name string
@ -327,9 +327,9 @@ func TestEndToEnd_RateLimiting(t *testing.T) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, &cfg)
handler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
handler := api.NewIngestionHandler(&cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
// Send requests until rate limited
rateLimited := false
@ -376,9 +376,9 @@ func TestEndToEnd_CORS(t *testing.T) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, &cfg)
handler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
handler := api.NewIngestionHandler(&cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
// Test OPTIONS preflight
t.Run("Preflight", func(t *testing.T) {
@ -435,9 +435,9 @@ func BenchmarkIngestionThroughput(b *testing.B) {
pathRegistry := aggregate.NewPathRegistry(cfg.Limits.MaxPaths)
refRegistry := normalize.NewReferrerRegistry(cfg.Limits.MaxSources)
eventRegistry := aggregate.NewCustomEventRegistry(cfg.Limits.MaxCustomEvents)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, cfg)
metricsAgg := aggregate.NewMetricsAggregator(pathRegistry, eventRegistry, &cfg)
handler := api.NewIngestionHandler(cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
handler := api.NewIngestionHandler(&cfg, pathNorm, pathRegistry, refRegistry, metricsAgg)
event := `{"d":"test.example.com","p":"/","r":"https://google.com","w":1920}`