def test_upload_loads_data_from_specified_directory(self, mock_exporter_class, mock_converter): mock_converter.return_value = {"foo": "bar"} cli._main(self.arguments) mock_converter.assert_called_once_with( {'my_account': {'id': '42', 'email': '*****@*****.**', 'owner': 'me'}})
def test_upload_loads_data_from_specified_directory( self, mock_exporter_class, mock_converter): mock_converter.return_value = {"foo": "bar"} cli._main(self.arguments) mock_converter.assert_called_once_with({ 'my_account': { 'id': '42', 'email': '*****@*****.**', 'owner': 'me' } })
def test_upload_calls_all_upload_tasks(self, mock_exporter_class, mock_converter): """Mock away S3 Uploader, see if all necessary methods were called""" mock_converter.return_value = {"foo": "bar"} mock_exporter_instance = Mock() mock_exporter_class.return_value = mock_exporter_instance cli._main(self.arguments) mock_exporter_class.assert_called_once_with( "bucketname42", allowed_ips=["123", "345"], allowed_aws_account_ids=['42']) mock_exporter_instance.setup_infrastructure.assert_called_once_with() mock_exporter_instance.upload_to_S3.assert_called_once_with({'foo': 'bar'})
def test_upload_uses_S3Uploader_correctly(self, _, mock_upload, mock_converter): """Check if the 'necessary methods' used above actually exist on S3Uploader""" upload_data = {"foo": "bar"} mock_converter.return_value = upload_data cli._main(self.arguments) mock_upload.assert_called_once_with(upload_data)