Feature/token

cmr/queries.py

@@ -271,6 +271,19 @@ def mode(self, mode=CMR_OPS):
 
         self._base_url = str(mode) + self._route
 
+    def token(self, token):
+        """
+        Add token into url request.
+
+        :param token: Token from EDL.
+        :returns: Query instance
+        """
+
+        if not token:
+            return self
+
+        self.params['token'] = token
+        return self
 
 class GranuleCollectionBaseQuery(Query):
     """

tests/test_collection.py

@@ -82,3 +82,11 @@ def test_invalid_concept_id(self):
 
         with self.assertRaises(ValueError):
             query.concept_id(["C1299783579-LPDAAC_ECS", "G1327299284-LPDAAC_ECS"])
+
+    def test_token(self):
+        query = CollectionQuery()
+
+        query.token("123TOKEN")
+
+        self.assertIn("token", query.params)
+        self.assertEqual(query.params["token"], "123TOKEN")

tests/test_granule.py

@@ -428,3 +428,11 @@ def test_valid_concept_id(self):
 
         query.concept_id(["C1299783579-LPDAAC_ECS", "G1441380236-PODAAC"])
         self.assertEqual(query.params["concept_id"], ["C1299783579-LPDAAC_ECS", "G1441380236-PODAAC"])
+
+    def test_token(self):
+        query = GranuleQuery()
+
+        query.token("123TOKEN")
+
+        self.assertIn("token", query.params)
+        self.assertEqual(query.params["token"], "123TOKEN")

tests/test_service.py

@@ -69,3 +69,11 @@ def test_invalid_concept_id(self):
 
         with self.assertRaises(ValueError):
             query.concept_id(["C1299783579-LPDAAC_ECS", "G1327299284-LPDAAC_ECS"])
+
+    def test_token(self):
+        query = ServiceQuery()
+
+        query.token("123TOKEN")
+
+        self.assertIn("token", query.params)
+        self.assertEqual(query.params["token"], "123TOKEN")

tests/test_tool.py

@@ -69,3 +69,11 @@ def test_invalid_concept_id(self):
 
         with self.assertRaises(ValueError):
             query.concept_id(["C1299783579-LPDAAC_ECS", "G1327299284-LPDAAC_ECS"])
+
+    def test_token(self):
+        query = ToolQuery()
+
+        query.token("123TOKEN")
+
+        self.assertIn("token", query.params)
+        self.assertEqual(query.params["token"], "123TOKEN")

tests/test_variable.py

@@ -69,3 +69,11 @@ def test_invalid_concept_id(self):
 
         with self.assertRaises(ValueError):
             query.concept_id(["C1299783579-LPDAAC_ECS", "G1327299284-LPDAAC_ECS"])
+
+    def test_token(self):
+        query = VariableQuery()
+
+        query.token("123TOKEN")
+
+        self.assertIn("token", query.params)
+        self.assertEqual(query.params["token"], "123TOKEN")