From 8ca66abc8806fc023aa24929fe26c48f0a3779f6 Mon Sep 17 00:00:00 2001 From: ANDRE Simon Date: Wed, 26 Apr 2023 02:04:17 +0200 Subject: [PATCH] Mix django unittest client and ninja unittest api to start a custom django ninja unittest client --- tests/testcase.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/testcase.py b/tests/testcase.py index ed47f0c..3249b3b 100644 --- a/tests/testcase.py +++ b/tests/testcase.py @@ -1,9 +1,16 @@ -from django.test import TestCase +from django.test import TestCase, Client from main.api import api from django.contrib.auth import get_user_model +from ninja.testing import TestClient + + +class NinjaClient(Client, TestClient): + pass class NinjaTestCase(TestCase): + client_class = NinjaClient + def create_client_helper(self): """Create test user and log them to dedicated client.""" self.user = get_user_model().objects.create_user( @@ -12,9 +19,9 @@ def create_client_helper(self): self.admin_user = get_user_model().objects.create_superuser( username="admin", password="admin" ) - self.user_client = self.client_class() + self.user_client = self.client_class(api) self.user_client.force_login(self.user) - self.admin_client = self.client_class() + self.admin_client = self.client_class(api) self.admin_client.force_login(self.admin_user) def link_api_helper(self):