1 files changed, 38 insertions, 0 deletions
diff --git a/tests/test_prepare_build_config.py b/tests/test_prepare_build_config.py
new file mode 100644
index 0000000..e8692c4
--- /dev/null
+++ b/tests/test_prepare_build_config.py
@@ -0,0 +1,38 @@
+import sys
+import os
+sys.path.append(os.path.dirname(__file__) + "/../node")
+import prepare_build_config
+
+
+def test_validate_config():
+    os.environ["JOB_NAME"] = "user_job"
+    config = {}
+    prepare_build_config.validate_config(config, "")
+
+    os.environ["JOB_NAME"] = "linaro-android-private_job"
+    config = {}
+    try:
+        prepare_build_config.validate_config(config, "")
+        assert False, "Mismatch wasn't caught"
+    except prepare_build_config.BuildConfigMismatchException:
+        pass
+
+    os.environ["JOB_NAME"] = "linaro-android-private_job"
+    config = {"BUILD_TYPE": "build-android-restricted"}
+    try:
+        prepare_build_config.validate_config(config, "")
+        assert False, "Mismatch wasn't caught"
+    except prepare_build_config.BuildConfigMismatchException:
+        pass
+
+    os.environ["JOB_NAME"] = "linaro-android-private_job"
+    config = {"BUILD_TYPE": "build-android-restricted"}
+    prepare_build_config.validate_config(config, prepare_build_config.SLAVE_TYPE_RESTRICTED)
+
+    os.environ["JOB_NAME"] = "linaro-android-private_job"
+    config = {}
+    try:
+        prepare_build_config.validate_config(config, prepare_build_config.SLAVE_TYPE_RESTRICTED)
+        assert False, "Mismatch wasn't caught"
+    except prepare_build_config.BuildConfigMismatchException:
+        pass