def test_disconnect(self):
fake_socket = Mock(socket.socket)
fake_socket.recv = Mock(return_value=bytearray())
fake_socket.close = Mock(return_value=None)
c = Cube('127.0.0.1', 62910)
c._socket = fake_socket
c.disconnect()
self.assertTrue(fake_socket.close.called)
self.assertIsNone(c._socket)
def test_disconnect(self):
fake_socket = Mock(socket.socket)
fake_socket.recv = Mock(return_value=bytearray())
fake_socket.close = Mock(return_value=None)
c = Cube('127.0.0.1', 62910)
c._socket = fake_socket
c.disconnect()
self.assertTrue(fake_socket.close.called)
self.assertIsNone(c._socket)
def update_room(settings, room_id, temperature):
logger.info('changing temperature of room %s to %s', room_id, temperature)
cube_wait1()
cube = Cube('evn-cube')
cube.connect()
rf_address = database.get_room_rf_address(room_id)
cube.set_mode_manual(room_id, rf_address, temperature)
cube.disconnect()
cube_wait2()
logger.info('changing temperature of room %s to %s done', room_id,
temperature)
def test_context_manager_connect_and_disconnect(self):
c = Cube((None, None))
c.connect = Mock()
c.disconnect = Mock()
with c:
self.assertTrue(c.connect.called)
self.assertTrue(c.disconnect.called)
def test_context_manager_connect_and_disconnect(self): c = Cube((None, None)) c.connect = Mock() c.disconnect = Mock() with c: self.assertTrue(c.connect.called) self.assertTrue(c.disconnect.called)
def test_context_manager_disconnect_on_exception(self):
try:
c = Cube((None, None))
c.connect = Mock()
c.disconnect = Mock()
with c as cube:
raise Exception("just a test")
except Exception as ex:
self.assertTrue(c.connect.called)
self.assertTrue(c.disconnect.called)
self.assertEqual(ex.args, ("just a test", ))
def test_context_manager_disconnect_on_exception(self):
try:
c = Cube((None, None))
c.connect = Mock()
c.disconnect = Mock()
with c as cube:
raise Exception("just a test")
except Exception as ex:
self.assertTrue(c.connect.called)
self.assertTrue(c.disconnect.called)
self.assertEqual(ex.args, ("just a test", ))
def update_cube(settings):
cube_wait1()
# TODO configurable hostname/ip address
cube = Cube('evn-cube')
cube.connect()
cube_rooms = cube.rooms
for cube_room in cube_rooms:
database.save_or_update_room(cube_room)
problems = []
for cube_device in cube.devices:
database.save_or_update_device(cube_device)
submit_temperature(settings, cube_device)
problems.extend(check_for_problems(cube_device, cube_rooms))
cube.disconnect()
cube_wait2()
return problems
def test_disconnect_not_connected(self):
c = Cube('127.0.0.1', 62910)
c.disconnect()
def test_disconnect_not_connected(self):
c = Cube('127.0.0.1', 62910)
c.disconnect()
