def test_tries_upload(self, monkeypatch): mock_vault = mock.Mock() mock_vault.name = 'ThreePoint' mock_vault.upload_archive.return_value = 3 mock_get_vault = mock.Mock() mock_get_vault.return_value = mock_vault mock_complete = mock.Mock() monkeypatch.setattr(backup, 'get_vault', mock_get_vault) monkeypatch.setattr(backup, '_push_archive_complete', mock_complete) backup._push_file_archive('Triangles', None, None, {}, {}) mock_vault.upload_archive.assert_called_once_with('Triangles', description='Triangles')
def test_upload_error(self, monkeypatch): mock_vault = mock.Mock() mock_vault.name = 'ThreePoint' mock_response = mock.Mock() mock_response.status = 400 mock_response.read.return_value = json.dumps({ 'status': 400, 'message': 'Jean Valjean means nothing now', }).encode('utf-8') error = UnexpectedHTTPResponseError(200, mock_response) mock_vault.upload_archive.side_effect = error mock_get_vault = mock.Mock() mock_get_vault.return_value = mock_vault mock_complete = mock.Mock() monkeypatch.setattr(backup, 'get_vault', mock_get_vault) monkeypatch.setattr(backup, '_push_archive_complete', mock_complete) with pytest.raises(UnexpectedHTTPResponseError): backup._push_file_archive('Triangles', None, None, {}, {}) assert not mock_complete.called
def test_calls_complete(self, monkeypatch, credentials, settings): mock_vault = mock.Mock() mock_complete = mock.Mock() mock_vault.name = 'ThreePoint' mock_vault.upload_archive.return_value = 3 mock_get_vault = mock.Mock() mock_get_vault.return_value = mock_vault monkeypatch.setattr(backup, 'get_vault', mock_get_vault) monkeypatch.setattr(backup, '_push_archive_complete', mock_complete) backup._push_file_archive('Triangles', 0, None, credentials, settings) mock_complete.delay.assert_called_once_with( 0, None, { 'vault': 'ThreePoint', 'archive': 3 }, )
def test_upload_error_empty_file(self, monkeypatch): mock_vault = mock.Mock() mock_vault.name = 'ThreePoint' mock_response = mock.Mock() mock_response.status = 400 mock_response.read.return_value = json.dumps({ 'status': 400, 'message': 'Invalid Content-Length: 0', }).encode('utf-8') error = UnexpectedHTTPResponseError(200, mock_response) mock_vault.upload_archive.side_effect = error mock_get_vault = mock.Mock() mock_get_vault.return_value = mock_vault mock_complete = mock.Mock() monkeypatch.setattr(backup, 'get_vault', mock_get_vault) monkeypatch.setattr(backup, '_push_archive_complete', mock_complete) backup._push_file_archive('Triangles', None, None, {}, {}) mock_vault.upload_archive.assert_called_once_with('Triangles', description='Triangles') assert not mock_complete.called
def test_upload_error_empty_file(self, monkeypatch): mock_vault = mock.Mock() mock_vault.name = 'ThreePoint' mock_response = mock.Mock() mock_response.status = 400 mock_response.read.return_value = json.dumps({ 'status': 400, 'message': 'Invalid Content-Length: 0', }).encode('utf-8') error = UnexpectedHTTPResponseError(200, mock_response) mock_vault.upload_archive.side_effect = error mock_get_vault = mock.Mock() mock_get_vault.return_value = mock_vault mock_complete = mock.Mock() monkeypatch.setattr(backup, 'get_vault', mock_get_vault) monkeypatch.setattr(backup, '_push_archive_complete', mock_complete) backup._push_file_archive('Triangles', None, None, {}, {}) mock_vault.upload_archive.assert_called_once_with( 'Triangles', description='Triangles') assert not mock_complete.called