summaryrefslogtreecommitdiff
path: root/tests/unit/objects/test_environments.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/objects/test_environments.py')
-rw-r--r--tests/unit/objects/test_environments.py27
1 files changed, 26 insertions, 1 deletions
diff --git a/tests/unit/objects/test_environments.py b/tests/unit/objects/test_environments.py
index b49a1db..5501471 100644
--- a/tests/unit/objects/test_environments.py
+++ b/tests/unit/objects/test_environments.py
@@ -4,7 +4,7 @@ GitLab API: https://docs.gitlab.com/ce/api/environments.html
import pytest
import responses
-from gitlab.v4.objects import ProjectEnvironment
+from gitlab.v4.objects import ProjectEnvironment, ProjectProtectedEnvironment
@pytest.fixture
@@ -22,9 +22,34 @@ def resp_get_environment():
yield rsps
+@pytest.fixture
+def resp_get_protected_environment():
+ content = {
+ "name": "protected_environment_name",
+ "last_deployment": "my birthday",
+ }
+
+ with responses.RequestsMock() as rsps:
+ rsps.add(
+ method=responses.GET,
+ url="http://localhost/api/v4/projects/1/protected_environments/2",
+ json=content,
+ content_type="application/json",
+ status=200,
+ )
+ yield rsps
+
+
def test_project_environments(project, resp_get_environment):
environment = project.environments.get(1)
assert isinstance(environment, ProjectEnvironment)
assert environment.id == 1
assert environment.last_deployment == "sometime"
assert environment.name == "environment_name"
+
+
+def test_project_protected_environments(project, resp_get_protected_environment):
+ protected_environment = project.protected_environments.get(2)
+ assert isinstance(protected_environment, ProjectProtectedEnvironment)
+ assert protected_environment.last_deployment == "my birthday"
+ assert protected_environment.name == "protected_environment_name"