class BaseTestCase(APITestCase): def setUp(self): """ Sets up a number of test entites in the database for testing. Included: user: A webapp user organization: An organization that user belongs to domain: A domain that belongs to organization """ # Creating Users self.user = MBUser( email='*****@*****.**', first_name='test', last_name='tester', is_admin=False, is_platform=False, is_webapp=True, ) self.user.set_password('test') self.user.save() self.client.login(username='******', password='******') # Organization self.organization = Organization( name='Test Organization', address_1='123 Any Street', address_2='Suite 001', city='New York', state='NY', phone='555-555-1234' ) self.organization.save() self.user_meta = UserMeta( user=self.user, organization=self.organization ) self.user_meta.save() # Domain self.domain = Domain( domain_name='test.com', organization=self.organization, whitelisted=False ) self.domain.save()
def get_queryset(self): domain_id = self.kwargs.get('pk') if not domain_id: raise Http404 domain = Domain.get_by_id(domain_id) metas = UserMeta.objects.filter(organization=domain.organization) return [meta.user for meta in metas]
def handle(self, *args, **options): # Create the organization org = Organization(**organization).save() if not org: self.stdout.write("Org failed") return self.stdout.write("Successfully created the MailBeaker organization") # Create the domain in the organization dom = Domain(organization=org, **domain) self.stdout.write("Successfully created the MailBeaker domain") # Create all of the users in the domain for user in users: mb_user = MBUser(**user).save() user_meta = UserMeta(user=mb_user, organization=org) self.stdout.write("Successfully created the %s user and associated meta" % mb_user.email) # Create all of the rules for rule in rules: rule_entity = Rule(domain=dom, **rule) self.stdout.write("Successfully created the rule: %s" % rule['description'])