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