def test_patch_about_details_succeeds(self):
"""
Test that the endpoint '/about/create_or_update' updates about details when a patch request is
made.
:return: None
"""
new_role = RoleFactory.create(name='admin')
UserRoleFactory.create(user_id=self.user_id(), role_id=new_role.id)
AboutFactory.create(
details=base64.b64encode(self.html_data['data'].encode('utf-8')))
response = self.client().patch(
self.make_url("/about/create_or_update"),
headers=self.headers(),
data=self.encode_to_json_string(self.html_data_update))
response_json = self.decode_from_json_string(
response.data.decode('utf-8'))
self.assertEqual(response.status_code, 201)
self.assertEqual(response_json['msg'], 'OK')
self.assertEqual(response_json['payload']['data']["details"],
self.html_data_update["data"])
def test_update_faq_succeeds(self):
new_role = RoleFactory.create(name='admin')
new_user_role = UserRoleFactory.create(user_id=self.user_id(),
role=new_role)
faq = FaqFactory()
update_faq_info = FaqFactory.build()
faq_data = dict(question=update_faq_info.question,
answer=update_faq_info.answer)
response = self.client().patch(
self.make_url(f"/faqs/{faq.id}"),
headers=self.headers(),
data=self.encode_to_json_string(faq_data))
response_json = self.decode_from_json_string(
response.data.decode('utf-8'))
self.assertEqual(response.status_code, 201)
self.assertEqual(response_json['msg'], 'OK')
self.assertEqual(response_json['payload']['faq']['question'],
update_faq_info.question)
self.assertEqual(response_json['payload']['faq']['answer'],
update_faq_info.answer)
self.assertEqual(response_json['payload']['faq']['question'],
faq.question)
self.assertEqual(response_json['payload']['faq']['answer'], faq.answer)
def test_create_about_endpoint_rejects_invalid_data(self):
"""
Test that the endpoint '/about/create_or_update' rejects json request without data field
:return: None
"""
new_role = RoleFactory.create(name='admin')
UserRoleFactory.create(user_id=self.user_id(), role=new_role)
response = self.client().post(
self.make_url("/about/create_or_update"), headers=self.headers(),
data=self.encode_to_json_string(self.html_data_wrong_key)
)
response_json = self.decode_from_json_string(response.data.decode('utf-8'))
self.assertEqual(response.status_code, 400)
self.assertEqual(response_json['msg'], 'Bad Request - data is required')
def test_exclude_works_user_role_instance(self):
user_role = UserRoleFactory.build()
new_user_role = self.repo.new_user_role(user_role.role_id.id,
user_role.user_id,
user_role.location_id,
user_role.email)
excluded_response = new_user_role.to_dict(exclude=["user_id"])
self.assertFalse(excluded_response.get("user_id", False))
def test_new_user_role_method_returns_new_user_role_object(self):
user_role = UserRoleFactory.build()
new_user_role = self.repo.new_user_role(user_role.role_id.id,
user_role.user_id,
user_role.location_id,
user_role.email)
self.assertIsInstance(new_user_role, UserRole)
self.assertEqual(str(new_user_role.user_id), str(user_role.user_id))
self.assertEqual(new_user_role.location_id, user_role.location_id)
def test_new_user_method_returns_new_user_object(self):
user = UserFactory.build(slack_id='-Lcel5YdCjsvKOG3zsrV')
role = RoleFactory()
user_role = UserRoleFactory(role_id=role.id)
new_user = self.repo.new_user(user.first_name, user.last_name, user.image_url,
user_id=user.user_id, slack_id=user.slack_id, user_type=user_role)
self.assertIsInstance(new_user, User)
self.assertEqual(str(new_user.slack_id), str(user.slack_id))
self.assertEqual(new_user.first_name, user.first_name)
self.assertEqual(new_user.image_url, user.image_url)
def test_create_about_details_succeeds(self):
"""
Test that the endpoint '/about/create_or_update' creates about details when a post request is
made in case details do not already exist in the database
:return: None
"""
new_role = RoleFactory.create(name='admin')
UserRoleFactory.create(user_id=self.user_id(), role=new_role)
response = self.client().post(
self.make_url("/about/create_or_update"), headers=self.headers(),
data=self.encode_to_json_string(self.html_data)
)
response_json = self.decode_from_json_string(response.data.decode('utf-8'))
self.assertEqual(response.status_code, 201)
self.assertEqual(response_json['msg'], 'OK')
self.assertEqual(response_json['payload']['data']["details"], self.html_data["data"])
def test_new_user_role_method_returns_new_user_role_object(self):
role = RoleFactory.create()
location = LocationFactory.create()
user_role = UserRoleFactory.build(role_id=role.id, location=location)
new_user_role = self.repo.new_user_role(user_role.role_id,
user_role.user_id,
user_role.location_id,
user_role.email)
self.assertIsInstance(new_user_role, UserRole)
self.assertEqual(str(new_user_role.user_id), str(user_role.user_id))
self.assertEqual(new_user_role.location.id, user_role.location.id)
def test_exclude_works_user_role_instance(self):
role = RoleFactory.create()
user = UserFactory.create()
user_role = UserRoleFactory.build(
role_id=role.id,
user_id=user.id,
)
new_user_role = self.repo.new_user_role(
user_id=user_role.user_id,
role_id=user_role.role_id,
)
excluded_response = new_user_role.to_dict(exclude=["user_id"])
self.assertFalse(excluded_response.get("user_id", False))
def test_delete_faq_succeeds(self):
new_role = RoleFactory.create(name='admin')
new_user_role = UserRoleFactory.create(user_id=self.user_id(),
role=new_role)
faq = FaqFactory()
response = self.client().delete(self.make_url(f"/faqs/{faq.id}"),
headers=self.headers())
response_json = self.decode_from_json_string(
response.data.decode('utf-8'))
self.assertEqual(response.status_code, 200)
self.assertEqual(response_json['msg'], 'FAQ deleted successfully')
def test_run_5_minute_method(self):
role = RoleFactory.create()
location = LocationFactory.create()
user_role = UserRoleFactory.build(role_id=role.id, location=location)
UserRoleRepo().new_user_role(user_role.role_id, user_role.user_id,
user_role.location_id, user_role.email)
Cron(self.app).run_5_minute()
results = self.redis_set.get(user_role.email[0])
self.assertEqual(user_role.email, results[0])
results = self.redis_set.get(user_role.email[0:1])
self.assertEqual(user_role.email, results[0])
results = self.redis_set.get(user_role.email[0:2])
self.assertEqual(user_role.email, results[0])
def test_new_user_role_updates_cache(self):
role = RoleFactory.create()
location = LocationFactory.create()
user_role = UserRoleFactory.build(role_id=role.id, location=location)
self.repo.new_user_role(user_role.role_id, user_role.user_id,
user_role.location_id, user_role.email)
results = self.redis_set.get(user_role.email[0:1])
self.assertTrue(user_role.email in results)
results = self.redis_set.get(user_role.email[0:3])
self.assertTrue(user_role.email in results)
results = self.redis_set.get(user_role.email[0:5])
self.assertTrue(user_role.email in results)
results = self.redis_set.get(user_role.email[0:len(user_role.email) -
1])
self.assertTrue(user_role.email in results)
def create_admin(self):
role = RoleFactory.create(name='admin')
user_id = self.user_id()
UserRoleFactory.create(user_id=user_id, role=role)
def create_user_role(keyword, role_name):
role = RoleFactory.create(name=role_name)
user_id = BaseTestCase.user_id()
PermissionFactory.create(keyword=keyword, role=role)
return UserRoleFactory.create(user_id=user_id, role=role), user_id
