cleanup and fix tests

founders4schools · Jan 28, 2016 · d6eefa0 · d6eefa0
1 parent 461b1ad
commit d6eefa0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 11 deletions.
diff --git a/duedil/resources/__init__.py b/duedil/resources/__init__.py
@@ -26,10 +26,6 @@
 from ..api import LiteClient, ProClient  # , InternationalClient
 
 
-class ReadOnlyException(Exception):
-    pass
-
-
 class Resource(Mapping):
     attribute_names = None
     locale = 'uk'

diff --git a/duedil/resources/pro/company/accounts/__init__.py b/duedil/resources/pro/company/accounts/__init__.py
@@ -6,10 +6,6 @@
 import sys
 
 
-class UnknownAccountTypeException(Exception):
-    pass
-
-
 class Account(RelatedResourceMixin, ProResource):
     'Abstraction of Accounts resource in duedil v3 pro api'
     attribute_names = [

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -25,7 +25,7 @@
 import requests_mock
 from requests.exceptions import HTTPError
 
-from duedil.api import LiteClient, ProClient, InternationalClient, Client
+from duedil.api import LiteClient, ProClient, InternationalClient, Client, APIMonthlyLimitException
 from duedil.resources.lite import Company as LiteCompany
 from duedil.search.pro import CompanySearchResult as ProCompanySearchResult, DirectorSearchResult
 from duedil.search.lite import CompanySearchResult as LiteCompanySearchResult
@@ -131,8 +131,9 @@ def test_throttling_quota(self, m):
                        {'status_code': 200, 'json': {'name': 'Duedil', 'id': '12345'}},
                       ])
         data = {'name': 'Duedil', 'id': '12345'}
-        response = client.get('12345')
-        self.assertEqual(response, {'name': 'Duedil', 'id': '12345'})
+        with self.assertRaises(APIMonthlyLimitException):
+            response = client.get('12345')
+            self.assertIsNone(response)
 
 
     def test_incorrect_query_params(self):