"""Test for the Stream utils module.""" from notifications.utils import stream def test_get_user_feed_id_no_user_feed_id(): """Test get user feed id when no user feed id is specified.""" user_id = 'alw:123' user_feed_id = None result = stream.get_user_feed_id(user_id, user_feed_id) assert result == 'alw_123' def test_get_user_feed_id_with_user_feed_id(): """Test get user feed id when a user feed id is specified.""" user_id = 'alw:123' user_feed_id = 'john-doe@gmail.com' result = stream.get_user_feed_id(user_id, user_feed_id) assert result == 'johnDASHSYMBOLdoeATSYMBOLgmailDOTSYMBOLcom'