StasonJatham
a366905a43
Logging: - structured slog with LOG_FORMAT=json|text and a stable key vocabulary - periodic stats record with queue depths, db pool, goroutines, memory, dns/probe - no per-certificate logging (hot path is counters only) UI (Bucketeer branding, DESIGN.md): - light+dark themes (OS default, persisted header toggle, no first-paint flash) - official mascot: lossless WebP light/dark logos + favicon, embedded and served with cache headers (1.2MB source PNGs -> ~27/22KB webp) - Assets/Contents views, pagination (offset) and native virtual scrolling (content-visibility) so large result sets do not blow up the DOM - search with debounce/abort + URL state, exposure chips, sort, summary counts Readability + indexing: - internal/index listers for S3-compatible, GCS, Azure, Firebase, OpenSearch (metadata only: keys/sizes/dates; never object contents) - asset_entries table + store write/search; readable resources are indexed and searchable in the UI - deep, unauthenticated, read-only DB probes (PROBE_DEEP): Redis PING, PostgreSQL StartupMessage, MongoDB OP_MSG listDatabases; OpenSearch/Neptune HTTP reads - Azure account-level listing fixed: 403 is reported reachable, not auth_required, since anonymous access is per-container DNS: - curated reliable resolvers by default; DNS_SERVERS=fresh adds the compiled-in Fresh-Resolvers snapshot with a bounded startup sample (warming all 6.7k at once broke the container resolver) - random healthy selection + failover + per-server cooldown Other: prober Close fixes a shutdown hang; config/README updated.
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package store
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"software.sslmate.com/src/certspotter/cttypes"
|
|
|
|
"tea.karlcom.de/karl/bucketeer/internal/model"
|
|
"tea.karlcom.de/karl/bucketeer/internal/pipeline"
|
|
)
|
|
|
|
func TestSTHKeyStableAndUnique(t *testing.T) {
|
|
a := &cttypes.SignedTreeHead{TreeSize: 10, Timestamp: 100}
|
|
if sthKey(a) != sthKey(a) {
|
|
t.Fatal("sthKey is not deterministic")
|
|
}
|
|
b := &cttypes.SignedTreeHead{TreeSize: 11, Timestamp: 100}
|
|
if sthKey(a) == sthKey(b) {
|
|
t.Fatal("sthKey collides for different tree sizes")
|
|
}
|
|
}
|
|
|
|
func TestNonNil(t *testing.T) {
|
|
if got := nonNil(nil); got == nil || len(got) != 0 {
|
|
t.Fatalf("nonNil(nil) = %#v", got)
|
|
}
|
|
in := []string{"a"}
|
|
if got := nonNil(in); len(got) != 1 || got[0] != "a" {
|
|
t.Fatalf("nonNil changed slice: %#v", got)
|
|
}
|
|
}
|
|
|
|
// TestIntegration exercises the real schema when DATABASE_URL is set.
|
|
func TestIntegration(t *testing.T) {
|
|
url := os.Getenv("DATABASE_URL")
|
|
if url == "" {
|
|
t.Skip("DATABASE_URL not set; skipping integration test")
|
|
}
|
|
ctx := t.Context()
|
|
s, err := Open(ctx, url, 2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer s.Close()
|
|
if err := s.Migrate(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f := pipeline.Finding{
|
|
Candidate: model.Candidate{
|
|
Hostname: "it-bucket.s3.eu-central-1.amazonaws.com",
|
|
Original: "it-bucket.s3.eu-central-1.amazonaws.com",
|
|
Provider: model.ProviderAWS,
|
|
Service: model.ServiceS3,
|
|
Source: model.SourceDirectCT,
|
|
},
|
|
IPv4: []string{"1.2.3.4"},
|
|
}
|
|
if err := s.WriteFindings(ctx, []pipeline.Finding{f}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := s.UpdateExposure(ctx, model.ProviderAWS.String(), model.ServiceS3.String(), f.Hostname, model.ExposureReachable, "test"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := s.ListAssets(ctx, AssetFilter{Provider: "aws", Limit: 10}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := s.CountAssets(ctx, AssetFilter{Provider: "aws"}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := s.Stats(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
n, err := s.ReplaceEntries(ctx, model.ProviderAWS.String(), model.ServiceS3.String(), f.Hostname, []model.Entry{
|
|
{Kind: "object", Name: "it/index/a.txt", Size: 10},
|
|
{Kind: "prefix", Name: "it/index/"},
|
|
})
|
|
if err != nil || n != 2 {
|
|
t.Fatalf("ReplaceEntries = %d, %v", n, err)
|
|
}
|
|
entries, total, err := s.SearchEntries(ctx, "index/a", 10, 0)
|
|
if err != nil || total != 1 || len(entries) != 1 || entries[0].ProviderHost != f.Hostname {
|
|
t.Fatalf("SearchEntries = %+v total=%d err=%v", entries, total, err)
|
|
}
|
|
}
|