Python get_user_model Examples

Example #1

File: test_api.py Project: gradel/django-stored-messages

    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")

Example #2

File: test_api.py Project: jnishiyama/django-stored-messages

    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")

Example #3

File: __init__.py Project: BadgerMaps/django-stored-messages

    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

Example #4

File: test_api.py Project: onionisi/django-stored-messages

    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")

Example #5

File: test_api.py Project: gradel/django-stored-messages

    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")

Example #6

    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")

Example #7

    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

Example #8

File: test_api.py Project: onionisi/django-stored-messages

    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")

Example #9

File: test_api.py Project: jnishiyama/django-stored-messages

    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")

Example #10

File: api.py Project: BadgerMaps/django-stored-messages

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)

Example #11

File: api.py Project: onionisi/django-stored-messages

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)

Example #12

    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")

Example #13

File: api.py Project: l1f7/django-stored-messages

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)

Example #14

File: api.py Project: onionisi/django-stored-messages

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)