Fix test methods

Author: kappsegla

src/test/java/com/example/MainTest.java

@@ -147,28 +147,36 @@ void displayMinMaxPrices_withValidData() {
 
     @Test
     void displaySortedPrices_whenRequested() {
-        String mockJson = """
-                [{"SEK_per_kWh":0.30,"EUR_per_kWh":0.03,"EXR":10.0,"time_start":"2025-09-04T00:00:00+02:00","time_end":"2025-09-04T01:00:00+02:00"},
-                 {"SEK_per_kWh":0.10,"EUR_per_kWh":0.01,"EXR":10.0,"time_start":"2025-09-04T01:00:00+02:00","time_end":"2025-09-04T02:00:00+02:00"},
-                 {"SEK_per_kWh":0.20,"EUR_per_kWh":0.02,"EXR":10.0,"time_start":"2025-09-04T02:00:00+02:00","time_end":"2025-09-04T03:00:00+02:00"},
-                 {"SEK_per_kWh":0.10,"EUR_per_kWh":0.01,"EXR":10.0,"time_start":"2025-09-04T03:00:00+02:00","time_end":"2025-09-04T04:00:00+02:00"}]""";
+        // This test ensures charging window can span days when next day data exists
+        LocalDate today = LocalDate.of(2025, 9, 4);
+        LocalDate tomorrow = today.plusDays(1);
 
-        ElpriserAPI.setMockResponse(mockJson);
+        String mockJsonToday = """
+                [{"SEK_per_kWh":0.30,"EUR_per_kWh":0.03,"EXR":10.0,"time_start":"2025-09-04T20:00:00+02:00","time_end":"2025-09-04T21:00:00+02:00"},
+                 {"SEK_per_kWh":0.10,"EUR_per_kWh":0.01,"EXR":10.0,"time_start":"2025-09-04T21:00:00+02:00","time_end":"2025-09-04T22:00:00+02:00"},
+                 {"SEK_per_kWh":0.20,"EUR_per_kWh":0.02,"EXR":10.0,"time_start":"2025-09-04T22:00:00+02:00","time_end":"2025-09-04T23:00:00+02:00"},
+                 {"SEK_per_kWh":0.10,"EUR_per_kWh":0.01,"EXR":10.0,"time_start":"2025-09-04T23:00:00+02:00","time_end":"2025-09-04T00:00:00+02:00"}]""";
+        String mockJsonTomorrow = """
+                [{"SEK_per_kWh":0.10,"EUR_per_kWh":0.01,"EXR":10.0,"time_start":"2025-09-05T00:00:00+02:00","time_end":"2025-09-05T01:00:00+02:00"},
+                 {"SEK_per_kWh":0.15,"EUR_per_kWh":0.015,"EXR":10.0,"time_start":"2025-09-05T01:00:00+02:00","time_end":"2025-09-05T02:00:00+02:00"},
+                 {"SEK_per_kWh":0.15,"EUR_per_kWh":0.015,"EXR":10.0,"time_start":"2025-09-05T02:00:00+02:00","time_end":"2025-09-05T03:00:00+02:00"}]""";
+
+        ElpriserAPI.setMockResponseForDate(today, mockJsonToday);
+        ElpriserAPI.setMockResponseForDate(tomorrow, mockJsonTomorrow);
 
         Main.main(new String[]{"--zone", "SE2", "--date", "2025-09-04", "--sorted"});
 
         String output = bos.toString();
 
         // Expected sorted output (ascending by price)
         List<String> expectedOrder = List.of(
-                "2025-09-04    00-01         30,00 öre",
-                "2025-09-04    00-01         30,00 öre",
-                "2025-09-04    02-03         20,00 öre",
-                "2025-09-04    02-03         20,00 öre",
-                "2025-09-04    01-02         10,00 öre",
-                "2025-09-04    03-04         10,00 öre",
-                "2025-09-04    01-02         10,00 öre",
-                "2025-09-04    03-04         10,00 öre"
+                "2025-09-04 20-21 30,00 öre",
+                "2025-09-04 22-23 20,00 öre",
+                "2025-09-05 01-02 15,00 öre",
+                "2025-09-05 02-03 15,00 öre",
+                "2025-09-04 21-22 10,00 öre",
+                "2025-09-04 23-00 10,00 öre",
+                "2025-09-05 00-01 10,00 öre"
         );
 
         // Extract actual lines that match the pattern