from unittest import TestCase
from unittest.mock import Mock
from owrx.property import PropertyCarousel, PropertyLayer, PropertyDeleted, PropertyWriteError


class PropertyCarouselTest(TestCase):
    def testInitiallyEmpty(self):
        pc = PropertyCarousel()
        with self.assertRaises(KeyError):
            x = pc["testkey"]

    def testPropertyAccess(self):
        pc = PropertyCarousel()
        pl = PropertyLayer(testkey="testvalue")
        pc.addLayer("test", pl)
        pc.switch("test")
        self.assertEqual(pc["testkey"], "testvalue")

    def testWriteAccess(self):
        pc = PropertyCarousel()
        pl = PropertyLayer(testkey="old_value")
        pc.addLayer("test", pl)
        pc.switch("test")
        pc["testkey"] = "new_value"
        self.assertEqual(pc["testkey"], "new_value")
        self.assertEqual(pl["testkey"], "new_value")

    def testForwardsEvents(self):
        pc = PropertyCarousel()
        pl = PropertyLayer(testkey="old_value")
        pc.addLayer("test", pl)
        pc.switch("test")
        mock = Mock()
        pc.wire(mock.method)
        pc["testkey"] = "new_value"
        mock.method.assert_called_once_with({"testkey": "new_value"})

    def testStopsForwardingAfterSwitch(self):
        pc = PropertyCarousel()
        pl_x = PropertyLayer(testkey="old_value")
        pc.addLayer("x", pl_x)
        pl_y = PropertyLayer(testkey="new_value")
        pc.addLayer("y", pl_y)
        pc.switch("x")
        pc.switch("y")
        mock = Mock()
        pc.wire(mock.method)
        pl_x["testkey"] = "new_value"
        mock.method.assert_not_called()

    def testEventsOnSwitch(self):
        pc = PropertyCarousel()
        pl_x = PropertyLayer(old_key="old_value")
        pc.addLayer("x", pl_x)
        pl_y = PropertyLayer(new_key="new_value")
        pc.addLayer("y", pl_y)
        pc.switch("x")
        mock = Mock()
        pc.wire(mock.method)
        pc.switch("y")
        mock.method.assert_called_once_with({"old_key": PropertyDeleted, "new_key": "new_value"})

    def testNoEventsIfKeysDontChange(self):
        pc = PropertyCarousel()
        pl_x = PropertyLayer(testkey="same_value")
        pc.addLayer("x", pl_x)
        pl_y = PropertyLayer(testkey="same_value")
        pc.addLayer("y", pl_y)
        pc.switch("x")
        mock = Mock()
        pc.wire(mock.method)
        pc.switch("y")
        mock.method.assert_not_called()

    def testKeyErrorOnInvalidSwitch(self):
        pc = PropertyCarousel()
        with self.assertRaises(KeyError):
            pc.switch("doesntmatter")

    def testHasLayer(self):
        pc = PropertyCarousel()
        pc.addLayer("testkey", PropertyLayer())
        self.assertTrue(pc.hasLayer("testkey"))
        self.assertFalse(pc.hasLayer("otherkey"))

    def testRemoveLayer(self):
        pc = PropertyCarousel()
        pl = PropertyLayer(testkey="testvalue")
        pc.addLayer("x", pl)
        pc.switch("x")
        self.assertEqual(pc["testkey"], "testvalue")
        pc.removeLayer("x")
        with self.assertRaises(KeyError):
            pc.switch("x")

    def testPropertyResetAfterRemoval(self):
        pc = PropertyCarousel()
        pl = PropertyLayer(testkey="testvalue")
        pc.addLayer("x", pl)
        pc.switch("x")
        self.assertEqual(pc["testkey"], "testvalue")
        pc.removeLayer("x")
        with self.assertRaises(KeyError):
            x = pc["testkey"]

    def testEmptySwitch(self):
        pc = PropertyCarousel()
        pl = PropertyLayer(testkey="testvalue")
        pc.addLayer("x", pl)
        pc.switch("x")
        self.assertEqual(pc["testkey"], "testvalue")
        pc.switch()
        with self.assertRaises(KeyError):
            x = pc["testkey"]

    def testErrorOnWriteOnDefaultLayer(self):
        pc = PropertyCarousel()
        with self.assertRaises(PropertyWriteError):
            pc["testkey"] = "testvalue"