def test_dashboard(self, mock_database,mock_response): mock_database.return_value.cursor.return_value.fetchall.return_value = [[1, 'C:/some_file', 'function_name']] #print mdb.connect().cursor().fetchall() request = {'user':'******'} import views views.dashboard(request) mock_response.assert_called_with('dashboard.html', {'traces': [[1, 'C:/some_file/function_name']]})
def login_required(func, *args, **kwargs): request = args[0] # request es el primer parametro que pasamos if request.user.is_authenticated(): if request.user.username is not None: return func(*args, **kwargs) else: # falta el nombre de usuario, redirigimos al panel from views import dashboard return dashboard(request) from views import login return login(args[0])
def username_required(func, *args, **kwargs): request = args[0] # request es el primer parametro que pasamos if request.user.is_authenticated() and request.user.username is None: from views import dashboard return dashboard(request) return func(*args, **kwargs)