def test_from_response(self):
fakes = [FakeFactory() for _ in range(10)]
response = es_factory()
response['facets'] = {
'test': {
'_type':
'terms',
'total':
229,
'other':
33,
'missing':
2,
'terms': [{
'term': str(f.id),
'count': faker.random_number(2)
} for f in fakes],
}
}
extracted = self.facet.from_response('test', response)
self.assertEqual(extracted['type'], 'models')
self.assertEqual(len(extracted['models']), 10)
for fake, row in zip(fakes, extracted['models']):
self.assertIsInstance(row[0], Fake)
self.assertIsInstance(row[1], int)
self.assertEqual(fake.id, row[0].id)
def hit_factory():
return {
"_score": float(faker.random_number(2)),
"_type": "fake",
"_id": faker.md5(),
"_source": {
"title": faker.sentence(),
"tags": [faker.word() for _ in range(faker.random_digit())]
},
"_index": "udata-test"
}
def hit_factory():
return {
"_score": float(faker.random_number(2)),
"_type": "fake",
"_id": faker.md5(),
"_source": {
"title": faker.sentence(),
"tags": [faker.word() for _ in range(faker.random_digit())]
},
"_index": "udata-test"
}
def test_from_response(self):
response = es_factory()
response['facets'] = {
'test': {
'_type': 'terms',
'total': 229,
'other': 33,
'missing': 2,
'terms': [{'term': faker.word(), 'count': faker.random_number(2)} for _ in range(10)],
}
}
extracted = self.facet.from_response('test', response)
self.assertEqual(extracted['type'], 'terms')
self.assertEqual(len(extracted['terms']), 10)
def test_from_response(self):
fakes = [FakeFactory() for _ in range(10)]
response = es_factory()
response['facets'] = {
'test': {
'_type': 'terms',
'total': 229,
'other': 33,
'missing': 2,
'terms': [{'term': str(f.id), 'count': faker.random_number(2)} for f in fakes],
}
}
extracted = self.facet.from_response('test', response)
self.assertEqual(extracted['type'], 'models')
self.assertEqual(len(extracted['models']), 10)
for fake, row in zip(fakes, extracted['models']):
self.assertIsInstance(row[0], Fake)
self.assertIsInstance(row[1], int)
self.assertEqual(fake.id, row[0].id)
def test_from_response(self):
response = es_factory()
response['facets'] = {
'test': {
'_type':
'terms',
'total':
229,
'other':
33,
'missing':
2,
'terms': [{
'term': faker.word(),
'count': faker.random_number(2)
} for _ in range(10)],
}
}
extracted = self.facet.from_response('test', response)
self.assertEqual(extracted['type'], 'terms')
self.assertEqual(len(extracted['terms']), 10)
