summaryrefslogtreecommitdiff
path: root/openid/test/utils.py
blob: 02199e18c4a3c2c988ccf6fc442d3633a972485e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
"""Test utilities."""
from __future__ import unicode_literals

from openid import message


class OpenIDTestMixin(object):
    """Mixin providing custom asserts."""

    def assertOpenIDValueEqual(self, msg, key, expected, ns=None):
        """Check OpenID message contains key with expected value."""
        if ns is None:
            ns = message.OPENID_NS

        actual = msg.getArg(ns, key)
        error_format = 'Wrong value for openid.%s: expected=%s, actual=%s'
        error_message = error_format % (key, expected, actual)
        self.assertEqual(actual, expected, error_message)

    def assertOpenIDKeyMissing(self, msg, key, ns=None):
        if ns is None:
            ns = message.OPENID_NS

        error_message = 'openid.%s unexpectedly present' % key
        self.assertFalse(msg.hasKey(ns, key), error_message)