#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# input-remapper - GUI for device specific keyboard mappings
# Copyright (C) 2025 sezanzeb <b8x45ygc9@mozmail.com>
#
# This file is part of input-remapper.
#
# input-remapper is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# input-remapper is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with input-remapper.  If not, see <https://www.gnu.org/licenses/>.


import asyncio
import unittest

from evdev.ecodes import EV_KEY

from inputremapper.configs.keyboard_layout import keyboard_layout
from inputremapper.injection.macros.parse import Parser
from tests.lib.test_setup import test_setup
from tests.unit.test_macros.macro_test_base import MacroTestBase, DummyMapping


@test_setup
class TestToggle(MacroTestBase):
    async def test_toggle(self):
        # repeats key(a) as long as macro is toggled
        macro = Parser.parse("toggle(key(a))", self.context, DummyMapping)
        code_a = keyboard_layout.get("a")

        self.assertEqual(self.count_child_macros(macro), 1)
        self.assertEqual(self.count_tasks(macro), 2)

        # Start
        macro.press_trigger()
        macro.release_trigger()
        asyncio.ensure_future(macro.run(self.handler))

        await asyncio.sleep(0.1)
        count_1 = self.result.count((EV_KEY, code_a, 1))
        self.assertGreater(count_1, 2)

        # Keeps writing more
        await asyncio.sleep(0.1)
        count_2 = self.result.count((EV_KEY, code_a, 1))
        self.assertGreater(count_2, count_1)

        # Stop
        macro.press_trigger()
        macro.release_trigger()

        count_3 = self.result.count((EV_KEY, code_a, 1))
        await asyncio.sleep(0.1)
        count_4 = self.result.count((EV_KEY, code_a, 1))
        # ensure that the macro has stopped
        self.assertEqual(count_3, count_4)


if __name__ == "__main__":
    unittest.main()