import report

from mqtt import mqtt_client
from simulation.rooms import house

from .help import testSession
from .help import STATES_SW
from tests.common_testcases import device_follow


def main_light_videv_shelly_sw(ts, testcase_id, room, single=False):
    ts.testCase(
        testcase_id, report.TCEL_SINGLE if single else report.TCEL_SMOKE, device_follow,
        room.videv_main_light, room.videv_main_light.KEY_OUTPUT_0,
        room.main_light, room.main_light.KEY_OUTPUT_0,
        STATES_SW
    )


def main_light_shelly_videv_sw(ts, testcase_id, room, single=False):
    ts.testCase(
        testcase_id, report.TCEL_SINGLE if single else report.TCEL_FULL, device_follow,
        room.main_light, room.main_light.KEY_OUTPUT_0,
        room.videv_main_light, room.videv_main_light.KEY_OUTPUT_0,
        STATES_SW
    )


def ffe(ts: testSession, mc: mqtt_client, h: house):
    # REQ 0001 - 0099
    room = h.ffe.livingroom
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0001', room)
    main_light_shelly_videv_sw(ts, 'REQ-0002', room)

    room = h.ffe.sleep
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0011', room)
    main_light_shelly_videv_sw(ts, 'REQ-0012', room)

    room = h.ffe.diningroom
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0021', room)
    main_light_shelly_videv_sw(ts, 'REQ-0022', room)

    room = h.ffe.kitchen
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0031', room)
    main_light_shelly_videv_sw(ts, 'REQ-0032', room)

    h.ffe.floor
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0041', room)
    main_light_shelly_videv_sw(ts, 'REQ-0042', room)


def ffw(ts: testSession, mc: mqtt_client, h: house):
    # REQ 0101 - 0199
    room = h.ffw.livingroom
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0101', room)
    main_light_shelly_videv_sw(ts, 'REQ-0102', room)

    room = h.ffw.sleep
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0111', room)
    main_light_shelly_videv_sw(ts, 'REQ-0112', room, True)

    room = h.ffw.julian
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0121', room)
    main_light_shelly_videv_sw(ts, 'REQ-0122', room)

    room = h.ffw.bath

    room = h.ffw.floor
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0141', room, True)
    main_light_shelly_videv_sw(ts, 'REQ-0142', room, True)


def gfe(ts: testSession, mc: mqtt_client, h: house):
    # REQ 0201 - 0299
    pass


def gfw(ts: testSession, mc: mqtt_client, h: house):
    # REQ 0301 - 0399
    room = h.gfw.dirk
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0301', room)
    main_light_shelly_videv_sw(ts, 'REQ-0302', room)

    room = h.gfw.marion
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0311', room)
    main_light_shelly_videv_sw(ts, 'REQ-0312', room)

    room = h.gfw.floor
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0321', room)
    main_light_shelly_videv_sw(ts, 'REQ-0322', room)


def stw(ts: testSession, mc: mqtt_client, h: house):
    # REQ 0401 - 0499
    room = h.stairway
    # main_light videv->shelly
    main_light_videv_shelly_sw(ts, 'REQ-0401', room)
    main_light_shelly_videv_sw(ts, 'REQ-0402', room)