def test_sign_and_move_file_error(self, remove_signed_file_mock, private_storage_mock, sign_app_mock): extension = Extension(uuid='12345678123456781234567812345678') version = ExtensionVersion(extension=extension, pk=123) sign_app_mock.side_effect = SigningError with self.assertRaises(SigningError): version.sign_and_move_file() eq_(remove_signed_file_mock.call_count, 1)
def test_sign_and_move_file(self, remove_signed_file_mock, private_storage_mock, sign_app_mock): extension = Extension(uuid='ab345678123456781234567812345678') version = ExtensionVersion(extension=extension, pk=123) version.sign_and_move_file() expected_args = ( private_storage_mock.open.return_value, version.signed_file_path, json.dumps({ 'id': 'ab345678123456781234567812345678', 'version': 123, }) ) eq_(sign_app_mock.call_args[0], expected_args) eq_(remove_signed_file_mock.call_count, 0)
def test_sign_and_move_file_no_version_pk(self, private_storage_mock, sign_app_mock): extension = Extension(uuid='12345678123456781234567812345678') version = ExtensionVersion(extension=extension) with self.assertRaises(SigningError): version.sign_and_move_file()