def test__get_image_search_match_from_index_result_no_post(self): with mock.patch.object(DuplicateImageService, '_get_post_from_index_id') as dup: dup.return_value = None dup_svc = DuplicateImageService(Mock(), Mock(), Mock(), config=MagicMock()) r = dup_svc._get_image_search_match_from_index_result( { 'id': 123, 'distance': .123 }, 'test.com', '40bec6703e3f3c2b0fc491a1c0c16cff273f00c00c020ff91b6807cc060c0014' ) self.assertIsNone(r)
def test__get_image_search_match_from_index_result_valid_post(self): with mock.patch.object(DuplicateImageService, '_get_post_from_index_id') as dup: dup.return_value = Post( id=456, dhash_h= '40bec6703e3f3c2b0fc491a1c0c16cff273f00c00c020ff91b6807cc060c0014' ) dup_svc = DuplicateImageService(Mock(), Mock(), Mock(), config=MagicMock()) r = dup_svc._get_image_search_match_from_index_result( { 'id': 123, 'distance': .123 }, 'test.com', '40bec6703e3f3c2b0fc491a1c0c16cff273f00c00c020ff91b6807cc060c0014' ) self.assertIsInstance(r, ImageSearchMatch) self.assertEqual(123, r.index_match_id) self.assertEqual(456, r.post.id)