class TestMethodError(TestCase):
def setUp(self):
self.aclient = Mock()
self.aclient.get_access_token = Mock(return_value='atoken')
self.aclient.get_api_key = Mock(return_value='apikey')
self.aclient.session = Mock(return_value={'access_token': 'atoken'})
self.vrs = VisualRecognition(self.aclient)
def test_add_face_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.add_face('collection_id', 'image.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
def test_add_face_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.add_face('collection_id', 'image.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_uuid_jpeg(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [False, True]
img = MagicMock()
img.format.lower.return_value = 'jpeg'
pil_open.return_value = img
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('ef0dce93-c2ac-4da5-bb2c-82ca7c770ad8',
'test_1.jpeg')
assert util.COMBINATION_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_uuid_uri(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [False, False]
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('ef0dce93-c2ac-4da5-bb2c-82ca7c770ad8',
'https://test.co,/test.jpg')
assert util.COMBINATION_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_uuid_uuid(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [False, False]
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('ef0dce93-c2ac-4da5-bb2c-82ca7c770ad8',
'06ef969b-4d2f-49bf-8f79-afc3bc072def')
assert util.COMBINATION_ERROR == str(excinfo.value)
class TestMethodOK(TestCase):
def setUp(self):
self.aclient = Mock()
self.aclient.get_access_token = Mock(return_value='atoken')
self.aclient.get_api_key = Mock(return_value='apikey')
self.aclient.session = Mock(return_value={'access_token': 'atoken'})
self.vrs = VisualRecognition(self.aclient)
self.__create_face_collection_expected = {
'collection_id': '728bee35-fa67-473b-91bf-79f088f46179'
}
self.__list_face_collections_expected = {
'face_collections': [{
'collection_id':
'728bee35-fa67-473b-91bf-79f088f46179'
}]
}
self.__list_faces_expected = {
'faces': [{
'face_id': '728bee35-fa67-473b-91bf-79f088f46179'
}]
}
self.__add_face_expected = {
'face_id': '728bee35-fa67-473b-91bf-79f088f46179',
'location': {
'left': 1085,
'top': 244,
'right': 1307,
'bottom': 466
}
}
self.__compare_face_to_collection_expected = {
'source': {
'location': {
'left': 1085,
'top': 244,
'right': 1307,
'bottom': 466
}
},
'target': {
'collection_id':
'80bf2bdc-d3de-491e-9106-0635df0a0a18',
'faces': [{
'face_id': '80bf2bdc-d3de-491e-9106-0635df0a0a18',
'score': 0.787753701210022,
}]
}
}
def test_param_err(self):
with pytest.raises(TypeError):
self.vrs.detect_faces()
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_list_collection(self, req):
req.return_value.text = json.dumps(
self.__list_face_collections_expected)
req.return_value.status_code = 200
assert self.__list_face_collections_expected == self.vrs.list_collections(
)
headers = make_headers('application/json')
req.assert_called_once_with('GET',
ENDPOINT + '/face_collections',
headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_create_collection(self, req):
req.return_value.text = json.dumps(
self.__create_face_collection_expected)
req.return_value.status_code = 201
assert self.__create_face_collection_expected == self.vrs.create_collection(
)
headers = make_headers('application/json')
req.assert_called_once_with('POST',
ENDPOINT + '/face_collections',
headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_delete_collection(self, req):
req.return_value.text = ''
req.return_value.status_code = 204
assert '' == self.vrs.delete_collection(
'728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179'
req.assert_called_once_with('DELETE', ENDPOINT + uri, headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_list_faces(self, req):
req.return_value.text = json.dumps(self.__list_faces_expected)
req.return_value.status_code = 200
assert self.__list_faces_expected == self.vrs.list_faces(
'728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
req.assert_called_once_with('GET', ENDPOINT + uri, headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_remove_face(self, req):
req.return_value.text = ''
req.return_value.status_code = 204
assert '' == self.vrs.remove_face(
'728bee35-fa67-473b-91bf-79f088f46179',
'79a68ab3-8c42-4c79-bc09-3ac363cd9ab1')
headers = make_headers('application/json')
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces/79a68ab3-8c42-4c79-bc09-3ac363cd9ab1'
req.assert_called_once_with('DELETE', ENDPOINT + uri, headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_add_face_jpeg(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__add_face_expected)
req.return_value.status_code = 201
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['jpeg']
pil_open.return_value = img
assert self.__add_face_expected == self.vrs.add_face(
'test.jpg', '728bee35-fa67-473b-91bf-79f088f46179')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
payload = {'image': opn()}
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
files=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_add_face_png(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__add_face_expected)
req.return_value.status_code = 201
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['png']
pil_open.return_value = img
assert self.__add_face_expected == self.vrs.add_face(
'test.png', '728bee35-fa67-473b-91bf-79f088f46179')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
payload = {'image': opn()}
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
files=payload)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_add_face_uri(self, req):
req.return_value.text = json.dumps(self.__add_face_expected)
req.return_value.status_code = 200
assert self.__add_face_expected == self.vrs.add_face(
'http://test.com/test.jpg', '728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
payload = json.dumps({'image': 'http://test.com/test.jpg'})
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_uri_to_collection(self, req):
req.return_value.text = json.dumps(
self.__compare_face_to_collection_expected)
req.return_value.status_code = 200
assert self.__compare_face_to_collection_expected == self.vrs.compare_faces(
'http://test.com/test_1.jpg',
'728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
payload = json.dumps({'image': 'http://test.com/test_1.jpg'})
uri = '/compare_faces/728bee35-fa67-473b-91bf-79f088f46179'
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_image_to_collection(self, req, opn, isfile,
pil_open):
req.return_value.text = json.dumps(
self.__compare_face_to_collection_expected)
req.return_value.status_code = 200
isfile.side_effect = [True, False]
img = MagicMock()
img.format.lower.side_effect = ['jpeg']
pil_open.return_value = img
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
assert self.__compare_face_to_collection_expected == self.vrs.compare_faces(
'test_1.jpg', '728bee35-fa67-473b-91bf-79f088f46179')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
payload = {'image': opn()}
uri = '/compare_faces/728bee35-fa67-473b-91bf-79f088f46179'
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
files=payload)
