added tests

jakopako · Jun 17, 2024 · 5db2ccd · 5db2ccd
1 parent 6e05b29
commit 5db2ccd
Showing 1 changed file with 184 additions and 4 deletions.
diff --git a/scraper/scraper_test.go b/scraper/scraper_test.go
@@ -650,14 +650,194 @@ func TestExtractFieldDate29Feb(t *testing.T) {
 }
 
 func TestGuessYearSimple(t *testing.T) {
-	// events span period around change of year
-
+	// items dates span period around change of year
+	s := &Scraper{
+		Fields: []Field{
+			{
+				Type:      "date",
+				GuessYear: true,
+				Name:      "date",
+			},
+		},
+	}
+	loc, _ := time.LoadLocation("CET")
+	items := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 1, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	expectedItems := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2024, 1, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	s.guessYear(items, time.Date(2023, 11, 30, 20, 30, 0, 0, loc))
+	for i, d := range items {
+		if d["date"] != expectedItems[i]["date"] {
+			t.Fatalf("expected '%v' as year of date but got '%v'", expectedItems[i]["date"], d["date"])
+		}
+	}
 }
 
 func TestGuessYearUnordered(t *testing.T) {
-	// events are not perfectly ordered
+	// items dates are not perfectly ordered and span
+	// period around change of year
+	s := &Scraper{
+		Fields: []Field{
+			{
+				Type:      "date",
+				GuessYear: true,
+				Name:      "date",
+			},
+		},
+	}
+	loc, _ := time.LoadLocation("CET")
+	items := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 11, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 14, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 1, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	expectedItems := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 11, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 14, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2024, 1, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	s.guessYear(items, time.Date(2023, 11, 1, 20, 30, 0, 0, loc))
+	for i, d := range items {
+		if d["date"] != expectedItems[i]["date"] {
+			t.Fatalf("expected '%v' as year of date but got '%v'", expectedItems[i]["date"], d["date"])
+		}
+	}
 }
 
 func TestGuessYear2Years(t *testing.T) {
-	// events span more than 2 years
+	// items dates span more than 2 years
+	s := &Scraper{
+		Fields: []Field{
+			{
+				Type:      "date",
+				GuessYear: true,
+				Name:      "date",
+			},
+		},
+	}
+	loc, _ := time.LoadLocation("CET")
+	items := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 1, 14, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 5, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 9, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 2, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	expectedItems := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2024, 1, 14, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2024, 5, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2024, 9, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2025, 2, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	s.guessYear(items, time.Date(2023, 11, 1, 20, 30, 0, 0, loc))
+	for i, d := range items {
+		if d["date"] != expectedItems[i]["date"] {
+			t.Fatalf("expected '%v' as year of date but got '%v'", expectedItems[i]["date"], d["date"])
+		}
+	}
+}
+
+func TestGuessYearStartBeforeReference(t *testing.T) {
+	// items date start before given reference
+	s := &Scraper{
+		Fields: []Field{
+			{
+				Type:      "date",
+				GuessYear: true,
+				Name:      "date",
+			},
+		},
+	}
+	loc, _ := time.LoadLocation("CET")
+	items := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 1, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	expectedItems := []map[string]interface{}{
+		{
+			"date": time.Date(2023, 12, 2, 20, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2023, 12, 24, 21, 30, 0, 0, loc),
+		},
+		{
+			"date": time.Date(2024, 1, 2, 20, 0, 0, 0, loc),
+		},
+	}
+	s.guessYear(items, time.Date(2024, 1, 30, 20, 30, 0, 0, loc))
+	for i, d := range items {
+		if d["date"] != expectedItems[i]["date"] {
+			t.Fatalf("expected '%v' as year of date but got '%v'", expectedItems[i]["date"], d["date"])
+		}
+	}
 }