def test_get_user(self): self.assertEqual(self.kba.get_user(self.token1), self.user1) self.assertEqual(self.kba.get_user(self.token2), self.user2) # test getting from cache self.assertEqual(self.kba.get_user(self.token1), self.user1) self.assertEqual(self.kba.get_user(self.token2), self.user2) # test default url kba2 = KBaseAuth() self.assertEqual(kba2.get_user(self.token1), self.user1) self.assertEqual(kba2.get_user(self.token2), self.user2)
def test_bad_url(self): kba2 = KBaseAuth('https://thisisasuperfakeurlihope.com') with self.assertRaises(ConnectionError) as context: kba2.get_user(self.token1) self.assertIn('not known', str(context.exception.message))