Initial commit

internal/classifier/gating_test.go

@@ -0,0 +1,74 @@
+package classifier
+
+import "testing"
+
+func TestGating_Stage1_DefinitePost(t *testing.T) {
+	tests := []struct {
+		name     string
+		headline string
+		lede     string
+	}{
+		{"head of state", "President signs new climate bill", "The president signed legislation today."},
+		{"international body", "WHO declares new pandemic phase", "The World Health Organization announced a new phase."},
+		{"casualties", "12 killed in factory explosion", "Authorities report 12 dead after an explosion."},
+		{"disaster", "Earthquake strikes northern Japan", "A major earthquake hit the region early Tuesday."},
+		{"conflict", "Airstrike hits hospital in conflict zone", "An airstrike damaged a hospital."},
+		{"crisis", "Economic crisis deepens in Argentina", "The ongoing crisis has intensified."},
+		{"financial scale", "EU approves $50bn climate fund", "The European Union approved funding."},
+		{"assassination", "Assassination attempt on opposition leader", "An armed assailant attacked the leader."},
+		{"NATO", "NATO expands eastern flank deployment", "Alliance members agreed to new deployments."},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result, reason := RunGating(tt.headline, tt.lede)
+			if result != GatingPost {
+				t.Errorf("expected GatingPost, got %d (reason: %s)", result, reason)
+			}
+		})
+	}
+}
+
+func TestGating_Stage2_DefiniteDiscard(t *testing.T) {
+	tests := []struct {
+		name     string
+		headline string
+		lede     string
+	}{
+		{"local government", "City council approves new parking meters", "The council voted 5-3 in favor."},
+		{"sports scores", "Lakers defeat Celtics 112-98 in regular season", "Final score 112-98."},
+		{"celebrity gossip", "Celebrity couple spotted on red carpet at Cannes", "The pair arrived together."},
+		{"routine earnings", "Regional bank reports Q2 results slightly above estimates", "Quarterly earnings were released."},
+		{"entertainment", "Pop star announces new album release date", "Fans have been waiting since 2023."},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result, reason := RunGating(tt.headline, tt.lede)
+			if result != GatingDiscard {
+				t.Errorf("expected GatingDiscard, got %d (reason: %s)", result, reason)
+			}
+		})
+	}
+}
+
+func TestGating_Stage3_Ambiguous(t *testing.T) {
+	tests := []struct {
+		name     string
+		headline string
+		lede     string
+	}{
+		{"tech product", "Apple unveils new MacBook Pro with M5 chip", "The new laptops feature significant performance gains."},
+		{"policy nuanced", "Government considers new data privacy framework", "Ministers met to discuss potential changes."},
+		{"science", "Researchers discover high-temperature superconductor", "A team at MIT published findings in Nature."},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result, reason := RunGating(tt.headline, tt.lede)
+			if result != GatingAmbiguous {
+				t.Errorf("expected GatingAmbiguous, got %d (reason: %s)", result, reason)
+			}
+		})
+	}
+}