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_get_values(self):
buckets = [{
'key': faker.word(),
'doc_count': faker.random_number(2)
} for _ in range(10)]
result = self.factory(aggregations=bucket_agg_factory(buckets))
assert len(result) == 10
for row in result:
assert isinstance(row[0], basestring)
assert isinstance(row[1], int)
assert isinstance(row[2], bool)
def test_get_values(self):
fakes = [FakeWithStringIdFactory() for _ in range(10)]
buckets = [{
'key': str(f.id),
'doc_count': faker.random_number(2)
} for f in fakes]
result = self.factory(aggregations=bucket_agg_factory(buckets))
assert len(result) == 10
for fake, row in zip(fakes, result):
assert isinstance(row[0], FakeWithStringId)
assert isinstance(row[1], int)
assert isinstance(row[2], bool)
assert fake.id == row[0].id
def test_get_values(self):
fakes = [FakeFactory() for _ in range(10)]
buckets = [{
'key': str(f.id),
'doc_count': faker.random_number(2)
} for f in fakes]
result = self.factory(aggregations=bucket_agg_factory(buckets))
self.assertEqual(len(result), 10)
for fake, row in zip(fakes, result):
self.assertIsInstance(row[0], Fake)
self.assertIsInstance(row[1], int)
self.assertIsInstance(row[2], bool)
self.assertEqual(fake.id, row[0].id)
