def test_converging(self):
"""Test it converges."""
desk = Desk({"back-of-desk": [[0, 4]], "monitor": [[9, 7], [6, 5]]})
desk.pixels = MagicMock()
converge = Converge(desk, [0, 255, 0])
converge.run()
desk.pixels.assert_has_calls(
[
call.__setitem__(0, [0, 255, 0]),
call.__setitem__(5, [0, 255, 0]),
call.show(),
call.__setitem__(1, [0, 255, 0]),
call.__setitem__(6, [0, 255, 0]),
call.show(),
call.__setitem__(2, [0, 255, 0]),
call.__setitem__(7, [0, 255, 0]),
call.show(),
call.__setitem__(3, [0, 255, 0]),
call.__setitem__(8, [0, 255, 0]),
call.show(),
call.__setitem__(4, [0, 255, 0]),
call.__setitem__(9, [0, 255, 0]),
call.show(),
]
)
def test_reverse_sweeping(self):
"""Test it sweeps backwards."""
desk = Desk({"back-of-desk": [[0, 4]], "monitor": [[9, 7], [6, 5]]})
desk.pixels = MagicMock()
sweep = Sweep(desk, [0, 255, 0], {"direction": "backwards"})
sweep.run()
desk.pixels.assert_has_calls([
call.__setitem__(5, [0, 255, 0]),
call.show(),
call.__setitem__(6, [0, 255, 0]),
call.show(),
call.__setitem__(7, [0, 255, 0]),
call.show(),
call.__setitem__(8, [0, 255, 0]),
call.show(),
call.__setitem__(9, [0, 255, 0]),
call.show(),
call.__setitem__(4, [0, 255, 0]),
call.show(),
call.__setitem__(3, [0, 255, 0]),
call.show(),
call.__setitem__(2, [0, 255, 0]),
call.show(),
call.__setitem__(1, [0, 255, 0]),
call.show(),
call.__setitem__(0, [0, 255, 0]),
call.show(),
])
def test__init__using_defaults__no_ssl(self):
connection_params_dict_mock = RLockedMagicMock()
connection_params_dict_mock.get.return_value = False
self.meth.__init__(connection_params_dict=connection_params_dict_mock,
exchange='my-exchange')
# attrs
self.assertIs(self.mock._connection_params_dict,
connection_params_dict_mock)
self.assertEqual(self.mock._exchange, {'exchange': 'my-exchange'})
self.assertEqual(self.mock._exchange_name, 'my-exchange')
self.assertEqual(self.mock._queues_to_declare, [])
self.assertIsNone(self.mock._serialize)
self.assertEqual(self.mock._prop_kwargs,
AMQPThreadedPusher.DEFAULT_PROP_KWARGS)
self.assertEqual(self.mock._mandatory, False)
self.assertIs(self.mock._output_fifo.__class__, queue.Queue)
self.assertEqual(self.mock._output_fifo.maxsize, 20000)
self.assertIsNone(self.mock._error_callback)
# calls
self.assertEqual(self.mock.mock_calls, [
call._setup_communication(),
call._start_publishing(),
])
self.assertEqual(
connection_params_dict_mock.mock_calls,
[
call.get('ssl'),
('__contains__', ('client_properties', ),
{}), # because cannot use `call.__contains__`
call.__setitem__('client_properties', ANY),
])
self.assertIsInstance(
# 2nd argument passed to __setitem__()
connection_params_dict_mock.__setitem__.mock_calls[0][-2][1],
dict)
def test_enable_token_env(self):
expected_token = 'secret'
json.load = MagicMock()
json.load.return_value = {'faulty_key': 'faulty_token'}
environment = MagicMock()
environment.get.return_value = expected_token
with patch('os.environ', environment):
enable_account(expected_token)
all_calls = environment.mock_calls
self.assertIn([call.__setitem__('QI_TOKEN', expected_token)],
all_calls)
token = load_account()
all_calls = environment.mock_calls
self.assertIn([call.get('QI_TOKEN', None)], all_calls)
self.assertEqual(expected_token, token)