Move event_timestamp into ChangeEventsAPISession

PagerDuty · Apr 22, 2021 · bd634df · bd634df
1 parent 4ae7cd3
commit bd634df
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pdpyras.py b/pdpyras.py
@@ -629,10 +629,6 @@ def user_agent(self):
             sys.version_info.minor
         )
 
-    @property
-    def event_timestamp(self):
-        return datetime.now(timezone.utc).isoformat()
-
 class EventsAPISession(PDSession):
 
     """
@@ -811,6 +807,10 @@ class ChangeEventsAPISession(PDSession):
     def auth_header(self):
         return {}
 
+    @property
+    def event_timestamp(self):
+        return datetime.now(timezone.utc).isoformat()
+
     def prepare_headers(self, method):
         """Add user agent and content type headers for Change Events API requests."""
         headers = deepcopy(self.headers)

diff --git a/test_pdpyras.py b/test_pdpyras.py
@@ -617,7 +617,7 @@ def test_truncated_token(self):
 
 
 class ChangeEventsSessionTest(SessionTest):
-    @patch('pdpyras.PDSession.event_timestamp', datetime.datetime(2020, 3, 25).replace(tzinfo=datetime.timezone.utc).isoformat())
+    @patch('pdpyras.ChangeEventsAPISession.event_timestamp', datetime.datetime(2020, 3, 25).replace(tzinfo=datetime.timezone.utc).isoformat())
     def test_submit_change_event(self):
         sess = pdpyras.ChangeEventsAPISession('routingkey')
         parent = MagicMock()
@@ -655,7 +655,7 @@ def test_submit_change_event(self):
                     'links': [{'href':'https://http.cat/502.jpg'}]
                 },
                 parent.request.call_args[1]['json'])
-    @patch('pdpyras.PDSession.event_timestamp', datetime.datetime(2020, 3, 25).replace(tzinfo=datetime.timezone.utc).isoformat())
+    @patch('pdpyras.ChangeEventsAPISession.event_timestamp', datetime.datetime(2020, 3, 25).replace(tzinfo=datetime.timezone.utc).isoformat())
     def test_submit_lite_change_event(self):
         sess = pdpyras.ChangeEventsAPISession('routingkey')
         parent = MagicMock()