|  | 
| 14 | 14 | 
 | 
| 15 | 15 | 
 | 
| 16 | 16 | class BaseTest(TestCase): | 
| 17 |  | -    def setUp(self): | 
| 18 |  | -        self.foo_user = UserModel.objects.create_user("foo_user", "test@example.com", "123456") | 
| 19 |  | -        self.bar_user = UserModel.objects.create_user("bar_user", "dev@example.com", "123456") | 
| 20 |  | - | 
| 21 |  | -    def tearDown(self): | 
| 22 |  | -        self.foo_user.delete() | 
| 23 |  | -        self.bar_user.delete() | 
|  | 17 | +    @classmethod | 
|  | 18 | +    def setUpTestData(cls): | 
|  | 19 | +        cls.foo_user = UserModel.objects.create_user("foo_user", "test@example.com", "123456") | 
|  | 20 | +        cls.bar_user = UserModel.objects.create_user("bar_user", "dev@example.com", "123456") | 
| 24 | 21 | 
 | 
| 25 | 22 | 
 | 
| 26 | 23 | @pytest.mark.usefixtures("oauth2_settings") | 
| @@ -67,29 +64,26 @@ def test_application_registration_user(self): | 
| 67 | 64 | 
 | 
| 68 | 65 | 
 | 
| 69 | 66 | class TestApplicationViews(BaseTest): | 
| 70 |  | -    def _create_application(self, name, user): | 
| 71 |  | -        app = Application.objects.create( | 
|  | 67 | +    @classmethod | 
|  | 68 | +    def _create_application(cls, name, user): | 
|  | 69 | +        return Application.objects.create( | 
| 72 | 70 |             name=name, | 
| 73 | 71 |             redirect_uris="http://example.com", | 
| 74 | 72 |             post_logout_redirect_uris="http://other_example.com", | 
| 75 | 73 |             client_type=Application.CLIENT_CONFIDENTIAL, | 
| 76 | 74 |             authorization_grant_type=Application.GRANT_AUTHORIZATION_CODE, | 
| 77 | 75 |             user=user, | 
| 78 | 76 |         ) | 
| 79 |  | -        return app | 
| 80 |  | - | 
| 81 |  | -    def setUp(self): | 
| 82 |  | -        super().setUp() | 
| 83 |  | -        self.app_foo_1 = self._create_application("app foo_user 1", self.foo_user) | 
| 84 |  | -        self.app_foo_2 = self._create_application("app foo_user 2", self.foo_user) | 
| 85 |  | -        self.app_foo_3 = self._create_application("app foo_user 3", self.foo_user) | 
| 86 | 77 | 
 | 
| 87 |  | -        self.app_bar_1 = self._create_application("app bar_user 1", self.bar_user) | 
| 88 |  | -        self.app_bar_2 = self._create_application("app bar_user 2", self.bar_user) | 
|  | 78 | +    @classmethod | 
|  | 79 | +    def setUpTestData(cls): | 
|  | 80 | +        super().setUpTestData() | 
|  | 81 | +        cls.app_foo_1 = cls._create_application("app foo_user 1", cls.foo_user) | 
|  | 82 | +        cls.app_foo_2 = cls._create_application("app foo_user 2", cls.foo_user) | 
|  | 83 | +        cls.app_foo_3 = cls._create_application("app foo_user 3", cls.foo_user) | 
| 89 | 84 | 
 | 
| 90 |  | -    def tearDown(self): | 
| 91 |  | -        super().tearDown() | 
| 92 |  | -        get_application_model().objects.all().delete() | 
|  | 85 | +        cls.app_bar_1 = cls._create_application("app bar_user 1", cls.bar_user) | 
|  | 86 | +        cls.app_bar_2 = cls._create_application("app bar_user 2", cls.bar_user) | 
| 93 | 87 | 
 | 
| 94 | 88 |     def test_application_list(self): | 
| 95 | 89 |         self.client.login(username="foo_user", password="123456") | 
| @@ -132,8 +126,8 @@ def test_application_udpate(self): | 
| 132 | 126 |         self.assertRedirects(response, reverse("oauth2_provider:detail", args=(self.app_foo_1.pk,))) | 
| 133 | 127 | 
 | 
| 134 | 128 |         self.app_foo_1.refresh_from_db() | 
| 135 |  | -        self.assertEquals(self.app_foo_1.client_id, form_data["client_id"]) | 
| 136 |  | -        self.assertEquals(self.app_foo_1.redirect_uris, form_data["redirect_uris"]) | 
| 137 |  | -        self.assertEquals(self.app_foo_1.post_logout_redirect_uris, form_data["post_logout_redirect_uris"]) | 
| 138 |  | -        self.assertEquals(self.app_foo_1.client_type, form_data["client_type"]) | 
| 139 |  | -        self.assertEquals(self.app_foo_1.authorization_grant_type, form_data["authorization_grant_type"]) | 
|  | 129 | +        self.assertEqual(self.app_foo_1.client_id, form_data["client_id"]) | 
|  | 130 | +        self.assertEqual(self.app_foo_1.redirect_uris, form_data["redirect_uris"]) | 
|  | 131 | +        self.assertEqual(self.app_foo_1.post_logout_redirect_uris, form_data["post_logout_redirect_uris"]) | 
|  | 132 | +        self.assertEqual(self.app_foo_1.client_type, form_data["client_type"]) | 
|  | 133 | +        self.assertEqual(self.app_foo_1.authorization_grant_type, form_data["authorization_grant_type"]) | 
0 commit comments