bug: Fix all_channels to match registration UAID format

* Fix tests to match how UAID is presented to functions. Closes #1330
mozilla-services · Apr 25, 2019 · bd21f06 · bd21f06
1 parent 64c64f7
commit bd21f06
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/autopush/db.py b/autopush/db.py
@@ -557,7 +557,7 @@ def all_channels(self, uaid):
         # update that list as well using register_channel()
         result = self.table.get_item(
             Key={
-                'uaid': hasher(uaid),
+                'uaid': hasher(uuid.UUID(uaid).hex),
                 'chidmessageid': ' ',
             },
             ConsistentRead=True

diff --git a/autopush/tests/test_db.py b/autopush/tests/test_db.py
@@ -249,7 +249,7 @@ def setUp(self):
             prefix="message_int_test",
             boto_resource=self.resource)
         self.real_table = table
-        self.uaid = str(uuid.uuid4())
+        self.uaid = uuid.uuid4().hex
 
     def test_non_rotating_tables(self):
         message_tablename = "message_int_test"