def test_destroy_source_and_provider_source_db_errort(self, mock_destory_provider, mock_destroy_source): """Test that destroys source where database error occurs.""" aws_source = Sources.objects.get(source_type="AWS") aws_provider = Provider.objects.get(type="AWS-local") # Associate models aws_source.source_uuid = aws_provider.uuid aws_source.koku_uuid = aws_provider.uuid aws_source.save() source_id = aws_source.source_id source_uuid = aws_source.source_uuid delete_source_and_provider(source_id, source_uuid, aws_source.auth_header) self.assertTrue(Sources.objects.filter(source_id=source_id).exists())
def test_destroy_source_and_provider_no_provider(self, mock_destory_provider): """Test that destroys source with no provider.""" aws_source = Sources.objects.get(source_type="AWS") aws_provider = Provider.objects.get(type="AWS-local") # Associate models aws_source.koku_uuid = aws_provider.uuid aws_source.save() source_id = aws_source.source_id source_uuid = uuid4() # Use incorrect UUID to simulate no provider. delete_source_and_provider(source_id, source_uuid, aws_source.auth_header) self.assertFalse(Sources.objects.filter(source_id=source_id).exists()) mock_destory_provider.assert_not_called()
def test_destroy_source_and_provider_source_does_not_exist(self, mock_destory_provider): """Test that destroys source where source does not exist.""" aws_source = Sources.objects.get(source_type="AWS") aws_provider = Provider.objects.get(type="AWS-local") # Associate models aws_source.source_uuid = aws_provider.uuid aws_source.koku_uuid = aws_provider.uuid aws_source.save() non_existent_source_id = 1 while Sources.objects.filter(source_id=non_existent_source_id).exists(): non_existent_source_id += 1 source_id = non_existent_source_id source_uuid = aws_source.source_uuid delete_source_and_provider(source_id, source_uuid, aws_source.auth_header) mock_destory_provider.assert_called()