package sources import ( "context" "net/http" "net/http/httptest" "testing" "time" "github.com/salvacybersec/keyhunter/pkg/providers" "github.com/salvacybersec/keyhunter/pkg/recon" ) func TestAPKMirror_Name(t *testing.T) { s := &APKMirrorSource{} if s.Name() != "apkmirror" { t.Fatalf("expected apkmirror, got %s", s.Name()) } } func TestAPKMirror_Enabled(t *testing.T) { s := &APKMirrorSource{} if !s.Enabled(recon.Config{}) { t.Fatal("APKMirrorSource should always be enabled") } } func TestAPKMirror_RespectsRobots(t *testing.T) { s := &APKMirrorSource{} if !s.RespectsRobots() { t.Fatal("APKMirrorSource should respect robots.txt") } } func TestAPKMirror_Sweep(t *testing.T) { srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "text/html") _, _ = w.Write([]byte(`

AI Chat Pro

Uses api_key = "sk-proj-ABCDEF1234567890abcdef" for backend

`)) })) defer srv.Close() reg := providers.NewRegistryFromProviders([]providers.Provider{ {Name: "openai", Keywords: []string{"sk-proj-"}}, }) s := &APKMirrorSource{ BaseURL: srv.URL, Registry: reg, Client: NewClient(), } out := make(chan recon.Finding, 10) ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() err := s.Sweep(ctx, "", out) close(out) if err != nil { t.Fatalf("Sweep error: %v", err) } var findings []recon.Finding for f := range out { findings = append(findings, f) } if len(findings) == 0 { t.Fatal("expected at least one finding from APKMirror") } if findings[0].SourceType != "recon:apkmirror" { t.Fatalf("expected recon:apkmirror, got %s", findings[0].SourceType) } } func TestAPKMirror_Sweep_NoMatch(t *testing.T) { srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "text/html") _, _ = w.Write([]byte(`

No API keys here