Added a simple reindex helper

2013-08-01 15:05:29 +02:00
parent 29977efff4
commit 1ee42ca17c
2 changed files with 27 additions and 0 deletions
@@ -24,3 +24,20 @@ class TestScan(ElasticTestCase):
        self.assertEquals(100, len(docs))
        self.assertEquals(set(map(str, range(100))), set(d['_id'] for d in docs))
        self.assertEquals(set(range(100)), set(d['_source']['answer'] for d in docs))
+
+class TestReindex(ElasticTestCase):
+    def test_all_documents_get_moved(self):
+        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)
+
+        helpers.reindex(self.client, "test_index", "prod_index")
+        self.client.indices.refresh()
+
+        self.assertTrue(self.client.indices.exists("prod_index"))
+        self.assertTrue(self.client.indices.exists_type("prod_index", "answers"))
+        self.assertTrue(self.client.indices.exists_type("prod_index", "questions"))
+        self.assertEquals(100, self.client.count(index='prod_index')['count'])
+