import json
import unittest

from wideq.client import Client, DeviceInfo
from wideq.washer import WasherDevice, WasherState, WasherStatus


POLL_DATA = {
    "APCourse": "10",
    "DryLevel": "0",
    "Error": "0",
    "Initial_Time_H": "0",
    "Initial_Time_M": "58",
    "LoadLevel": "4",
    "OPCourse": "0",
    "Option1": "0",
    "Option2": "0",
    "Option3": "2",
    "PreState": "23",
    "Remain_Time_H": "0",
    "Remain_Time_M": "13",
    "Reserve_Time_H": "0",
    "Reserve_Time_M": "0",
    "RinseOption": "1",
    "SmartCourse": "51",
    "Soil": "0",
    "SpinSpeed": "5",
    "State": "30",
    "TCLCount": "15",
    "WaterTemp": "4",
}


class WasherStatusTest(unittest.TestCase):
    def setUp(self):
        super().setUp()
        with open("./tests/fixtures/client.json") as fp:
            state = json.load(fp)
        self.client = Client.load(state)
        self.device_info = DeviceInfo(
            {
                "alias": "WASHER",
                "deviceId": "33330ba80-107d-11e9-96c8-0051ede85d3f",
                "deviceType": 201,
                "modelJsonUrl": (
                    "https://aic.lgthinq.com:46030/api/webContents/modelJSON?"
                    "modelName=F3L2CYV5W_WIFI&countryCode=WW&contentsId="
                    "JS1217232703654216&authKey=thinq"
                ),
                "modelNm": "F3L2CYV5W_WIFI",
            }
        )
        self.washer = WasherDevice(self.client, self.device_info)

    def test_properties(self):
        status = WasherStatus(self.washer, POLL_DATA)
        self.assertEqual(WasherState.RINSING, status.state)
        self.assertEqual(WasherState.RUNNING, status.previous_state)
        self.assertTrue(status.is_on)
        self.assertEqual(13, status.remaining_time)
        self.assertEqual(58, status.initial_time)
        self.assertEqual("Towels", status.course)
        self.assertEqual("SmallLoad", status.smart_course)
        self.assertEqual("No Error", status.error)