def test_save_image_before_upload(self, _, api_upload_image, image_save): # Check if upload image is called after saving image to file. # If not, an exception will be raised. image_save.side_effect = ( lambda *_: setattr(self, 'called_upload', True)) api_upload_image.side_effect = ( lambda *_: raise_(Exception()) if not getattr(self, 'called_upload', False) else None) pdu = UpgradeFaceSamples() # Init message message = UpgradeFaceSamplesFactory() pdu.process_message(message)
def test_pdu_calls_betaface_api_with_correct_params(self, _, api_upload_image, image_save): pdu = UpgradeFaceSamples() # Init message message = UpgradeFaceSamplesFactory() # Validate message ok_(pdu.validate_message(message), "Message is valid") # Call tested methods pdu.process_message(message) image_save.assert_called_once_with(message['head_image']['image'], message['head_image']['width'], message['head_image']['height'], ANY) path = image_save.call_args_list[0][0][3] api_upload_image.assert_called_once_with(path, message['person_name'])