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