added more unit tests

chimpler · Apr 12, 2015 · 4e8c162 · 4e8c162
1 parent a767df2
commit 4e8c162
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/samples/aws.conf b/samples/aws.conf
@@ -0,0 +1,9 @@
+{
+    // dict merge
+	data-center-generic = { cluster-size = 6 }
+	data-center-east = ${data-center-generic} { name = "east" }
+
+	// list merge
+	default-jvm-opts = [-XX:+UseParNewGC]
+	large-jvm-opts = ${default-jvm-opts} [-Xm16g]
+}
diff --git a/tests/test_config_parser.py b/tests/test_config_parser.py
@@ -280,6 +280,19 @@ def test_dict_substitutions(self):
         assert config3.get('data-center-east.name') == 'east'
         assert config3.get('data-center-east.opts') == '-Xmx4g'
 
+        config4 = ConfigFactory.parse_string(
+            """
+                data-center-generic = { cluster-size = 6 }
+                data-center-east = {name = "east"} ${data-center-generic}
+                data-center-east-prod = ${data-center-east} {tmpDir=/tmp}
+            """
+        )
+
+        assert config4.get('data-center-east.cluster-size') == 6
+        assert config4.get('data-center-east.name') == 'east'
+        assert config4.get('data-center-east-prod.cluster-size') == 6
+        assert config4.get('data-center-east-prod.tmpDir') == '/tmp'
+
     def test_list_substitutions(self):
         config = ConfigFactory.parse_string(
             """
@@ -306,8 +319,21 @@ def test_list_substitutions(self):
             """
         )
 
+        assert config3.get('common_modules') == ['php', 'python']
         assert config3.get('host_modules') == ['java', 'php', 'python', 'perl']
 
+        config4 = ConfigFactory.parse_string(
+            """
+                common_modules = [php, python]
+                host_modules = [java] ${common_modules} [perl]
+                full_modules = ${host_modules} [c, go]
+            """
+        )
+
+        assert config4.get('common_modules') == ['php', 'python']
+        assert config4.get('host_modules') == ['java', 'php', 'python', 'perl']
+        assert config4.get('full_modules') == ['java', 'php', 'python', 'perl', 'c', 'go']
+
     def test_include_dict(self):
         config = ConfigFactory.parse_file("samples/animals.conf")
         assert config.get('cat.garfield.say') == 'meow'