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)