def test_should_add_view(self):
view = View(
view_name="Test"
)
view.add()
self.assertIn(view, self.db.session)
class AdminTest(Test):
def setUp(self):
Test.setUp(self)
self.user = User(username="******", password="******")
self.user.add()
self.view = View(view_name="Test View")
self.view.add()
class TestGetAll(ViewTest):
def setUp(self):
ViewTest.setUp(self)
self.view2 = View(
view_name="Test2"
)
self.view2.add()
def test_should_return_list_of_all_views(self):
views = View.get_all()
self.assertEqual(views, [self.view, self.view2])
def setUp(self):
UserTest.setUp(self)
self.admin_permission = Permission(permission_name="Admin")
self.admin_permission.add()
self.admin_view = View(view_name="Admin View")
self.admin_view.add()
self.admin_view.add_permission(self.admin_permission)
self.admin_user = User(username="******", password="******")
self.admin_user.add()
self.admin_user.add_permission(self.admin_permission)
self.quality_permission = Permission(permission_name="quality")
self.quality_permission.add()
self.quality_view = View(view_name="Quality View")
self.quality_view.add()
self.quality_view.add_permission(self.quality_permission)
self.normal_user = User(username="******", password="******")
self.normal_user.add()
def set_view_permissions(view_name, permission_names):
view = View.search(view_name)
permissions = get_permissions(permission_names)
if not view:
view = View(
view_name=view_name
)
view.add()
view.add_permissions(permissions)
def has_view_permissions(self, view_name):
view = View.search(view_name)
if self.is_dev() or not view:
return True
if self.is_admin() and not view.requires_dev():
return True
for permission in self.permissions:
if permission in set(view.permissions):
return True
return False
def main_page():
users = User.get_all()
views = View.get_all()
if request.method == "POST":
search_term = request.form['search_term']
user = User.search(search_term)
if user:
return redirect(
url_for('user.update', id=user.id)
)
view = View.search(search_term)
if view:
return redirect(
url_for('view.update', id=view.id)
)
return render_template(
"admin/main-page.html",
user_heads=username_head,
users=users,
views=views
)
def update(id):
view = View.get(id)
try:
receipt_image = request.files["receipt_image"]
except KeyError:
receipt_image = None
if receipt_image:
save_image(receipt_image)
checked_permissions = get_checked_permissions()
view.update_permissions(checked_permissions)
if not request.referrer:
url = url_for('home.main_page')
else:
url = request.referrer
return redirect(url)
def test_should_return_list_of_all_views(self):
views = View.get_all()
self.assertEqual(views, [self.view, self.view2])
def setUp(self):
Test.setUp(self)
self.view = View(view_name="Test View")
self.view.add()
class ViewTest(Test):
def setUp(self):
Test.setUp(self)
self.view = View(view_name="Test View")
self.view.add()
def inject_view_and_permissions():
return dict(
view=View.search(request.endpoint),
permissions=Permission.get_all()
)
def test_should_return_none_given_invalid_name(self):
view = View.search("Non existent name")
self.assertEqual(view, None)
def test_should_return_permission_given_valid_name(self):
view = View.search("Test View")
self.assertEqual(view, self.view)
def test_should_return_view_given_valid_id(self):
view = View.get(1)
self.assertEqual(view, self.view)
def test_should_return_none_given_invalid_id(self):
view = View.get(2)
self.assertEqual(view, None)
def setUp(self):
ViewTest.setUp(self)
self.view2 = View(
view_name="Test2"
)
self.view2.add()
