class TestGateway: def setup(self): driver = DummyDriver() protocol = DummyProtocol() self.gw = Gateway(driver, protocol, "my gateway") def tearDown(self): self.gw = None def test_set_protocol_robustness(self): assert_raises(PyZigBeeBadArgument, self.gw.set_protocol, "invalid object") def test_set_driver_robustness(self): assert_raises(PyZigBeeBadArgument, self.gw.set_driver, "invalid object") def test_get_info(self): assert_equal("my gateway", self.gw.get_info()["description"]) assert_equal({'description': 'Dummy driver (reading from it what you previously wrote to it)'}, self.gw.get_info()["driver"]) assert_equal({'description': 'Dummy protocol'}, self.gw.get_info()["protocol"]) def test_get_firmware_version(self): def my_get_answer(*args, **kwargs): return "nothing" self.gw._get_answer = my_get_answer assert_equal("1.0.0", self.gw.get_firmware_version()) def test_get_hardware_version(self): def my_get_answer(*args, **kwargs): return "nothing" self.gw._get_answer = my_get_answer assert_equal("1.2.0", self.gw.get_hardware_version())
def setup(self): driver = DummyDriver() protocol = DummyProtocol() self.gw = Gateway(driver, protocol, "my gateway")