package com.battery.device.bts;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import java.time.Duration;
import java.time.ZoneId;
import java.util.Set;
import org.junit.jupiter.api.Test;
import com.battery.device.api.DeviceAdapter;
import com.battery.device.model.ChannelIdentity;
import com.battery.device.model.DeviceControlCommand;
import com.battery.device.model.DeviceWorkStatus;

class BtsDeviceAdapterContractTest
{
    @Test
    void observesCompleteReadOnlyContractWithoutSendingControlCommands() throws Exception
    {
        try (ScriptedBtsTcpSimulator simulator = new ScriptedBtsTcpSimulator();
             BtsTcpTransport transport = new BtsTcpTransport("127.0.0.1", simulator.port(),
                     Duration.ofSeconds(1), Duration.ofSeconds(2));
             DeviceAdapter adapter = new BtsDeviceAdapter("lab-bts", transport,
                     new BtsXmlCodec("lab-bts", ZoneId.of("Asia/Hong_Kong"))))
        {
            ChannelIdentity first = new ChannelIdentity("lab-bts", 22, 3, 2, 8);
            ChannelIdentity second = new ChannelIdentity("lab-bts", 22, 3, 1, 6);
            simulator.enqueue("<bts><cmd>getdevinfo_resp</cmd><middle count=\"2\">"
                    + "<channel ip=\"192.168.1.35\" devtype=\"22\" devid=\"3\" subdevid=\"2\" Channelid=\"8\">true</channel>"
                    + "<channel ip=\"192.168.1.36\" devtype=\"22\" devid=\"3\" subdevid=\"1\" Channelid=\"6\">true</channel>"
                    + "</middle></bts>");
            simulator.enqueue("<bts><cmd>inquire_resp</cmd><list count=\"2\">"
                    + "<inquire dev=\"22-3-2-8-0\" workstatus=\"working\" voltage=\"3.5\"/>"
                    + "<inquire dev=\"22-3-1-6-0\" workstatus=\"futureState\" voltage=\"3.6\"/>"
                    + "</list></bts>");
            simulator.enqueue("<bts><cmd>inquiredf_resp</cmd><list count=\"2\">"
                    + "<chl devtype=\"22\" devid=\"3\" subdevid=\"2\" chlid=\"8\" testid=\"11\" count=\"3\">true</chl>"
                    + "<chl devtype=\"22\" devid=\"3\" subdevid=\"1\" chlid=\"6\" testid=\"10\" count=\"11\">false</chl>"
                    + "</list></bts>");
            simulator.enqueue("<bts><cmd>download_resp</cmd>"
                    + "<download devtype=\"22\" devid=\"3\" subdevid=\"2\" chlid=\"8\" testid=\"11\"/>"
                    + "<list count=\"3\">"
                    + historyData(3) + historyData(1) + historyData(3) + "</list></bts>");

            assertEquals(2, adapter.discoverChannels().size());
            var snapshots = adapter.inquire(Set.of(first, second));
            var history = adapter.inquireHistory(Set.of(first, second));
            var page = adapter.downloadHistory(first, "11", 1, 1000);

            assertTrue(adapter.capabilities().discovery());
            assertTrue(adapter.capabilities().realtimeObservation());
            assertTrue(adapter.capabilities().historyDownload());
            assertFalse(adapter.capabilities().control());
            assertEquals(DeviceWorkStatus.UNKNOWN, snapshots.get(second).workStatus());
            assertFalse(history.get(1).uploadComplete());
            assertEquals(4, page.nextPosition());
            assertEquals(java.util.List.of(3L, 1L, 3L),
                    page.samples().stream().map(sample -> sample.sequenceId()).toList());

            int observedRequests = simulator.requests().size();
            for (DeviceControlCommand command : DeviceControlCommand.values())
                assertThrows(UnsupportedOperationException.class, () -> adapter.control(first, command));
            assertEquals(observedRequests, simulator.requests().size());
            assertEquals(4, observedRequests);
        }
    }

    @Test
    void surfacesMalformedResponseScript() throws Exception
    {
        try (ScriptedBtsTcpSimulator simulator = new ScriptedBtsTcpSimulator();
             BtsTcpTransport transport = new BtsTcpTransport("127.0.0.1", simulator.port(),
                     Duration.ofSeconds(1), Duration.ofSeconds(1));
             DeviceAdapter adapter = new BtsDeviceAdapter("lab-bts", transport,
                     new BtsXmlCodec("lab-bts", ZoneId.of("Asia/Hong_Kong"))))
        {
            simulator.enqueue("<bts><cmd>getdevinfo_resp</cmd>");

            BtsProtocolException exception = assertThrows(BtsProtocolException.class, adapter::discoverChannels);

            assertEquals(BtsProtocolException.Reason.MALFORMED_XML, exception.reason());
        }
    }

    private static String historyData(long sequenceId)
    {
        return "<data seqid=\"" + sequenceId + "\" stepid=\"1\" cycleid=\"1\" steptype=\"cc\" "
                + "atime=\"2026-08-25 18.00.00\" volt=\"3.5\" curr=\"1.2\" cap=\"0.1\" eng=\"0.35\"/>";
    }
}