def test_remove_signed_file_not_exists(self, public_storage_mock):
extension = Extension(pk=42, slug='mocked_ext')
version = ExtensionVersion(extension=extension, pk=123)
public_storage_mock.exists.return_value = False
version.remove_signed_file()
eq_(public_storage_mock.exists.call_args[0][0],
version.signed_file_path)
eq_(public_storage_mock.delete.call_count, 0)
def test_remove_signed_file(self, mocked_public_storage):
extension = Extension(pk=42, slug='mocked_ext')
version = ExtensionVersion(extension=extension, pk=123)
mocked_public_storage.exists.return_value = True
version.remove_signed_file()
eq_(mocked_public_storage.exists.call_args[0][0],
version.signed_file_path)
eq_(mocked_public_storage.delete.call_args[0][0],
version.signed_file_path)
