def test_get_database_table(data_fixture): user = data_fixture.create_user() table = data_fixture.create_database_table(user=user) table_2 = data_fixture.create_database_table() handler = TableHandler() with pytest.raises(UserNotInGroupError): handler.get_table(user=user, table_id=table_2.id) with pytest.raises(TableDoesNotExist): handler.get_table(user=user, table_id=99999) table_copy = handler.get_table(user=user, table_id=table.id) assert table_copy.id == table.id
def can_add(self, user, web_socket_id, table_id, **kwargs): """ The user should only have access to this page if the table exists and if he has access to the table. """ if not table_id: return False try: handler = TableHandler() handler.get_table(user, table_id) except (UserNotInGroupError, TableDoesNotExist): return False return True
def test_get_database_table(data_fixture): user = data_fixture.create_user() table = data_fixture.create_database_table(user=user) data_fixture.create_database_table() handler = TableHandler() with pytest.raises(TableDoesNotExist): handler.get_table(table_id=99999) # If the error is raised we know for sure that the base query has resolved. with pytest.raises(AttributeError): handler.get_table( table_id=table.id, base_queryset=Table.objects.prefetch_related("UNKNOWN")) table_copy = handler.get_table(table_id=table.id) assert table_copy.id == table.id
def can_add(self, user, web_socket_id, table_id, **kwargs): """ The user should only have access to this page if the table exists and if he has access to the table. """ if not table_id: return False try: handler = TableHandler() table = handler.get_table(table_id) table.database.group.has_user(user, raise_error=True) except (UserNotInGroup, TableDoesNotExist): return False return True