def test_get_for_message_cls(self): assert Subscription.get_for_message_cls('mymsg').count() == 0 Subscription.create('aaa', 'mymsg', 'messenger') Subscription.create('bbb', 'mymsg', 'messenger2') assert Subscription.get_for_message_cls('mymsg').count() == 2
def test_get_for_user(self, user): r = Subscription.get_for_user(user) assert list(r) == [] assert Subscription.get_for_user(user).count() == 0 Subscription.create(user, 'message', 'messenger') assert Subscription.get_for_user(user).count() == 1
def test_prepare_undispatched(): m, d = Message.create('testplain', {MessageBase.SIMPLE_TEXT_ID: 'abc'}) Subscription.create('fred', 'testplain', 'test_messenger') Subscription.create('colon', 'testplain', 'test_messenger') dispatches = prepare_dispatches() assert len(dispatches) == 2 assert dispatches[0].address == 'fred' assert dispatches[1].address == 'colon'
def test_subscription(user_create): user1 = user_create(attributes=dict(username='******')) user2 = user_create(attributes=dict(username='******')) user2.is_active = False user2.save() Subscription.create(user1.id, MessagePlainForTest, MessengerForTest) Subscription.create(user2.id, MessagePlainForTest, MessengerForTest) assert len(MessagePlainForTest.get_subscribers(active_only=False)) == 2 assert len(MessagePlainForTest.get_subscribers(active_only=True)) == 1
def test_create(self, user): s = Subscription.create('abc', 'message', 'messenger') assert s.time_created is not None assert s.recipient is None assert s.address == 'abc' assert s.message_cls == 'message' assert s.messenger_cls == 'messenger' s = Subscription.create(user, 'message', 'messenger') assert s.time_created is not None assert s.address is None assert s.recipient_id == user.id assert s.message_cls == 'message' assert s.messenger_cls == 'messenger'
def test_replace_for_user(self, user): new_prefs = [('message3', 'messenger3')] assert Subscription.replace_for_user(user, new_prefs) Subscription.create(user, 'message', 'messenger') Subscription.create(user, 'message2', 'messenger2') assert Subscription.get_for_user(user).count() == 3 Subscription.replace_for_user(user, new_prefs) s = Subscription.get_for_user(user) assert s.count() == 1 s = s[0] assert s.message_cls == 'message3' assert s.messenger_cls == 'messenger3'
def test_cancel(self, user): Subscription.create('abc', 'message', 'messenger') Subscription.create('abc', 'message1', 'messenger') Subscription.create('abc', 'message', 'messenger1') assert Subscription.objects.filter(address='abc').count() == 3 Subscription.cancel('abc', 'message', 'messenger') assert Subscription.objects.filter(address='abc').count() == 2 Subscription.create(user, 'message', 'messenger') assert Subscription.objects.filter(recipient=user).count() == 1 Subscription.cancel(user, 'message', 'messenger') assert Subscription.objects.filter(recipient=user).count() == 0
def test_get_user_preferences_for_ui(template_render_tag, template_context, user): messengers_titles, prefs = get_user_preferences_for_ui(user) assert len(prefs.keys()) == 3 assert len(messengers_titles) == 8 from .testapp.sitemessages import MessageForTest, MessengerForTest Subscription.create(user, MessageForTest, MessengerForTest) user_prefs = get_user_preferences_for_ui( user, message_filter=lambda m: m.alias == 'test_message', messenger_filter=lambda m: m.alias in ['smtp', 'test_messenger'] ) messengers_titles, prefs = user_prefs assert len(prefs.keys()) == 1 assert len(messengers_titles) == 2 assert 'E-mail' in messengers_titles assert 'Test messenger' in messengers_titles html = template_render_tag( 'sitemessage', 'sitemessage_prefs_table from user_prefs', template_context({'user_prefs': user_prefs}) ) assert 'class="sitemessage_prefs' in html assert 'E-mail</th>' in html assert 'Test messenger</th>' in html assert 'value="test_message|smtp"' in html assert 'value="test_message|test_messenger" checked' in html prefs_row = prefs.popitem() assert prefs_row[0] == 'Test message type' assert ('test_message|smtp', True, False) in prefs_row[1] assert ('test_message|test_messenger', True, True) in prefs_row[1]