more tests for config stack

funbringer · funbringer · commit dd8c5000fae3 · 2018-02-20T23:51:06.000+03:00
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -23,7 +23,10 @@
     TimeoutException
 
 from testgres import \
-    scoped_config
+    TestgresConfig, \
+    configure_testgres, \
+    scoped_config, \
+    pop_config
 
 from testgres import \
     NodeStatus, \
@@ -529,6 +532,29 @@ def test_pg_config(self):
             b = get_pg_config()
             self.assertNotEqual(id(a), id(b))
 
+    def test_config_stack(self):
+        # no such option
+        with self.assertRaises(TypeError):
+            configure_testgres(dummy=True)
+
+        # we have only 1 config in stack
+        with self.assertRaises(IndexError):
+            pop_config()
+
+        d0 = TestgresConfig.cached_initdb_dir
+        d1 = 'dummy_abc'
+        d2 = 'dummy_def'
+
+        with scoped_config(cached_initdb_dir=d1) as c1:
+            self.assertEqual(c1.cached_initdb_dir, d1)
+
+            with scoped_config(cached_initdb_dir=d2) as c2:
+                self.assertEqual(c2.cached_initdb_dir, d2)
+
+            self.assertEqual(c1.cached_initdb_dir, d1)
+
+        self.assertEqual(TestgresConfig.cached_initdb_dir, d0)
+
     def test_unix_sockets(self):
         with get_new_node() as node:
             node.init(unix_sockets=False, allow_streaming=True)
