Merge pull request #106 from reddit/update_DeciderContext

fix `T` TypeVar

Author: mrlevitas

reddit_decider/__init__.py

@@ -20,7 +20,6 @@
 from baseplate.lib.events import DebugLogger
 from baseplate.lib.events import EventLogger
 from baseplate.lib.file_watcher import FileWatcher
-from baseplate.lib.file_watcher import T
 from baseplate.lib.file_watcher import WatchedFileNotAvailableError
 from reddit_edgecontext import ValidatedAuthenticationToken
 from rust_decider import Decider as RustDecider
@@ -62,33 +61,43 @@ class ExperimentConfig:
 
 
 class DeciderContext:
-    """DeciderContext() is used to contain all fields necessary for
+    """:code:`DeciderContext` is used to contain all fields necessary for
     bucketing, targeting, and overrides.
-    :code:`DeciderContext()` is populated in :code:`make_object_for_context()`.
+    :code:`DeciderContext` is populated in :code:`make_object_for_context()`
+
+    :param user_id: user's t2 id
+    :param device_id: device installation uuid
+    :param country_code: 2-letter country codes
+    :param locale: ISO 639-1 primary language subtag and an optional ISO 3166-1 alpha-2 region subtag
+    :param user_is_employee:
+    :param logged_in: is user logged in
+    :param oauth_client_id: OAuth Client ID
+    :param origin_service: Service where request originated
+    :param cookie_created_timestamp: When the authentication cookie was created
+    :param loid_created_timestamp: Epoch milliseconds when the current LoID cookie was created
+    :param extracted_fields: Optional dict of additional fields, e.g. app_name & build_number
     """
 
-    T = TypeVar("T")
-
     def __init__(
         self,
         user_id: Optional[str] = None,
+        device_id: Optional[str] = None,
         country_code: Optional[str] = None,
         locale: Optional[str] = None,
         user_is_employee: Optional[bool] = None,
         logged_in: Optional[bool] = None,
-        device_id: Optional[str] = None,
         oauth_client_id: Optional[str] = None,
         origin_service: Optional[str] = None,
         cookie_created_timestamp: Optional[float] = None,
         loid_created_timestamp: Optional[float] = None,
         extracted_fields: Optional[dict] = None,
     ):
         self._user_id = user_id
+        self._device_id = device_id
         self._country_code = country_code
         self._locale = locale
         self._user_is_employee = user_is_employee
         self._logged_in = logged_in
-        self._device_id = device_id
         self._oauth_client_id = oauth_client_id
         self._origin_service = origin_service
         self._cookie_created_timestamp = cookie_created_timestamp
@@ -100,11 +109,11 @@ def to_dict(self) -> Dict:
 
         return {
             "user_id": self._user_id,
+            "device_id": self._device_id,
             "country_code": self._country_code,
             "locale": self._locale,
             "user_is_employee": self._user_is_employee,
             "logged_in": self._logged_in,
-            "device_id": self._device_id,
             "oauth_client_id": self._oauth_client_id,
             "origin_service": self._origin_service,
             "cookie_created_timestamp": self._cookie_created_timestamp,
@@ -175,6 +184,8 @@ class Decider:
     It will automatically reload the cache when changed.
     """
 
+    T = TypeVar("T")
+
     def __init__(
         self,
         decider_context: DeciderContext,

tests/decider_tests.py

@@ -3,14 +3,12 @@
 import logging
 import tempfile
 import unittest
-import warnings
 
 from unittest import mock
 
 from baseplate import RequestContext
 from baseplate import ServerSpan
 from baseplate.lib.events import DebugLogger
-from baseplate.lib.file_watcher import FileWatcher
 from reddit_edgecontext import ValidatedAuthenticationToken
 
 from reddit_decider import Decider
@@ -737,6 +735,17 @@ def test_get_variant_for_identifier_subreddit_id(self):
             # `identifier` passed to correct event field of experiment's `bucket_val` config
             self.assertEqual(event_fields["subreddit_id"], identifier)
 
+            # exposure assertions
+            self.assertEqual(self.event_logger.log.call_count, 1)
+            event_fields = self.event_logger.log.call_args[1]
+            self.assert_exposure_event_fields(
+                experiment_name="exp_1",
+                variant=variant,
+                event_fields=event_fields,
+                bucket_val=bucket_val,
+                identifier=identifier,
+            )
+
     def test_get_variant_for_identifier_ad_account_id(self):
         identifier = AD_ACCOUNT_ID
         bucket_val = "ad_account_id"
@@ -1409,7 +1418,7 @@ def test_get_all_variants_for_identifier_without_expose_bogus_identifier_type(se
 
             with self.assertLogs() as captured:
                 variant_arr = decider.get_all_variants_for_identifier_without_expose(
-                    identifier=identifier, identifier_type="blah"
+                    identifier=identifier, identifier_type=identifier_type
                 )
 
                 self.assertEqual(len(variant_arr), 0)
@@ -1497,7 +1506,7 @@ def test_get_variant_without_expose_with_HG_as_control_1_and_child_returns_none_
             self.assertEqual(self.event_logger.log.call_count, 0)
             variant = decider.get_variant_without_expose("exp_1")
 
-            assert variant == None
+            assert variant is None
 
             # exposure from control_1 of "hg"
             self.assertEqual(self.event_logger.log.call_count, 1)