Grouper: Add option for test stem #42

UAL-RE · Sep 15, 2020 · 8e50cfd · 8e50cfd
1 parent a5c64b0
commit 8e50cfd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requiam/grouper_admin.py b/requiam/grouper_admin.py
@@ -33,7 +33,7 @@ def url(self, endpoint):
     def get_group_list(self, group_type):
         """Retrieve list of groups in a Grouper stem"""
 
-        if group_type not in ['portal', 'quota', '']:
+        if group_type not in ['portal', 'quota', 'test', '']:
             raise ValueError("Incorrect [group_type] input")
 
         grouper_stem = figshare_stem(group_type, production=self.grouper_production)
@@ -51,7 +51,7 @@ def get_group_list(self, group_type):
     def check_group_exists(self, group, group_type):
         """Check whether a Grouper group exists within a Grouper stem"""
 
-        if group_type not in ['portal', 'quota']:
+        if group_type not in ['portal', 'quota', 'test']:
             raise ValueError("Incorrect [group_type] input")
 
         result = self.get_group_list(group_type)
@@ -71,7 +71,7 @@ def add_group(self, group, group_type, description):
 
         endpoint = self.url("")
 
-        if group_type not in ['portal', 'quota']:
+        if group_type not in ['portal', 'quota', 'test']:
             raise ValueError("Incorrect [group_type] input")
 
         grouper_name = figshare_group(group, group_type,