Produits Entr'ouvert » Authentic 2

class UserProfileTests(TestCase):

    def setUp(self):

        from django.contrib.auth import get_user_model

        User = get_user_model()

        user = User.objects.create(username='testbot')

        user.set_password('secret')

        user.save()

        self.client = Client()

    def test_edit_profile_attributes(self):

        models.Attribute.objects.create(

            label=u'custom',

            name='custom',

            required=True,

            user_visible=True,

            user_editable=True,

            kind='string')

        models.Attribute.objects.create(

            label=u'ID',

            name='national_number',

            user_editable=True,

            user_visible=True,

            kind='string')

        self.assertTrue(self.client.login(username='testbot', password='secret'))

        # get the edit page in order to check form's prefix

        response = self.client.get(reverse('profile_edit'))

        form = get_response_form(response)

        kwargs = {'custom': 'random data',

                  'national_number': 'xx20153566342yy'}

        if form.prefix:

            kwargs = dict(('%s-%s' % (form.prefix, k), v)

                          for k, v in kwargs.iteritems())

        response = self.client.post(reverse('profile_edit'), kwargs)

        self.assertEqual(response.status_code, 302)

        response = self.client.get(reverse('account_management'))

        self.assertContains(response, 'random data')

        self.assertContains(response, 'xx20153566342yy')

        response = self.client.get(reverse('profile_edit'))

        form = get_response_form(response)

        self.assertEqual(form['custom'].value(), 'random data')

        self.assertEqual(form['national_number'].value(), 'xx20153566342yy')