From c17f67b0a9fe7ff01bd7ebedb8afb931f695cb39 Mon Sep 17 00:00:00 2001
From: Seif Eddine GHAZOUANI <se.ghazouani@labsoft.fr>
Date: Tue, 3 Dec 2024 17:12:11 +0100
Subject: [PATCH] Add test case to validate the removal of calendars with
 today's date

---
 .../impl/RemoveOldCalendarStatementsTest.java | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 onebusaway-gtfs-transformer/src/test/java/org/onebusaway/gtfs_transformer/impl/RemoveOldCalendarStatementsTest.java

diff --git a/onebusaway-gtfs-transformer/src/test/java/org/onebusaway/gtfs_transformer/impl/RemoveOldCalendarStatementsTest.java b/onebusaway-gtfs-transformer/src/test/java/org/onebusaway/gtfs_transformer/impl/RemoveOldCalendarStatementsTest.java
new file mode 100644
index 00000000..e246045d
--- /dev/null
+++ b/onebusaway-gtfs-transformer/src/test/java/org/onebusaway/gtfs_transformer/impl/RemoveOldCalendarStatementsTest.java
@@ -0,0 +1,71 @@
+package org.onebusaway.gtfs_transformer.impl;
+
+import static org.junit.Assert.assertEquals;
+
+import java.io.IOException;
+import java.time.LocalDate;
+import java.time.format.DateTimeFormatter;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.onebusaway.gtfs.services.GtfsMutableRelationalDao;
+import org.onebusaway.gtfs.services.MockGtfs;
+import org.onebusaway.gtfs_transformer.services.TransformContext;
+
+public class RemoveOldCalendarStatementsTest {
+
+    private RemoveOldCalendarStatements removeOldCalendarStatements = new RemoveOldCalendarStatements();
+    private TransformContext _context = new TransformContext();
+    private MockGtfs _gtfs;
+
+    @Before
+    public void setup() throws IOException{
+
+        _gtfs = MockGtfs.create();
+        _gtfs.putAgencies(1);
+        _gtfs.putStops(1);
+        _gtfs.putRoutes(1);
+        _gtfs.putTrips(1, "r0", "sid0");
+        _gtfs.putStopTimes("t0", "s0");
+
+        // Insert a calendar entry wtih start and end dates set to today's date 
+        String startDate = getCurrentDateFormatted(-3);
+        String endDate = getCurrentDateFormatted(null);
+
+        _gtfs.putCalendars(
+            1,
+            "start_date="+startDate,
+            "end_date="+endDate
+        );
+    }
+
+    @Test
+    public void testRemoveCalendarForToday() throws IOException  {
+        GtfsMutableRelationalDao dao = _gtfs.read();
+        // Set removeToday to true to allow the removal of the calendar for today's date
+        removeOldCalendarStatements.setRemoveToday(true);
+        removeOldCalendarStatements.run(_context, dao);
+        // Verify that GtfsMutableRelationalDao object no longer contains any calendar entries after removing the calendar for today's date
+        assertEquals(0,dao.getAllCalendars().size());
+    }
+
+    @Test
+    public void testRemoveCalendar() throws IOException  {
+        GtfsMutableRelationalDao dao = _gtfs.read();
+        removeOldCalendarStatements.setRemoveToday(false);
+        removeOldCalendarStatements.run(_context, dao);
+        // Verify that GtfsMutableRelationalDao object still contain the initially added calendar entry
+        assertEquals(1,dao.getAllCalendars().size());
+    }
+
+    // Helper function to get today's date in the required format
+    public static String getCurrentDateFormatted(Integer daysOffset) {
+        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyyMMdd");
+        LocalDate date = LocalDate.now();
+        if (daysOffset != null){
+            date = date.plusDays(daysOffset);
+        }
+        // Format date as "yyyyMMdd"
+        return date.format(formatter);
+    }
+}