def setUp(self):
translation.activate('en') # Set English
super(SkillSetListCreateAPIViewWithTenantTestCase, self).setUp()
# Load up the dependat.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
call_command('populate_tenant_sample_db',
TEST_SCHEMA_NAME,
verbosity=0)
# Get objects.
self.customer = Customer.objects.get(owner__email='*****@*****.**')
self.associate = Associate.objects.get(
owner__email='*****@*****.**')
# Get users.
exec_user = SharedUser.objects.get(email='*****@*****.**')
manager_user = SharedUser.objects.get(email='*****@*****.**')
frontline_user = SharedUser.objects.get(email='*****@*****.**')
associate_user = SharedUser.objects.get(email='*****@*****.**')
customer_user = SharedUser.objects.get(email='*****@*****.**')
# Get tokens.
exec_token, exec_orig_iat = get_jwt_token_and_orig_iat(exec_user)
manager_token, manager_orig_iat = get_jwt_token_and_orig_iat(
manager_user)
frontline_token, frontline_orig_iat = get_jwt_token_and_orig_iat(
frontline_user)
associate_token, associate_orig_iat = get_jwt_token_and_orig_iat(
associate_user)
customer_token, customer_orig_iat = get_jwt_token_and_orig_iat(
customer_user)
# Setup.
self.unauthorized_client = TenantClient(self.tenant)
self.exec_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(exec_token))
self.exec_client.login(username='******',
password=TEST_USER_PASSWORD)
self.manager_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(manager_token))
self.manager_client.login(username='******',
password=TEST_USER_PASSWORD)
self.staff_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(manager_token))
self.staff_client.login(username='******',
password=TEST_USER_PASSWORD)
self.customer_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(manager_token))
self.customer_client.login(username='******',
password=TEST_USER_PASSWORD)
self.skill_set = SkillSet.objects.create(
category="One",
sub_category="Two",
)
def setUp(self):
"""
Run at the beginning of every unit test.
"""
super(TestTenantTeamViews, self).setUp()
# Setup our app and account.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
call_command('populate_tenant_sample_db',
TEST_SCHEMA_NAME,
verbosity=0)
# Get user and credentials.
user = SharedUser.objects.get(email="*****@*****.**")
user2 = SharedUser.objects.get(email="*****@*****.**")
token, orig_iat = get_jwt_token_and_orig_iat(user)
token2, orig_iat2 = get_jwt_token_and_orig_iat(user2)
# Setup our clients.
self.anon_c = TenantClient(self.tenant)
self.exec_auth_c = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.exec_auth_c.login(username="******",
password="******")
self.frontline_auth_c = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(token2))
self.frontline_auth_c.login(username="******",
password="******")
TaskItem.objects.update_or_create(
id=1,
defaults={
'id':
1,
'type_of':
FOLLOW_UP_CUSTOMER_SURVEY_TASK_ITEM_TYPE_OF_ID,
'title':
_('Completion Survey'),
'description':
_('Please call up the client and perform the satisfaction survey.'
),
'due_date':
timezone.now(),
'is_closed':
False,
'job':
None,
'ongoing_job':
None,
# 'created_by': None,
# created_from
# created_from_is_public
# last_modified_by
})
def setUp(self):
"""
Run at the beginning of every unit test.
"""
super(TestTenantDashboardViews, self).setUp()
# Setup our app and account.
call_command('init_app', verbosity=0)
call_command(
'create_shared_account',
TEST_USER_EMAIL,
TEST_USER_PASSWORD,
"Bart",
"Mika",
verbosity=0
)
# Get user and credentials.
user = SharedUser.objects.get()
token, orig_iat = get_jwt_token_and_orig_iat(user)
# Setup our clients.
self.anon_c = TenantClient(self.tenant)
self.auth_c = TenantClient(self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.auth_c.login(
username=TEST_USER_USERNAME,
password=TEST_USER_PASSWORD
)
def test_api_logout(self):
# Log in the the account.
user = SharedUser.objects.get()
token, orig_iat = get_jwt_token_and_orig_iat(user)
# Log out.
logout_url = reverse('workery_logout_api_endpoint')
data = {
'email_or_username': TEST_USER_EMAIL,
'password': TEST_USER_PASSWORD,
}
response = self.c.post(logout_url,
json.dumps(data),
HTTP_AUTHORIZATION='JWT {0}'.format(token),
content_type='application/json')
# Confirm.
self.assertEqual(response.status_code, status.HTTP_200_OK)
def setUp(self):
"""
Run at the beginning of every unit test.
"""
super(TestTenantTeamViews, self).setUp()
# Setup our app and account.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
# Create the account.
call_command(
'create_tenant_account',
TEST_SCHEMA_NAME,
constants.MANAGEMENT_GROUP_ID,
TEST_USER_EMAIL,
TEST_USER_PASSWORD,
"Bart",
"Mika",
TEST_USER_TEL_NUM,
TEST_USER_TEL_EX_NUM,
TEST_USER_CELL_NUM,
"CA",
"London",
"Ontario",
"", # Post Offic #
"N6H 1B4",
"78 Riverside Drive",
"", # Extra line.
verbosity=0)
# Get user and credentials.
user = SharedUser.objects.get()
token, orig_iat = get_jwt_token_and_orig_iat(user)
# Setup our clients.
self.anon_c = TenantClient(self.tenant)
self.auth_c = TenantClient(self.tenant,
HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.auth_c.login(username=TEST_USER_USERNAME,
password=TEST_USER_PASSWORD)
def setUp(self):
"""
Run at the beginning of every unit test.
"""
super(TestTenantAssociateViews, self).setUp()
# Setup our app and account.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
call_command('populate_tenant_sample_db', TEST_SCHEMA_NAME, verbosity=0)
# Get user and credentials.
user = SharedUser.objects.get(email="*****@*****.**")
token, orig_iat = get_jwt_token_and_orig_iat(user)
# Setup our clients.
self.anon_c = TenantClient(self.tenant)
self.auth_c = TenantClient(self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.auth_c.login(
username = "******",
password = "******"
)
def setUp(self):
translation.activate('en') # Set English
SharedUser.objects.delete_all()
super(SharedFranchiseListAPIViewWithPublicSchemaTestCase, self).setUp()
self.anon_client = TenantClient(self.tenant)
call_command('init_app', verbosity=0)
# Update the tenant.
self.tenant.name = 'Over 55 (London) Inc.',
self.tenant.alternate_name = "Over55",
self.tenant.description = "Located at the Forks of the Thames in ...",
self.tenant.address_country = "CA",
self.tenant.address_locality = "London",
self.tenant.address_region = "Ontario",
self.tenant.post_office_box_number = "", # Post Offic #
self.tenant.postal_code = "N6H 1B4",
self.tenant.street_address = "78 Riverside Drive",
self.tenant.street_address_extra = "", # Extra line.
self.tenant.save()
# Create our test user.
call_command('create_shared_account',
TEST_USER_EMAIL,
TEST_USER_PASSWORD,
"Bart",
"Mika",
verbosity=0)
# Initialize our test data.
self.user = SharedUser.objects.get(email=TEST_USER_EMAIL)
token, orig_iat = get_jwt_token_and_orig_iat(self.user)
self.authorized_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.authorized_client.login(username=TEST_USER_USERNAME,
password=TEST_USER_PASSWORD)
def setUp(self):
"""
Run at the beginning of every unit test.
"""
super(TestTenantReportViews, self).setUp()
# Setup our app and account.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
call_command('populate_tenant_sample_db',
TEST_SCHEMA_NAME,
verbosity=0)
# Create the account.
call_command(
'create_tenant_account',
TEST_SCHEMA_NAME,
MANAGEMENT_GROUP_ID,
TEST_USER_EMAIL,
TEST_USER_PASSWORD,
"Bart",
"Mika",
TEST_USER_TEL_NUM,
TEST_USER_TEL_EX_NUM,
TEST_USER_CELL_NUM,
"CA",
"London",
"Ontario",
"", # Post Offic #
"N6H 1B4",
"78 Riverside Drive",
"", # Extra line.
verbosity=0)
# Get user and credentials.
user = SharedUser.objects.filter(email=TEST_USER_EMAIL).first()
token, orig_iat = get_jwt_token_and_orig_iat(user)
# Setup our clients.
self.anon_c = TenantClient(self.tenant)
self.auth_c = TenantClient(self.tenant,
HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.auth_c.login(username=TEST_USER_USERNAME,
password=TEST_USER_PASSWORD)
TaskItem.objects.update_or_create(
id=1,
defaults={
'id':
1,
'type_of':
FOLLOW_UP_CUSTOMER_SURVEY_TASK_ITEM_TYPE_OF_ID,
'title':
_('Completion Survey'),
'description':
_('Please call up the client and perform the satisfaction survey.'
),
'due_date':
timezone.now(),
'is_closed':
False,
'job':
None,
'ongoing_job':
None,
# 'created_by': None,
# created_from
# created_from_is_public
# last_modified_by
})
def setUp(self):
translation.activate('en') # Set English
super(AssociateListCreateAPIViewWithTenantTestCase, self).setUp()
# Load up the dependat.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
# Create the account.
call_command(
'create_tenant_account',
TEST_SCHEMA_NAME,
constants.MANAGEMENT_GROUP_ID,
TEST_USER_EMAIL,
TEST_USER_PASSWORD,
"Bart",
"Mika",
TEST_USER_TEL_NUM,
TEST_USER_TEL_EX_NUM,
TEST_USER_CELL_NUM,
"CA",
"London",
"Ontario",
"", # Post Offic #
"N6H 1B4",
"78 Riverside Drive",
"", # Extra line.
verbosity=0)
# Create the account.
call_command(
'create_tenant_account',
TEST_SCHEMA_NAME,
constants.MANAGEMENT_GROUP_ID,
TEST_ALERNATE_USER_EMAIL,
TEST_USER_PASSWORD,
"Bart",
"Mika",
TEST_USER_TEL_NUM,
TEST_USER_TEL_EX_NUM,
TEST_USER_CELL_NUM,
"CA",
"London",
"Ontario",
"", # Post Offic #
"N6H 1B4",
"78 Riverside Drive",
"", # Extra line.
verbosity=0)
# Initialize our test data.
self.user = SharedUser.objects.get(email=TEST_USER_EMAIL)
token, orig_iat = get_jwt_token_and_orig_iat(self.user)
# Setup.
self.unauthorized_client = TenantClient(self.tenant)
self.authorized_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(token))
self.authorized_client.login(username=TEST_USER_USERNAME,
password=TEST_USER_PASSWORD)
# Create our associate.
connection.set_schema(TEST_SCHEMA_NAME, True) # Switch to Tenant.
self.associate = Associate.objects.create(owner=self.user,
given_name="Bart",
last_name="Mika")
self.alernate_associate = Associate.objects.create(
owner=SharedUser.objects.get(email=TEST_ALERNATE_USER_EMAIL),
given_name="Rodolfo",
last_name="Martinez")
def setUp(self):
translation.activate('en') # Set English
super(WorkOrderListCreateAPIViewWithTenantTestCase, self).setUp()
# Load up the dependat.
call_command('init_app', verbosity=0)
call_command('populate_tenant_content', TEST_SCHEMA_NAME, verbosity=0)
call_command('populate_tenant_sample_db',
TEST_SCHEMA_NAME,
verbosity=0)
# Get objects.
self.customer = Customer.objects.get(owner__email='*****@*****.**')
self.associate = Associate.objects.get(
owner__email='*****@*****.**')
# Get users.
exec_user = SharedUser.objects.get(email='*****@*****.**')
manager_user = SharedUser.objects.get(email='*****@*****.**')
frontline_user = SharedUser.objects.get(email='*****@*****.**')
associate_user = SharedUser.objects.get(email='*****@*****.**')
customer_user = SharedUser.objects.get(email='*****@*****.**')
# Get tokens.
exec_token, exec_orig_iat = get_jwt_token_and_orig_iat(exec_user)
manager_token, manager_orig_iat = get_jwt_token_and_orig_iat(
manager_user)
frontline_token, frontline_orig_iat = get_jwt_token_and_orig_iat(
frontline_user)
associate_token, associate_orig_iat = get_jwt_token_and_orig_iat(
associate_user)
customer_token, customer_orig_iat = get_jwt_token_and_orig_iat(
customer_user)
# Setup.
self.unauthorized_client = TenantClient(self.tenant)
self.exec_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(exec_token))
self.exec_client.login(username='******',
password=TEST_USER_PASSWORD)
self.manager_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(manager_token))
self.manager_client.login(username='******',
password=TEST_USER_PASSWORD)
self.staff_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(manager_token))
self.staff_client.login(username='******',
password=TEST_USER_PASSWORD)
self.customer_client = TenantClient(
self.tenant, HTTP_AUTHORIZATION='JWT {0}'.format(manager_token))
self.customer_client.login(username='******',
password=TEST_USER_PASSWORD)
# Load up the tenant.
connection.set_schema(TEST_SCHEMA_NAME, True) # Switch to Tenant.
# Tag
self.tag = Tag.objects.create(text="Test Tag")
# Create our order.
self.order = WorkOrder.objects.create(
customer=Customer.objects.get(owner__email="*****@*****.**"),
associate=Associate.objects.get(
owner__email="*****@*****.**"),
assignment_date=timezone.now(),
created_by=SharedUser.objects.get(email="*****@*****.**"),
last_modified_by=None)
self.order.tags.set([self.tag])
