Tests for environment module

2016-10-04 14:56:06 +02:00
parent 2a1276f4f1
commit 3844cfc226
2 changed files with 62 additions and 0 deletions
@@ -19,10 +19,25 @@ class ServerEnvironment(object):
    def __str__(self):
        return self._environment_str

+    @property
+    def is_development(self):
+        return self._environment_str == self.DEVELOPMENT
+
+    @property
+    def is_staging(self):
+        return self._environment_str == self.STAGING
+
+    @property
+    def is_production(self):
+        return self._environment_str == self.PRODUCTION
+
    @property
    def is_onpremise(self):
        return self._environment_str == self.ONPREMISE

+    def __eq__(self, other):
+        return self._environment_str == other._environment_str
+

 class ServerEnvironmentBuilder(object):