From 1f0852bb925ec4109d5055ea7444130ec76aadc7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Honza=20Kr=C3=A1l?= Date: Sat, 3 Jan 2015 01:54:34 +0100 Subject: [PATCH] Tests for reindex accepting a query --- test_elasticsearch/test_server/test_helpers.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/test_elasticsearch/test_server/test_helpers.py b/test_elasticsearch/test_server/test_helpers.py index 2674f324..df0a3f70 100644 --- a/test_elasticsearch/test_server/test_helpers.py +++ b/test_elasticsearch/test_server/test_helpers.py @@ -165,13 +165,25 @@ class TestScan(ElasticsearchTestCase): self.assertEquals(set(range(100)), set(d['_source']['answer'] for d in docs)) class TestReindex(ElasticsearchTestCase): - def test_all_documents_get_moved(self): + def setUp(self): + super(TestReindex, self).setUp() bulk = [] for x in range(100): bulk.append({"index": {"_index": "test_index", "_type": "answers" if x % 2 == 0 else "questions", "_id": x}}) bulk.append({"answer": x, "correct": x == 42}) self.client.bulk(bulk, refresh=True) + def test_reindex_accepts_a_query(self): + helpers.reindex(self.client, "test_index", "prod_index", query={"query": {"filtered": {"filter": {"term": {"_type": "answers"}}}}}) + self.client.indices.refresh() + + self.assertTrue(self.client.indices.exists("prod_index")) + self.assertFalse(self.client.indices.exists_type(index='prod_index', doc_type='questions')) + self.assertEquals(50, self.client.count(index='prod_index', doc_type='answers')['count']) + + self.assertEquals({"answer": 42, "correct": True}, self.client.get(index="prod_index", doc_type="answers", id=42)['_source']) + + def test_all_documents_get_moved(self): helpers.reindex(self.client, "test_index", "prod_index") self.client.indices.refresh()