Merge pull request #24 from MarcinWieczorek/patch-testclient

Add APIClient type to test cases

Author: sobolevn

README.md

@@ -2,7 +2,7 @@
 
 # pep484 stubs for Django REST framework
 
-[![Build Status](https://travis-ci.org/mkurnikov/djangorestframework-stubs.svg?branch=master)](https://travis-ci.org/mkurnikov/djangorestframework-stubs)
+[![Build Status](https://travis-ci.com/typeddjango/djangorestframework-stubs.svg?branch=master)](https://travis-ci.com/typeddjango/djangorestframework-stubs)
 [![Checked with mypy](http://www.mypy-lang.org/static/mypy_badge.svg)](http://mypy-lang.org/)
 
 
@@ -30,4 +30,4 @@ in your `mypy.ini` file.
 
 We have Gitter here https://gitter.im/mypy-django/Lobby.
 
-If you think you have more generic typing issue, please refer to https://github.com/python/mypy and their Gitter.
+If you think you have more generic typing issue, please refer to https://github.com/python/mypy and their Gitter.

rest_framework-stubs/test.pyi

@@ -60,8 +60,16 @@ class APIClient(DjangoClient):
         **extra: Any
     ) -> Response: ...
 
-class APITransactionTestCase(testcases.TransactionTestCase): ...
-class APITestCase(testcases.TestCase): ...
-class APISimpleTestCase(testcases.SimpleTestCase): ...
-class APILiveServerTestCase(testcases.LiveServerTestCase): ...
+class APITransactionTestCase(testcases.TransactionTestCase):
+    client: APIClient
+
+class APITestCase(testcases.TestCase):
+    client: APIClient
+
+class APISimpleTestCase(testcases.SimpleTestCase):
+    client: APIClient
+
+class APILiveServerTestCase(testcases.LiveServerTestCase):
+    client: APIClient
+
 class URLPatternsTestCase(testcases.SimpleTestCase): ...