def test_broadcast_message(self): user1 = get_user_model().objects.create_user("user1", "*****@*****.**", "123456") user2 = get_user_model().objects.create_user("user2", "*****@*****.**", "123456") user3 = get_user_model().objects.create_user("user3", "*****@*****.**", "123456") now = timezone.now() + timezone.timedelta(days=-1) url = 'http://example.com/error' broadcast_message( stored_messages.STORED_INFO, 'broadcast test message', 'extra', now, url) self.assertEqual(Inbox.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user3.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user1.id).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=user2.id).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=user3.id).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=user1.id).message.date, now) self.assertEqual(Inbox.objects.get(user=user2.id).message.date, now) self.assertEqual(Inbox.objects.get(user=user3.id).message.date, now) self.assertEqual(Inbox.objects.get(user=user1.id).message.url, url) self.assertEqual(Inbox.objects.get(user=user2.id).message.url, url) self.assertEqual(Inbox.objects.get(user=user3.id).message.url, url) self.assertEqual(MessageArchive.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user3.id).message.message, "broadcast test message")
def test_broadcast_message(self): user1 = get_user_model().objects.create_user("user1", "*****@*****.**", "123456") user2 = get_user_model().objects.create_user("user2", "*****@*****.**", "123456") user3 = get_user_model().objects.create_user("user3", "*****@*****.**", "123456") broadcast_message(stored_messages.STORED_INFO, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user3.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user3.id).message.message, "broadcast test message")
def setUp(self): self.factory = RequestFactory() self.user = get_user_model().objects.create_user("test_user", "*****@*****.**", "123456") self.request = RequestFactory().get('/') self.request.session = mock.MagicMock() self.request.user = self.user
def test_add_message_for(self): now = timezone.now() + timezone.timedelta(days=-1) url = 'http://example.com/error' user2 = get_user_model().objects.create_user("another_user", "*****@*****.**", "123456") add_message_for([user2, self.user], stored_messages.STORED_ERROR, 'Multiple errors', 'extra', now, url) self.assertEqual(Inbox.objects.count(), 2) self.assertEqual(MessageArchive.objects.count(), 2) self.assertEqual( Inbox.objects.get(user=user2.id).message.tags, 'extra') self.assertEqual( Inbox.objects.get(user=self.user).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=user2.id).message.date, now) self.assertEqual(Inbox.objects.get(user=self.user).message.date, now) self.assertEqual(Inbox.objects.get(user=user2.id).message.url, url) self.assertEqual(Inbox.objects.get(user=self.user).message.url, url) self.assertEqual( Inbox.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual( Inbox.objects.get(user=self.user).message.message, "Multiple errors") self.assertEqual( MessageArchive.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual( MessageArchive.objects.get(user=self.user).message.message, "Multiple errors")
def test_add_message_for(self): now = timezone.now() + timezone.timedelta(days=-1) url = 'http://example.com/error' user2 = get_user_model().objects.create_user("another_user", "*****@*****.**", "123456") add_message_for([user2, self.user], stored_messages.STORED_ERROR, 'Multiple errors', 'extra', now, url) self.assertEqual(Inbox.objects.count(), 2) self.assertEqual(MessageArchive.objects.count(), 2) self.assertEqual(Inbox.objects.get(user=user2.id).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=self.user).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=user2.id).message.date, now) self.assertEqual(Inbox.objects.get(user=self.user).message.date, now) self.assertEqual(Inbox.objects.get(user=user2.id).message.url, url) self.assertEqual(Inbox.objects.get(user=self.user).message.url, url) self.assertEqual(Inbox.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual(Inbox.objects.get(user=self.user).message.message, "Multiple errors") self.assertEqual(MessageArchive.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual(MessageArchive.objects.get(user=self.user).message.message, "Multiple errors")
def test_broadcast_message(self): user1 = get_user_model().objects.create_user("user1", "*****@*****.**", "123456") user2 = get_user_model().objects.create_user("user2", "*****@*****.**", "123456") user3 = get_user_model().objects.create_user("user3", "*****@*****.**", "123456") broadcast_message( stored_messages.STORED_INFO, 'broadcast test message') self.assertEqual(Inbox.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual(Inbox.objects.get(user=user3.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual(MessageArchive.objects.get(user=user3.id).message.message, "broadcast test message")
def test_broadcast_message(self): user1 = get_user_model().objects.create_user("user1", "*****@*****.**", "123456") user2 = get_user_model().objects.create_user("user2", "*****@*****.**", "123456") user3 = get_user_model().objects.create_user("user3", "*****@*****.**", "123456") now = timezone.now() + timezone.timedelta(days=-1) url = 'http://example.com/error' broadcast_message(stored_messages.STORED_INFO, 'broadcast test message', 'extra', now, url) self.assertEqual( Inbox.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual( Inbox.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual( Inbox.objects.get(user=user3.id).message.message, "broadcast test message") self.assertEqual( Inbox.objects.get(user=user1.id).message.tags, 'extra') self.assertEqual( Inbox.objects.get(user=user2.id).message.tags, 'extra') self.assertEqual( Inbox.objects.get(user=user3.id).message.tags, 'extra') self.assertEqual(Inbox.objects.get(user=user1.id).message.date, now) self.assertEqual(Inbox.objects.get(user=user2.id).message.date, now) self.assertEqual(Inbox.objects.get(user=user3.id).message.date, now) self.assertEqual(Inbox.objects.get(user=user1.id).message.url, url) self.assertEqual(Inbox.objects.get(user=user2.id).message.url, url) self.assertEqual(Inbox.objects.get(user=user3.id).message.url, url) self.assertEqual( MessageArchive.objects.get(user=user1.id).message.message, "broadcast test message") self.assertEqual( MessageArchive.objects.get(user=user2.id).message.message, "broadcast test message") self.assertEqual( MessageArchive.objects.get(user=user3.id).message.message, "broadcast test message")
def test_add_message_for(self): user2 = get_user_model().objects.create_user("another_user", "*****@*****.**", "123456") add_message_for([user2, self.user], stored_messages.STORED_ERROR, "Multiple errors") self.assertEqual(Inbox.objects.count(), 2) self.assertEqual(MessageArchive.objects.count(), 2) self.assertEqual(Inbox.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual(Inbox.objects.get(user=self.user).message.message, "Multiple errors") self.assertEqual(MessageArchive.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual(MessageArchive.objects.get(user=self.user).message.message, "Multiple errors")
def broadcast_message(level, message_text, extra_tags='', fail_silently=False): """ Send a message to all users aka broadcast. :param level: message level :param message_text: the string containing the message :param extra_tags: like the Django api, a string containing extra tags for the message :param fail_silently: not used at the moment """ users = get_user_model().objects.all() add_message_for(users, level, message_text, extra_tags=extra_tags, fail_silently=fail_silently)
def broadcast_message(level, message_text, extra_tags="", date=None, url=None, fail_silently=False): """ Send a message to all users aka broadcast. :param level: message level :param message_text: the string containing the message :param extra_tags: like the Django api, a string containing extra tags for the message :param date: a date, different than the default timezone.now :param url: an optional url :param fail_silently: not used at the moment """ users = get_user_model().objects.all() add_message_for(users, level, message_text, extra_tags=extra_tags, date=date, url=url, fail_silently=fail_silently)
def test_add_message_for(self): user2 = get_user_model().objects.create_user("another_user", "*****@*****.**", "123456") add_message_for([user2, self.user], stored_messages.STORED_ERROR, 'Multiple errors') self.assertEqual(Inbox.objects.count(), 2) self.assertEqual(MessageArchive.objects.count(), 2) self.assertEqual(Inbox.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual(Inbox.objects.get(user=self.user).message.message, "Multiple errors") self.assertEqual(MessageArchive.objects.get(user=user2.id).message.message, "Multiple errors") self.assertEqual(MessageArchive.objects.get(user=self.user).message.message, "Multiple errors")
def broadcast_message(level, message_text, extra_tags='', date=None, url=None, fail_silently=False): """ Send a message to all users aka broadcast. :param level: message level :param message_text: the string containing the message :param extra_tags: like the Django api, a string containing extra tags for the message :param date: a date, different than the default timezone.now :param url: an optional url :param fail_silently: not used at the moment """ users = get_user_model().objects.all() add_message_for(users, level, message_text, extra_tags=extra_tags, date=date, url=url, fail_silently=fail_silently)