import array
import unittest

import pythoncom
import win32con
import win32gui
import win32gui_struct


class TestBase(unittest.TestCase):
    def assertDictEquals(self, d, **kw):
        checked = {}
        for n, v in kw.items():
            self.assertEqual(v, d[n], f"'{n}' doesn't match: {v!r} != {d[n]!r}")
            checked[n] = True
        checked_keys = list(checked.keys())
        passed_keys = list(kw.keys())
        checked_keys.sort()
        passed_keys.sort()
        self.assertEqual(checked_keys, passed_keys)


class TestMenuItemInfo(TestBase):
    def _testPackUnpack(self, text):
        vals = {
            "fType": win32con.MFT_MENUBARBREAK,
            "fState": win32con.MFS_CHECKED,
            "wID": 123,
            "hSubMenu": 1234,
            "hbmpChecked": 12345,
            "hbmpUnchecked": 123456,
            "dwItemData": 1234567,
            "text": text,
            "hbmpItem": 321,
        }
        mii, extras = win32gui_struct.PackMENUITEMINFO(**vals)
        (
            fType,
            fState,
            wID,
            hSubMenu,
            hbmpChecked,
            hbmpUnchecked,
            dwItemData,
            text,
            hbmpItem,
        ) = win32gui_struct.UnpackMENUITEMINFO(mii)
        self.assertDictEquals(
            vals,
            fType=fType,
            fState=fState,
            wID=wID,
            hSubMenu=hSubMenu,
            hbmpChecked=hbmpChecked,
            hbmpUnchecked=hbmpUnchecked,
            dwItemData=dwItemData,
            text=text,
            hbmpItem=hbmpItem,
        )

    def testPackUnpack(self):
        self._testPackUnpack("Hello")

    def testPackUnpackNone(self):
        self._testPackUnpack(None)

    def testEmptyMenuItemInfo(self):
        mii, extra = win32gui_struct.EmptyMENUITEMINFO()
        (
            fType,
            fState,
            wID,
            hSubMenu,
            hbmpChecked,
            hbmpUnchecked,
            dwItemData,
            text,
            hbmpItem,
        ) = win32gui_struct.UnpackMENUITEMINFO(mii)
        self.assertEqual(fType, 0)
        self.assertEqual(fState, 0)
        self.assertEqual(wID, 0)
        self.assertEqual(hSubMenu, 0)
        self.assertEqual(hbmpChecked, 0)
        self.assertEqual(hbmpUnchecked, 0)
        self.assertEqual(dwItemData, 0)
        self.assertEqual(hbmpItem, 0)
        # it's not clear if UnpackMENUITEMINFO() should ignore cch, instead
        # assuming it is a buffer size rather than 'current length' - but it
        # never has (and this gives us every \0 in the string), and actually
        # helps us test the unicode/str semantics.
        self.assertEqual(text, "\0" * len(text))


class TestMenuInfo(TestBase):
    def testPackUnpack(self):
        vals = {
            "dwStyle": 1,
            "cyMax": 2,
            "hbrBack": 3,
            "dwContextHelpID": 4,
            "dwMenuData": 5,
        }

        mi = win32gui_struct.PackMENUINFO(**vals)
        (
            dwStyle,
            cyMax,
            hbrBack,
            dwContextHelpID,
            dwMenuData,
        ) = win32gui_struct.UnpackMENUINFO(mi)

        self.assertDictEquals(
            vals,
            dwStyle=dwStyle,
            cyMax=cyMax,
            hbrBack=hbrBack,
            dwContextHelpID=dwContextHelpID,
            dwMenuData=dwMenuData,
        )

    def testEmptyMenuItemInfo(self):
        mi = win32gui_struct.EmptyMENUINFO()
        (
            dwStyle,
            cyMax,
            hbrBack,
            dwContextHelpID,
            dwMenuData,
        ) = win32gui_struct.UnpackMENUINFO(mi)
        self.assertEqual(dwStyle, 0)
        self.assertEqual(cyMax, 0)
        self.assertEqual(hbrBack, 0)
        self.assertEqual(dwContextHelpID, 0)
        self.assertEqual(dwMenuData, 0)


class TestTreeViewItem(TestBase):
    def _testPackUnpack(self, text):
        vals = {
            "hitem": 1,
            "state": 2,
            "stateMask": 3,
            "text": text,
            "image": 4,
            "selimage": 5,
            "citems": 6,
            "param": 7,
        }

        ti, extra = win32gui_struct.PackTVITEM(**vals)
        (
            hitem,
            state,
            stateMask,
            text,
            image,
            selimage,
            citems,
            param,
        ) = win32gui_struct.UnpackTVITEM(ti)

        self.assertDictEquals(
            vals,
            hitem=hitem,
            state=state,
            stateMask=stateMask,
            text=text,
            image=image,
            selimage=selimage,
            citems=citems,
            param=param,
        )

    def testPackUnpack(self):
        self._testPackUnpack("Hello")

    def testPackUnpackNone(self):
        self._testPackUnpack(None)

    def testEmpty(self):
        ti, extras = win32gui_struct.EmptyTVITEM(0)
        (
            hitem,
            state,
            stateMask,
            text,
            image,
            selimage,
            citems,
            param,
        ) = win32gui_struct.UnpackTVITEM(ti)
        self.assertEqual(hitem, 0)
        self.assertEqual(state, 0)
        self.assertEqual(stateMask, 0)
        self.assertEqual(text, "")
        self.assertEqual(image, 0)
        self.assertEqual(selimage, 0)
        self.assertEqual(citems, 0)
        self.assertEqual(param, 0)


class TestListViewItem(TestBase):
    def _testPackUnpack(self, text):
        vals = {
            "item": None,
            "subItem": None,
            "state": 1,
            "stateMask": 2,
            "text": text,
            "image": 3,
            "param": 4,
            "indent": 5,
        }

        ti, extra = win32gui_struct.PackLVITEM(**vals)
        (
            item,
            subItem,
            state,
            stateMask,
            text,
            image,
            param,
            indent,
        ) = win32gui_struct.UnpackLVITEM(ti)

        # patch expected values.
        vals["item"] = 0
        vals["subItem"] = 0
        self.assertDictEquals(
            vals,
            item=item,
            subItem=subItem,
            state=state,
            stateMask=stateMask,
            text=text,
            image=image,
            param=param,
            indent=indent,
        )

    def testPackUnpack(self):
        self._testPackUnpack("Hello")

    def testPackUnpackNone(self):
        self._testPackUnpack(None)

    def testEmpty(self):
        ti, extras = win32gui_struct.EmptyLVITEM(1, 2)
        (
            item,
            subItem,
            state,
            stateMask,
            text,
            image,
            param,
            indent,
        ) = win32gui_struct.UnpackLVITEM(ti)
        self.assertEqual(item, 1)
        self.assertEqual(subItem, 2)
        self.assertEqual(state, 0)
        self.assertEqual(stateMask, 0)
        self.assertEqual(text, "")
        self.assertEqual(image, 0)
        self.assertEqual(param, 0)
        self.assertEqual(indent, 0)


class TestLVColumn(TestBase):
    def _testPackUnpack(self, text):
        vals = {"fmt": 1, "cx": 2, "text": text, "subItem": 3, "image": 4, "order": 5}

        ti, extra = win32gui_struct.PackLVCOLUMN(**vals)
        fmt, cx, text, subItem, image, order = win32gui_struct.UnpackLVCOLUMN(ti)

        self.assertDictEquals(
            vals, fmt=fmt, cx=cx, text=text, subItem=subItem, image=image, order=order
        )

    def testPackUnpack(self):
        self._testPackUnpack("Hello")

    def testPackUnpackNone(self):
        self._testPackUnpack(None)

    def testEmpty(self):
        ti, extras = win32gui_struct.EmptyLVCOLUMN()
        fmt, cx, text, subItem, image, order = win32gui_struct.UnpackLVCOLUMN(ti)
        self.assertEqual(fmt, 0)
        self.assertEqual(cx, 0)
        self.assertEqual(text, "")
        self.assertEqual(subItem, 0)
        self.assertEqual(image, 0)
        self.assertEqual(order, 0)


class TestDEV_BROADCAST_HANDLE(TestBase):
    def testPackUnpack(self):
        s = win32gui_struct.PackDEV_BROADCAST_HANDLE(123)
        c = array.array("b", s)
        got = win32gui_struct.UnpackDEV_BROADCAST(c.buffer_info()[0])
        self.assertEqual(got.handle, 123)

    def testGUID(self):
        s = win32gui_struct.PackDEV_BROADCAST_HANDLE(123, guid=pythoncom.IID_IUnknown)
        c = array.array("b", s)
        got = win32gui_struct.UnpackDEV_BROADCAST(c.buffer_info()[0])
        self.assertEqual(got.handle, 123)
        self.assertEqual(got.eventguid, pythoncom.IID_IUnknown)


class TestDEV_BROADCAST_DEVICEINTERFACE(TestBase):
    def testPackUnpack(self):
        s = win32gui_struct.PackDEV_BROADCAST_DEVICEINTERFACE(
            pythoncom.IID_IUnknown, "hello"
        )
        c = array.array("b", s)
        got = win32gui_struct.UnpackDEV_BROADCAST(c.buffer_info()[0])
        self.assertEqual(got.classguid, pythoncom.IID_IUnknown)
        self.assertEqual(got.name, "hello")


class TestDEV_BROADCAST_VOLUME(TestBase):
    def testPackUnpack(self):
        s = win32gui_struct.PackDEV_BROADCAST_VOLUME(123, 456)
        c = array.array("b", s)
        got = win32gui_struct.UnpackDEV_BROADCAST(c.buffer_info()[0])
        self.assertEqual(got.unitmask, 123)
        self.assertEqual(got.flags, 456)


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