def test_handle_local( self, mock_gps=MagicMock(), mock_sense=MagicMock(), mock_web=MagicMock(), mock_trans=MagicMock(), ): mock_url = "localhost" with patch.dict(os.environ, {"DJANGO_SERVER_API_ENDPOINT": mock_url}): str_data = '{"key": "value"}' mock_trans.return_value.listen.return_value = str_data send_data_mock = MagicMock() mock_web.return_value.send = send_data_mock mock_web.return_value.send.side_effect = ErrorAfter(0) with self.assertRaises(CallableExhausted): main.handleLocal() mock_trans.assert_called_once() # assert init mock_web.assert_called_with(server_url=mock_url) # assert init self.assertEquals(1, send_data_mock.call_count) send_data_mock.assert_has_calls([call(json.loads(str_data))], any_order=True)
def test_handle_local_no_url( self, mock_print=MagicMock(), mock_gps=MagicMock(), mock_sense=MagicMock(), mock_web=MagicMock(), mock_trans=MagicMock(), ): with patch.dict(os.environ, {"DJANGO_SERVER_API_ENDPOINT": ""}): main.handleLocal() mock_trans.assert_called_once() mock_print.assert_has_calls( [call("DJANGO_SERVER_API_ENDPOINT not set")])
def test_handle_local_no_data( self, mock_gps=MagicMock(), mock_sense=MagicMock(), mock_web=MagicMock(), mock_trans=MagicMock(), ): mock_url = "localhost" with patch.dict(os.environ, {"DJANGO_SERVER_API_ENDPOINT": mock_url}): str_data = None mock_trans.return_value.listen.return_value = str_data mock_trans.return_value.listen.side_effect = ErrorAfter(1) send_data_mock = MagicMock() mock_web.return_value.send = send_data_mock with self.assertRaises(CallableExhausted): main.handleLocal() mock_trans.assert_called_once() # assert init mock_web.assert_called_with(server_url=mock_url) # assert init send_data_mock.assert_not_called()