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)