fix: return simple uaid to hello

Closes #1185
mozilla-services · Apr 23, 2018 · ae2f3b4 · ae2f3b4
1 parent d46f48c
commit ae2f3b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/autopush/tests/test_integration.py b/autopush/tests/test_integration.py
@@ -122,6 +122,7 @@ def hello(self, uaid=None, services=None):
         result = json.loads(self.ws.recv())
         log.debug("Recv: %s", result)
         assert result["status"] == 200
+        assert "-" not in result["uaid"]
         if self.uaid and self.uaid != result["uaid"]:  # pragma: nocover
             log.debug("Mismatch on re-using uaid. Old: %s, New: %s",
                       self.uaid, result["uaid"])
@@ -1234,7 +1235,7 @@ def test_webpush_monthly_rotation(self):
         yield deferToThread(
             self.conn.db.message.table.delete_item,
             Key={'uaid': client.uaid, 'chidmessageid': ' '}
-         )
+        )
 
         # Verify the channel is gone
         exists, chans = yield deferToThread(

diff --git a/autopush_rs/src/client.rs b/autopush_rs/src/client.rs
@@ -408,7 +408,7 @@ where
             message_month,
             connected_at,
             stats: SessionStatistics {
-                uaid: uaid.hyphenated().to_string(),
+                uaid: uaid.simple().to_string(),
                 uaid_reset: reset_uaid,
                 existing_uaid: check_storage,
                 connection_type: String::from("webpush"),
@@ -420,7 +420,7 @@ where
         srv.connect_client(RegisteredClient { uaid, uid, tx });
 
         let response = ServerMessage::Hello {
-            uaid: uaid.hyphenated().to_string(),
+            uaid: uaid.simple().to_string(),
             status: 200,
             use_webpush: Some(true),
             broadcasts: Service::into_hashmap(broadcasts),