diff --git a/tests/endtoend/test_cosmosdb_functions.py b/tests/endtoend/test_cosmosdb_functions.py
index ede7b7642..4404e16a1 100644
--- a/tests/endtoend/test_cosmosdb_functions.py
+++ b/tests/endtoend/test_cosmosdb_functions.py
@@ -16,7 +16,8 @@ def get_script_dir(cls):
     def test_cosmosdb_trigger(self):
         time.sleep(5)
         data = str(round(time.time()))
-        doc = {'id': 'cosmosdb-trigger-test', 'data': data}
+        doc = {'id': 'cosmosdb-trigger-test',
+               'data': data}
         r = self.webhost.request('POST', 'put_document',
                                  data=json.dumps(doc))
         self.assertEqual(r.status_code, 200)
@@ -35,12 +36,13 @@ def test_cosmosdb_trigger(self):
                 response = r.json()
                 response.pop('_metadata', None)
 
-                self.assertEqual(doc["id"], response["id"])
-                self.assertEqual(doc["data"], response["data"])
-                self.assertIsNotNone(response["_etag"])
-                self.assertIsNotNone(response["_rid"])
-                self.assertIsNotNone(response["_self"])
-                self.assertIsNotNone(response["_ts"])
+                self.assertEqual(response['id'], doc['id'])
+                self.assertEqual(response['data'], doc['data'])
+                self.assertTrue('_etag' in response)
+                self.assertTrue('_lsn' in response)
+                self.assertTrue('_rid' in response)
+                self.assertTrue('_self' in response)
+                self.assertTrue('_ts' in response)
             except AssertionError:
                 if try_no == max_retries - 1:
                     raise
@@ -51,7 +53,8 @@ def test_cosmosdb_trigger(self):
     def test_cosmosdb_input(self):
         time.sleep(5)
         data = str(round(time.time()))
-        doc = {'id': 'cosmosdb-input-test', 'data': data}
+        doc = {'id': 'cosmosdb-input-test',
+               'data': data}
         r = self.webhost.request('POST', 'put_document',
                                  data=json.dumps(doc))
         self.assertEqual(r.status_code, 200)
@@ -69,12 +72,14 @@ def test_cosmosdb_input(self):
                 self.assertEqual(r.status_code, 200)
                 response = r.json()
 
-                self.assertEqual(doc["id"], response["id"])
-                self.assertEqual(doc["data"], response["data"])
-                self.assertIsNotNone(response["_etag"])
-                self.assertIsNotNone(response["_rid"])
-                self.assertIsNotNone(response["_self"])
-                self.assertIsNotNone(response["_ts"])
+                # _lsn is present for cosmosdb change feed only,
+                # ref https://aka.ms/cosmos-change-feed
+                self.assertEqual(response['id'], doc['id'])
+                self.assertEqual(response['data'], doc['data'])
+                self.assertTrue('_etag' in response)
+                self.assertTrue('_rid' in response)
+                self.assertTrue('_self' in response)
+                self.assertTrue('_ts' in response)
             except AssertionError:
                 if try_no == max_retries - 1:
                     raise