Skip to content

Commit 0b163fb

Browse files
committed
Changed tests to not use self.assertRaises like a context manager (e.g. 'with self.assertRaises...') to enuser python 2.6 test compatibility. Added self.fail() to try expect statements for points that should be reached as an exception should occur.
1 parent 3bf84cf commit 0b163fb

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

tests/unit/test_user.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,7 @@ def test_find_by_email(self):
165165
self.assertEqual('Santiago', user.location_data.city_name)
166166
self.assertTrue(isinstance(user.location_data, LocationData))
167167
self.assertEqual('johnny', user.custom_data['nick'])
168-
with self.assertRaises(AttributeError):
169-
user.companies
168+
self.assertRaises(AttributeError, lambda: user.companies)
170169

171170
@patch('requests.request', create_response(200, 'get_user_id_valid.json'))
172171
def test_find_by_user_id(self):
@@ -206,6 +205,7 @@ def test_properties(self):
206205
try:
207206
# cannot set the relationship score
208207
user.relationship_score = 50
208+
self.fail()
209209
except AttributeError:
210210
pass
211211

@@ -218,30 +218,33 @@ def test_properties(self):
218218
self.assertEqual(1331764344,
219219
time.mktime(user.created_at.timetuple()))
220220
self.assertEqual('Ace', user.custom_data['name'])
221-
with self.assertRaises(AttributeError):
222-
user.companies
221+
self.assertRaises(AttributeError, lambda: user.companies)
223222

224223
def test_company(self):
225224
user = User()
226225
user.company = {
227226
'id': 1,
228227
'name':' Intercom',
229228
'created_at': datetime.fromtimestamp(1331764344)}
230-
with self.assertRaises(AttributeError):
231-
user.company
232-
with self.assertRaises(ValueError):
229+
self.assertRaises(AttributeError, lambda: user.company)
230+
try:
233231
user.company = ['foo']
232+
self.fail()
233+
except ValueError:
234+
pass
234235

235236
def test_companies(self):
236237
user = User()
237238
user.companies = [{
238239
'id': 1,
239240
'name':' Intercom',
240241
'created_at': datetime.fromtimestamp(1331764344)}]
241-
with self.assertRaises(AttributeError):
242-
user.companise
243-
with self.assertRaises(ValueError):
242+
self.assertRaises(AttributeError, lambda: user.companies)
243+
try:
244244
user.companies = {'foo':'bar'}
245+
self.fail()
246+
except ValueError:
247+
pass
245248

246249
class LocationDataTest(TestCase):
247250

0 commit comments

Comments
 (0)