コード例 #1
0
ファイル: test_utils.py プロジェクト: zixtor/e-cidadania
    def init(self):
        """Performs set up for the tests.
        """
        self.client = Client(enforce_csrf_checks=False)
        self.username = '******'
        self.user_password = '******'
        self.admin_username = '******'
        self.admin_password = '******'
        self.foo_user = self.create_user('foo_user', 'foo_user_password')
        self.foo_admin = self.create_user('foo_admin', 'foo_admin_password')
        self.foo_mod = self.create_user('foo_mod', 'foo_mod_password')

        self.bar_user = self.create_user('bar_user', 'bar_user_password')
        self.bar_admin = self.create_user('bar_admin', 'bar_admin_password')
        self.bar_mod = self.create_user('bar_mod', 'bar_mod_password')

        space_properties = {
            'name': 'foo_space',
            'url': 'foo_space_url',
            'author': self.foo_admin,
            'public': False,
            'mod_debate': True,
            'mod_proposals': True,
            'mod_news': True,
            'mod_cal': True,
            'mod_docs': True,
        }
        self.foo_space = Space(**space_properties)
        self.foo_space.save()
        self.foo_space.admins.add(self.foo_admin)
        self.foo_space.mods.add(self.foo_mod)
        self.foo_space.users.add(self.foo_user)
        self.foo_space.save()

        space_properties.update({
            'author': self.bar_admin,
            'name': 'bar_space',
            'url': 'bar_space_url',
            'public': True,
        })
        self.bar_space = Space(**space_properties)
        self.bar_space.save()
        self.bar_space.admins.add(self.bar_admin)
        self.bar_space.mods.add(self.bar_mod)
        self.bar_space.users.add(self.bar_user)
        self.bar_space.save()

        debate_properties = {'space': self.foo_space, 'author': self.foo_admin}
        self.foo_debate = self.seed(Debate, properties=debate_properties)

        debate_properties.update({
            'space': self.bar_space,
            'author': self.bar_admin
        })
        self.bar_debate = self.seed(Debate, debate_properties)

        column_properties = {'debate': self.foo_debate, 'criteria': 'private'}
        self.foo_column = Column(**column_properties)
        self.foo_column.save()

        column_properties.update({
            'debate': self.bar_debate,
            'criteria': 'public'
        })
        self.bar_column = Column(**column_properties)
        self.bar_column.save()

        row_properties = column_properties.copy()
        self.bar_row = Row(**row_properties)
        self.bar_row.save()

        row_properties.update({'debate': self.foo_debate})
        self.foo_row = Row(**row_properties)
        self.foo_row.save()

        note_properties = {
            'column': self.foo_column,
            'row': self.foo_row,
            'debate': self.foo_debate,
            'author': self.foo_admin
        }
        self.foo_note = Note(**note_properties)
        self.foo_note.save()

        note_properties.update({
            'column': self.bar_column,
            'row': self.bar_row,
            'debate': self.bar_debate,
            'author': self.bar_admin
        })
        self.bar_note = Note(**note_properties)
        self.bar_note.save()

        post_properties = {
            'title': 'Foo news post',
            'author': self.foo_user,
            'pub_index': True,
            'space': self.foo_space
        }
        self.foo_post = Post(**post_properties)
        self.foo_post.save()

        post_properties.update({
            'title': 'Bar news post',
            'author': self.bar_user,
            'space': self.bar_space
        })
        self.bar_post = Post(**post_properties)
        self.bar_post.save()

        proposal_set_properties = {
            'name': 'Foo Proposal Set',
            'space': self.foo_space,
            'author': self.foo_admin,
            'debate': self.foo_debate
        }
        self.foo_proposalset = ProposalSet(**proposal_set_properties)
        self.foo_proposalset.save()

        proposal_set_properties.update({
            'name': 'Bar Proposal Set',
            'space': self.bar_space,
            'author': self.bar_admin,
            'debate': self.bar_debate
        })
        self.bar_proposalset = ProposalSet(**proposal_set_properties)
        self.bar_proposalset.save()
コード例 #2
0
ファイル: test_utils.py プロジェクト: MiguelPin/e-cidadania
 def init(self):
     """Performs set up for the tests.
     """
     self.client = Client(enforce_csrf_checks=False)
     self.username = '******'
     self.user_password = '******'
     self.admin_username = '******'
     self.admin_password = '******'
     self.foo_user = self.create_user('foo_user', 'foo_user_password')
     self.foo_admin = self.create_user('foo_admin', 'foo_admin_password')
     self.foo_mod = self.create_user('foo_mod', 'foo_mod_password')
     
     self.bar_user = self.create_user('bar_user', 'bar_user_password')
     self.bar_admin = self.create_user('bar_admin', 'bar_admin_password')
     self.bar_mod = self.create_user('bar_mod', 'bar_mod_password')
         
     space_properties = {'name': 'foo_space', 'url': 'foo_space_url',
                         'author': self.foo_admin, 'public': False, 
                         'mod_debate': True, 'mod_proposals': True,
                         'mod_news': True, 'mod_cal': True, 'mod_docs': True,
                         }
     self.foo_space = Space(**space_properties)
     self.foo_space.save()
     self.foo_space.admins.add(self.foo_admin)
     self.foo_space.mods.add(self.foo_mod)
     self.foo_space.users.add(self.foo_user)
     self.foo_space.save()
     
     space_properties.update({'author': self.bar_admin, 'name': 'bar_space',
                              'url': 'bar_space_url', 'public': True,})
     self.bar_space = Space(**space_properties)
     self.bar_space.save()
     self.bar_space.admins.add(self.bar_admin)
     self.bar_space.mods.add(self.bar_mod)
     self.bar_space.users.add(self.bar_user)
     self.bar_space.save()
     
     debate_properties = {'space': self.foo_space, 'author': self.foo_admin}
     self.foo_debate = self.seed(Debate,properties=debate_properties)
     
     debate_properties.update({'space': self.bar_space, 
                               'author': self.bar_admin})
     self.bar_debate = self.seed(Debate,debate_properties)
     
     column_properties = {'debate': self.foo_debate, 'criteria': 'private'}
     self.foo_column = Column(**column_properties)
     self.foo_column.save()
     
     column_properties.update({'debate': self.bar_debate,
                               'criteria': 'public'})
     self.bar_column = Column(**column_properties)
     self.bar_column.save()
     
     row_properties = column_properties.copy()
     self.bar_row = Row(**row_properties)
     self.bar_row.save()
     
     row_properties.update({'debate': self.foo_debate})
     self.foo_row = Row(**row_properties)
     self.foo_row.save()
     
     note_properties = {'column': self.foo_column, 'row': self.foo_row,
                        'debate': self.foo_debate, 'author': self.foo_admin}
     self.foo_note = Note(**note_properties)
     self.foo_note.save()
     
     note_properties.update({'column': self.bar_column, 
                             'row': self.bar_row,
                             'debate': self.bar_debate,
                             'author': self.bar_admin})
     self.bar_note = Note(**note_properties)
     self.bar_note.save()
     
     post_properties = {'title': 'Foo news post', 'author': self.foo_user,
                        'pub_index': True, 'space': self.foo_space}
     self.foo_post = Post(**post_properties)
     self.foo_post.save()
     
     post_properties.update({'title': 'Bar news post',
                             'author': self.bar_user, 'space': self.bar_space})
     self.bar_post = Post(**post_properties)
     self.bar_post.save()
     
     proposal_set_properties = {'name': 'Foo Proposal Set', 
                                'space': self.foo_space,
                                'author': self.foo_admin, 
                                'debate': self.foo_debate}
     self.foo_proposalset = ProposalSet(**proposal_set_properties)
     self.foo_proposalset.save()
     
     proposal_set_properties.update({'name': 'Bar Proposal Set',
                                     'space': self.bar_space,
                                     'author': self.bar_admin,
                                     'debate': self.bar_debate})
     self.bar_proposalset = ProposalSet(**proposal_set_properties)
     self.bar_proposalset.save()
コード例 #3
0
ファイル: test_utils.py プロジェクト: zixtor/e-cidadania
class ECDTestCase(TestCase):
    """Class which extends Django TestCase and adds methods specific to 
    e-cidadania for testing.
    """
    def init(self):
        """Performs set up for the tests.
        """
        self.client = Client(enforce_csrf_checks=False)
        self.username = '******'
        self.user_password = '******'
        self.admin_username = '******'
        self.admin_password = '******'
        self.foo_user = self.create_user('foo_user', 'foo_user_password')
        self.foo_admin = self.create_user('foo_admin', 'foo_admin_password')
        self.foo_mod = self.create_user('foo_mod', 'foo_mod_password')

        self.bar_user = self.create_user('bar_user', 'bar_user_password')
        self.bar_admin = self.create_user('bar_admin', 'bar_admin_password')
        self.bar_mod = self.create_user('bar_mod', 'bar_mod_password')

        space_properties = {
            'name': 'foo_space',
            'url': 'foo_space_url',
            'author': self.foo_admin,
            'public': False,
            'mod_debate': True,
            'mod_proposals': True,
            'mod_news': True,
            'mod_cal': True,
            'mod_docs': True,
        }
        self.foo_space = Space(**space_properties)
        self.foo_space.save()
        self.foo_space.admins.add(self.foo_admin)
        self.foo_space.mods.add(self.foo_mod)
        self.foo_space.users.add(self.foo_user)
        self.foo_space.save()

        space_properties.update({
            'author': self.bar_admin,
            'name': 'bar_space',
            'url': 'bar_space_url',
            'public': True,
        })
        self.bar_space = Space(**space_properties)
        self.bar_space.save()
        self.bar_space.admins.add(self.bar_admin)
        self.bar_space.mods.add(self.bar_mod)
        self.bar_space.users.add(self.bar_user)
        self.bar_space.save()

        debate_properties = {'space': self.foo_space, 'author': self.foo_admin}
        self.foo_debate = self.seed(Debate, properties=debate_properties)

        debate_properties.update({
            'space': self.bar_space,
            'author': self.bar_admin
        })
        self.bar_debate = self.seed(Debate, debate_properties)

        column_properties = {'debate': self.foo_debate, 'criteria': 'private'}
        self.foo_column = Column(**column_properties)
        self.foo_column.save()

        column_properties.update({
            'debate': self.bar_debate,
            'criteria': 'public'
        })
        self.bar_column = Column(**column_properties)
        self.bar_column.save()

        row_properties = column_properties.copy()
        self.bar_row = Row(**row_properties)
        self.bar_row.save()

        row_properties.update({'debate': self.foo_debate})
        self.foo_row = Row(**row_properties)
        self.foo_row.save()

        note_properties = {
            'column': self.foo_column,
            'row': self.foo_row,
            'debate': self.foo_debate,
            'author': self.foo_admin
        }
        self.foo_note = Note(**note_properties)
        self.foo_note.save()

        note_properties.update({
            'column': self.bar_column,
            'row': self.bar_row,
            'debate': self.bar_debate,
            'author': self.bar_admin
        })
        self.bar_note = Note(**note_properties)
        self.bar_note.save()

        post_properties = {
            'title': 'Foo news post',
            'author': self.foo_user,
            'pub_index': True,
            'space': self.foo_space
        }
        self.foo_post = Post(**post_properties)
        self.foo_post.save()

        post_properties.update({
            'title': 'Bar news post',
            'author': self.bar_user,
            'space': self.bar_space
        })
        self.bar_post = Post(**post_properties)
        self.bar_post.save()

        proposal_set_properties = {
            'name': 'Foo Proposal Set',
            'space': self.foo_space,
            'author': self.foo_admin,
            'debate': self.foo_debate
        }
        self.foo_proposalset = ProposalSet(**proposal_set_properties)
        self.foo_proposalset.save()

        proposal_set_properties.update({
            'name': 'Bar Proposal Set',
            'space': self.bar_space,
            'author': self.bar_admin,
            'debate': self.bar_debate
        })
        self.bar_proposalset = ProposalSet(**proposal_set_properties)
        self.bar_proposalset.save()

    def seed(self,
             model,
             properties=None,
             constraints=None,
             follow_fk=None,
             generate_fk=None,
             follow_m2m=None,
             factory=None,
             commit=True):
        """Generates and returns a new instance of the `model` with 
        properties in `properties`.
        """
        instance = seeder.seed(model=model,
                               constraints=constraints,
                               follow_fk=follow_fk,
                               generate_fk=None,
                               follow_m2m=None,
                               factory=None,
                               model_properties=properties,
                               commit=commit)

        return instance

    def seedn(self,
              count,
              model,
              properties,
              constraints=None,
              follow_fk=None,
              generate_fk=None,
              follow_m2m=None,
              factory=None,
              commit=True):
        """Generates and returns `count` number of instances of `model` with
        properties in `properties`.
        """

        obj_list = seeder.seedn(count=count,
                                model=model,
                                constraints=constraints,
                                follow_fk=follow_fk,
                                generate_fk=generate_fk,
                                follow_m2m=follow_m2m,
                                factory=factory,
                                model_properties=properties,
                                commit=True)
        return obj_list

    def create_user(self,
                    username,
                    password,
                    email=None,
                    properties=None,
                    logged_in=False):
        """Creates, saves and returns a user with a given username, password
        and email. If `properties` is supplied, it will be applied to the 
        created user.
        """

        user = User.objects.create_user(username=username,
                                        password=password,
                                        email=email)
        if properties:
            for key in properties:
                setattr(user, key, properties[key])
            user.save()

        if logged_in:
            # log out the current user
            self.logout()
            # log in the new user
            user = self.login(username=username,
                              password=password,
                              email=email)
        return user

    def create_super_user(self,
                          username='******',
                          password='******',
                          email='*****@*****.**',
                          properties=None,
                          logged_in=False):
        """Creates, saves and returns  a super user with a given username, 
        password and email. If `properties` is supplied, it will be applied
        to the created user.
        """

        super_user = User.objects.create_superuser(username=username,
                                                   password=password,
                                                   email=email)
        if properties:
            for key in properties:
                setattr(super_user, key, properties[key])
            super_user.save()

        if logged_in:
            self.logout()
            super_user = self.login(username=username,
                                    password=password,
                                    email=email)
        return super_user

    def login(self, username, password, email=None):
        """Logs in a user with the given username and password. If the user is 
        not present in the database, it will be created and logged in.
        
        We assume `username` is unique across the database.
        """

        try:
            user = User.objects.get(username=username)
        except Exception:
            user = None
        if user is None:
            user = self.create_user(username=username,
                                    password=password,
                                    email=email)

        self.client.login(username=username, password=password)

        return user

    def logout(self):
        """Logs out the currently logged in user.
        """

        self.client.logout()

    def isLoggedIn(self, user=None):
        """Checks and returns True if a user is logged in otherwise returns
        False.
        """

        if '_auth_user_id' not in self.client.session:
            return False

        if (user.pk == self.client.session['_auth_user_id']):
            return True

        return False

    def getURL(self, name, args=None, kwargs=None):
        """Returns the url for the given `name` which may be a function name or
        url name.
        """
        return reverse(name, args=args, kwargs=kwargs)

    def get(self, url=None, url_name=None, data={}, follow=False, **extra):
        """
        Performs a get to the given url and returns the response.
        """
        if url is None and url_name is None:
            raise Exception("Please pass either url or url name")

        if url_name:
            url = self.getURL(url_name)

        response = self.client.get(url, data=data, follow=follow, extra=extra)
        return response

    def post(self, url, data={}, follow=False, **extra):
        """
        Performs a post to the supplied url and returns the response.
        """

        response = self.client.post(path=url,
                                    data=data,
                                    follow=follow,
                                    extra=extra)
        return response

    def printResponse(self, response):
        """Prints the response to the terminal.
        We need this method because the response is a unicode string and
        results in exception when printed directly i.e print response.
        """

        print smart_str(response)

    def assertResponseCode(self, response, status_code):
        """Asserts that the response status is status_code.
        """
        if response.status_code != status_code:
            verbose_codes = [
                httplib.FOUND,
            ]
            message_codes = [
                httplib.FORBIDDEN,
                httplib.BAD_REQUEST,
                httplib.NOT_FOUND,
            ]
            url_codes = [httplib.NOT_FOUND]

            if response.status_code in verbose_codes:
                print response

            if response.context and response.status_code in message_codes:
                try:
                    print response.context['message']
                except KeyError:
                    pass

            if response.status_code in url_codes:
                print response.request['PATH_INFO']

        self.assertEqual(status_code, response.status_code)

    def assertResponseOK(self, response):
        """Asserts that the response status is OK.
        """
        self.assertResponseCode(response, httplib.OK)

    def assertResponseRedirect(self, response):
        """Asserts that the response status is FOUND.
        """
        self.assertResponseCode(response, httplib.FOUND)

    def assertResponseForbidden(self, response):
        """Asserts that the response status is FORBIDDEN.
        """
        self.assertResponseCode(response, httplib.FORBIDDEN)

    def assertResponseBadRequest(self, response):
        """Asserts that the response status is BAD_REQUEST.
        """
        self.assertResponseCode(response, httplib.BAD_REQUEST)

    def assertResponseNotFound(self, response):
        """Asserts that the response status is NOT_FOUND.
        """
        self.assertResponseCode(response, httplib.NOT_FOUND)
コード例 #4
0
ファイル: test_utils.py プロジェクト: MiguelPin/e-cidadania
class ECDTestCase(TestCase):
    """Class which extends Django TestCase and adds methods specific to 
    e-cidadania for testing.
    """
    
    def init(self):
        """Performs set up for the tests.
        """
        self.client = Client(enforce_csrf_checks=False)
        self.username = '******'
        self.user_password = '******'
        self.admin_username = '******'
        self.admin_password = '******'
        self.foo_user = self.create_user('foo_user', 'foo_user_password')
        self.foo_admin = self.create_user('foo_admin', 'foo_admin_password')
        self.foo_mod = self.create_user('foo_mod', 'foo_mod_password')
        
        self.bar_user = self.create_user('bar_user', 'bar_user_password')
        self.bar_admin = self.create_user('bar_admin', 'bar_admin_password')
        self.bar_mod = self.create_user('bar_mod', 'bar_mod_password')
            
        space_properties = {'name': 'foo_space', 'url': 'foo_space_url',
                            'author': self.foo_admin, 'public': False, 
                            'mod_debate': True, 'mod_proposals': True,
                            'mod_news': True, 'mod_cal': True, 'mod_docs': True,
                            }
        self.foo_space = Space(**space_properties)
        self.foo_space.save()
        self.foo_space.admins.add(self.foo_admin)
        self.foo_space.mods.add(self.foo_mod)
        self.foo_space.users.add(self.foo_user)
        self.foo_space.save()
        
        space_properties.update({'author': self.bar_admin, 'name': 'bar_space',
                                 'url': 'bar_space_url', 'public': True,})
        self.bar_space = Space(**space_properties)
        self.bar_space.save()
        self.bar_space.admins.add(self.bar_admin)
        self.bar_space.mods.add(self.bar_mod)
        self.bar_space.users.add(self.bar_user)
        self.bar_space.save()
        
        debate_properties = {'space': self.foo_space, 'author': self.foo_admin}
        self.foo_debate = self.seed(Debate,properties=debate_properties)
        
        debate_properties.update({'space': self.bar_space, 
                                  'author': self.bar_admin})
        self.bar_debate = self.seed(Debate,debate_properties)
        
        column_properties = {'debate': self.foo_debate, 'criteria': 'private'}
        self.foo_column = Column(**column_properties)
        self.foo_column.save()
        
        column_properties.update({'debate': self.bar_debate,
                                  'criteria': 'public'})
        self.bar_column = Column(**column_properties)
        self.bar_column.save()
        
        row_properties = column_properties.copy()
        self.bar_row = Row(**row_properties)
        self.bar_row.save()
        
        row_properties.update({'debate': self.foo_debate})
        self.foo_row = Row(**row_properties)
        self.foo_row.save()
        
        note_properties = {'column': self.foo_column, 'row': self.foo_row,
                           'debate': self.foo_debate, 'author': self.foo_admin}
        self.foo_note = Note(**note_properties)
        self.foo_note.save()
        
        note_properties.update({'column': self.bar_column, 
                                'row': self.bar_row,
                                'debate': self.bar_debate,
                                'author': self.bar_admin})
        self.bar_note = Note(**note_properties)
        self.bar_note.save()
        
        post_properties = {'title': 'Foo news post', 'author': self.foo_user,
                           'pub_index': True, 'space': self.foo_space}
        self.foo_post = Post(**post_properties)
        self.foo_post.save()
        
        post_properties.update({'title': 'Bar news post',
                                'author': self.bar_user, 'space': self.bar_space})
        self.bar_post = Post(**post_properties)
        self.bar_post.save()
        
        proposal_set_properties = {'name': 'Foo Proposal Set', 
                                   'space': self.foo_space,
                                   'author': self.foo_admin, 
                                   'debate': self.foo_debate}
        self.foo_proposalset = ProposalSet(**proposal_set_properties)
        self.foo_proposalset.save()
        
        proposal_set_properties.update({'name': 'Bar Proposal Set',
                                        'space': self.bar_space,
                                        'author': self.bar_admin,
                                        'debate': self.bar_debate})
        self.bar_proposalset = ProposalSet(**proposal_set_properties)
        self.bar_proposalset.save()
        
    def seed(self, model, properties=None, constraints=None, follow_fk=None, 
             generate_fk=None, follow_m2m=None, factory=None, commit=True):
        """Generates and returns a new instance of the `model` with 
        properties in `properties`.
        """
        instance = seeder.seed(model=model, constraints=constraints, 
                               follow_fk=follow_fk, generate_fk=None, 
                               follow_m2m=None, factory=None,
                               model_properties=properties, commit=commit)
        
        return instance
    
    def seedn(self, count, model, properties, constraints=None, follow_fk=None, 
              generate_fk=None, follow_m2m=None, factory=None, commit=True):
        """Generates and returns `count` number of instances of `model` with
        properties in `properties`.
        """
        
        obj_list = seeder.seedn(count=count, model=model, constraints=constraints,
                                follow_fk=follow_fk, generate_fk=generate_fk,
                                follow_m2m=follow_m2m, factory=factory,
                                model_properties=properties, commit=True)
        return obj_list
    
    def create_user(self, username, password, email=None, properties=None,
                    logged_in=False):
        """Creates, saves and returns a user with a given username, password
        and email. If `properties` is supplied, it will be applied to the 
        created user.
        """
        
        user = User.objects.create_user(username=username, password=password, 
                                        email=email)
        if properties:
            for key in properties:
                setattr(user, key, properties[key])
            user.save()
            
        if logged_in:
            # log out the current user
            self.logout()
            # log in the new user
            user = self.login(username=username, password=password, email=email)
        return user
    
    def create_super_user(self, username='******', password='******', 
                          email='*****@*****.**', properties=None, 
                          logged_in=False):
        """Creates, saves and returns  a super user with a given username, 
        password and email. If `properties` is supplied, it will be applied
        to the created user.
        """
        
        super_user = User.objects.create_superuser(username=username,
                                                   password=password, 
                                                   email=email)
        if properties:
            for key in properties:
                setattr(super_user, key, properties[key])
            super_user.save()
        
        if logged_in:
            self.logout()
            super_user = self.login(username=username, password=password, 
                                    email=email)
        return super_user
    
    def login(self, username, password, email=None):
        """Logs in a user with the given username and password. If the user is 
        not present in the database, it will be created and logged in.
        
        We assume `username` is unique across the database.
        """
        
        try:
            user = User.objects.get(username=username)
        except Exception:
            user = None
        if user is None:
            user = self.create_user(username=username, password=password, 
                                    email=email)
        
        self.client.login(username=username, password=password)
        
        return user
    
    def logout(self):
        """Logs out the currently logged in user.
        """
        
        self.client.logout()
    
    def isLoggedIn(self, user=None):
        """Checks and returns True if a user is logged in otherwise returns
        False.
        """
        
        if '_auth_user_id' not in self.client.session:
            return False
        
        if (user.pk == self.client.session['_auth_user_id']):
            return True
    
        return False
    
    def getURL(self, name, args=None, kwargs=None):
        """Returns the url for the given `name` which may be a function name or
        url name.
        """
        return reverse(name, args=args, kwargs=kwargs)
    
    def get(self, url=None, url_name=None, data={}, follow=False, **extra):
        """
        Performs a get to the given url and returns the response.
        """
        if url is None and url_name is None:
            raise Exception("Please pass either url or url name")
            
        if url_name:
            url = self.getURL(url_name)
        
        response = self.client.get(url, data=data, follow=follow, extra=extra)
        return response
    
    def post(self, url,  data={}, follow=False, **extra):
        """
        Performs a post to the supplied url and returns the response.
        """
        
        response = self.client.post(path=url, data=data, follow=follow, 
                                    extra=extra)
        return response
    
    def printResponse(self, response):
        """Prints the response to the terminal.
        We need this method because the response is a unicode string and
        results in exception when printed directly i.e print response.
        """
        
        print smart_str(response)
    
    
    def assertResponseCode(self, response, status_code):
        """Asserts that the response status is status_code.
        """
        if response.status_code != status_code:
            verbose_codes = [
            httplib.FOUND,
            ]
            message_codes = [
            httplib.FORBIDDEN, httplib.BAD_REQUEST, httplib.NOT_FOUND,
            ]
            url_codes = [httplib.NOT_FOUND]
            
            if response.status_code in verbose_codes:
                print response

            if response.context and response.status_code in message_codes:
                try:
                    print response.context['message']
                except KeyError:
                    pass
    
            if response.status_code in url_codes:
                print response.request['PATH_INFO']

        self.assertEqual(status_code, response.status_code)

    def assertResponseOK(self, response):
        """Asserts that the response status is OK.
        """
        self.assertResponseCode(response, httplib.OK)

    def assertResponseRedirect(self, response):
        """Asserts that the response status is FOUND.
        """
        self.assertResponseCode(response, httplib.FOUND)

    def assertResponseForbidden(self, response):
        """Asserts that the response status is FORBIDDEN.
        """
        self.assertResponseCode(response, httplib.FORBIDDEN)

    def assertResponseBadRequest(self, response):
        """Asserts that the response status is BAD_REQUEST.
        """
        self.assertResponseCode(response, httplib.BAD_REQUEST)

    def assertResponseNotFound(self, response):
        """Asserts that the response status is NOT_FOUND.
        """
        self.assertResponseCode(response, httplib.NOT_FOUND)