Fixes #33: message formating

chat/consumers.py

@@ -1,5 +1,6 @@
 from typing import Set
 
+import bleach
 from channels.generic.websocket import AsyncJsonWebsocketConsumer
 from django.conf import settings
 
@@ -182,6 +183,6 @@ async def chat_message(self, event: dict):
                 "msg_type": settings.MSG_TYPE_MESSAGE,
                 "room": event["room_id"],
                 "username": event["username"],
-                "message": event["message"],
+                "message": bleach.linkify(event["message"]),
             },
         )

chat/tests/test_chat_consumer.py

@@ -59,6 +59,17 @@ async def test_chat_consumer() -> None:
                         'username': user.username,
                         'message': message}
 
+    # Testing message formatting
+    # First making urls into links
+    message: str = 'My website is gonevis.com'
+    expected_message: str = 'My website is <a href="http://gonevis.com" rel="nofollow">gonevis.com</a>'
+    await communicator.send_json_to({"command": "send", "room": room.id, 'message': message})
+    response = await communicator.receive_json_from()
+    assert response == {'msg_type': settings.MSG_TYPE_MESSAGE,
+                        'room': room.id,
+                        'username': user.username,
+                        'message': expected_message}
+
     # Getting list of users in the room
     await communicator.send_json_to({'command': 'room_users', 'room': room.id})
     response = await communicator.receive_json_from()

requirements.txt

@@ -10,3 +10,4 @@ whitenoise==3.3.1
 Twisted[tls,http2]==18.4.0
 psycopg2-binary==2.7.4
 Fabric3==1.14.post1
+bleach==2.1.3