Merge pull request #55 from contentstack/feat/CS-42561_early_access

Added Early access feature

Author: sunil-lakshman

contentstack/stack.py

@@ -40,6 +40,7 @@ def __init__(self, api_key: str, delivery_token: str, environment: str,
                      total=5, backoff_factor=0, status_forcelist=[408, 429]),
                  live_preview=None,
                  branch=None,
+                 early_access =  None,
                  ):
         """
         # Class that wraps the credentials of the authenticated user. Think of
@@ -92,6 +93,7 @@ def __init__(self, api_key: str, delivery_token: str, environment: str,
         self.branch = branch
         self.retry_strategy = retry_strategy
         self.live_preview = live_preview
+        self.early_access = early_access
         self._validate_stack()
 
     def _validate_stack(self):
@@ -123,6 +125,9 @@ def _validate_stack(self):
             'access_token': self.delivery_token,
             'environment': self.environment
         }
+        if self.early_access is not None:
+            early_access_str = ', '.join(self.early_access)
+            self.headers['x-header-ea'] = early_access_str
 
         if self.branch is not None:
             self.headers['branch'] = self.branch
@@ -141,6 +146,13 @@ def get_api_key(self):
         :return: api_key of the stack
         """
         return self.api_key
+    
+    @property
+    def get_early_access(self):
+        """
+        :return: early access
+        """
+        return self.early_access
 
     @property
     def get_delivery_token(self):

tests/test_stack.py

@@ -15,6 +15,7 @@ class TestStack(unittest.TestCase):
 
     def setUp(self):
         self.stack = contentstack.Stack(API_KEY, DELIVERY_TOKEN, ENVIRONMENT, host=HOST)
+        self.early_access = ['taxonomy', 'teams']
 
     def test_01_stack_credentials(self):
         self.assertEqual(ENVIRONMENT, stack_instance.environment)
@@ -167,3 +168,13 @@ def test_check_region(self):
                                     host=config.HOST, region=ContentstackRegion.AZURE_NA)
         var = _stack.region.value
         self.assertEqual('azure-na', var)
+    
+    def test_22_check_early_access_headers(self):
+        stack = contentstack.Stack(
+            config.APIKEY, config.DELIVERYTOKEN, config.ENVIRONMENT, early_access=[])
+        self.assertEqual(True, 'x-header-ea' in stack.get_headers)
+
+    def test_23_get_early_access(self):
+        stack = contentstack.Stack(
+            config.APIKEY, config.DELIVERYTOKEN, config.ENVIRONMENT, early_access=["taxonomy", "teams"])
+        self.assertEqual(self.early_access, stack.get_early_access)