package com.battery.business.service;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;
import java.time.Instant;
import java.util.*;
import com.battery.business.domain.*;
import com.battery.business.mapper.TTestMapper;
import org.h2.jdbcx.JdbcDataSource;
import org.junit.jupiter.api.Test;
import org.springframework.jdbc.core.JdbcTemplate;

class DashboardServiceTest {
    @Test
    void aggregatesAllTestsAtHongKongMidnightAndKeepsDeviceFactsSeparate() {
        var database = new JdbcDataSource(); database.setURL("jdbc:h2:mem:dashboard_" + UUID.randomUUID() + ";MODE=MySQL;DB_CLOSE_DELAY=-1");
        var jdbc = new JdbcTemplate(database);
        jdbc.execute("create table t_battery_group(group_id bigint, del_flag int)");
        jdbc.execute("insert into t_battery_group values(1,0),(2,1)");
        jdbc.execute("create table t_test_realtime_record(record_id bigint,test_id bigint,test_battery_id bigint,step_no int,cycle_no int)");
        jdbc.execute("create table t_test_battery(test_battery_id bigint,battery_no varchar(20),del_flag int)");
        jdbc.execute("create table t_test_channel_binding(channel_id bigint,test_id bigint,test_battery_id bigint)");
        jdbc.execute("create table t_detection_channel(channel_id bigint,cabinet_id bigint,del_flag int)");
        jdbc.execute("create table t_test(test_id bigint,test_status varchar(20),del_flag int)");
        jdbc.execute("insert into t_test values(1,'RUNNING',0),(2,'COMPLETED',0)");
        jdbc.execute("insert into t_test_battery values(1,'电池1',0),(2,'电池2',0)");
        jdbc.execute("insert into t_detection_channel values(1,7,0)");
        jdbc.execute("insert into t_test_channel_binding values(1,1,1),(1,2,2)");
        var tests = mock(ITTestService.class);
        var cabinets = mock(ITDetectionCabinetService.class);
        var mapper = mock(TTestMapper.class);
        var rows = new ArrayList<TTest>();
        for (int index = 0; index < 1001; index++) {
            var test = new TTest(); test.setTestId((long) index + 1); test.setTestStatus("RUNNING"); test.setReceptionStatus("PAUSED");
            test.setCreateTime(Date.from(Instant.parse(index == 0 ? "2026-08-31T15:59:59Z" : "2026-08-31T16:00:00Z"))); rows.add(test);
        }
        when(tests.selectTestList(any())).thenReturn(rows);
        var cabinet = new TDetectionCabinet(); cabinet.setCabinetId(7L); cabinet.setChannelCount(8); cabinet.setOnlineStatus("0");
        when(cabinets.selectDetectionCabinetList(any())).thenReturn(List.of(cabinet));
        var warning = new TTestWarningException(); warning.setExceptionId(3L); warning.setTestId(1L); warning.setExceptionType("DEVICE_OFFLINE");
        when(mapper.selectWarningExceptions(null, "OPEN")).thenReturn(List.of(warning));
        var result = new DashboardService(tests, cabinets, mapper, database).read(Instant.parse("2026-09-01T00:00:00Z"));
        assertEquals(1001, result.summary().testCount());
        assertEquals(1, result.summary().batteryGroupCount());
        assertEquals(1000, result.summary().todayCount());
        assertEquals(1000, result.summary().monthCount());
        assertEquals(List.of(1000L), result.trends().get("month").values());
        assertEquals(1, result.trends().get("year").values().get(7));
        assertEquals(1000, result.trends().get("year").values().get(8));
        assertEquals("在线", result.cabinets().get(0).onlineStatus());
        assertEquals(1, result.cabinets().get(0).usedChannels());
        assertEquals(1, result.cabinets().get(0).testCount());
        assertEquals("RUNNING", result.tests().get(0).status());
        assertEquals("PAUSED", result.tests().get(0).receptionStatus());
        assertEquals("设备离线", result.warnings().get(0).item());
    }
}