Test for reading store_versions from Riak manager config.

praekeltfoundation · Dec 3, 2014 · 27b73b2 · 27b73b2
1 parent 38caf11
commit 27b73b2
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/vumi/persist/tests/test_txriak_manager.py b/vumi/persist/tests/test_txriak_manager.py
@@ -87,6 +87,20 @@ def test_from_config_with_mapreduce_timeout(self):
                                            })
         self.assertEqual(manager.mapreduce_timeout, 1000)
 
+    def test_from_config_with_store_versions(self):
+        manager_cls = self.manager.__class__
+        manager = manager_cls.from_config({
+            'bucket_prefix': 'test.',
+            'store_versions': {
+                'foo.Foo': 3,
+                'bar.Bar': None,
+            },
+        })
+        self.assertEqual(manager.store_versions, {
+            'foo.Foo': 3,
+            'bar.Bar': None,
+        })
+
     def test_sub_manager(self):
         sub_manager = self.manager.sub_manager("foo.")
         self.assertEqual(sub_manager.client, self.manager.client)