def test_planset_exception(self):
self.run_api_call.side_effect = MockException("DEADBEEF")
with self.assertRaises(api_interface.ManipulateUserFailed) as cm:
res = api_interface.set_user_plan(sentinel.api_root,
self.test_user)
the_exception = cm.exception
self.assertEqual(str(the_exception), "DEADBEEF")
def test_planset_exception(self):
self.run_api_call.side_effect = MockException("DEADBEEF")
with self.assertRaises(api_interface.ManipulateUserFailed) as cm:
res = api_interface.set_user_plan(sentinel.api_root,
self.test_user)
the_exception = cm.exception
self.assertEqual(str(the_exception), "DEADBEEF")
def test_planset_success_promo(self):
ret_val = {'success': True}
self.run_api_call.return_value = ret_val
self.assertEqual(
api_interface.set_user_plan(sentinel.api_root, self.test_user,
sentinel.promo_code), ret_val)
args, _ = self.run_api_call.call_args
assert len(args) == 3
self.assertIs(args[2]['promo_code'], sentinel.promo_code)
def test_planset_failure(self):
ret_val = {'success' : False,
'reason' : "DEADBEEF",
}
self.run_api_call.return_value = ret_val
with self.assertRaises(api_interface.ManipulateUserFailed) as cm:
res = api_interface.set_user_plan(sentinel.api_root,
self.test_user)
the_exception = cm.exception
self.assertEqual(str(the_exception), "DEADBEEF")
def test_planset_success_promo(self):
ret_val = {'success': True}
self.run_api_call.return_value = ret_val
self.assertEqual(api_interface.set_user_plan(sentinel.api_root,
self.test_user,
sentinel.promo_code),
ret_val)
args, _ = self.run_api_call.call_args
assert len(args) == 3
self.assertIs(args[2]['promo_code'], sentinel.promo_code)
def test_planset_failure(self):
ret_val = {
'success': False,
'reason': "DEADBEEF",
}
self.run_api_call.return_value = ret_val
with self.assertRaises(api_interface.ManipulateUserFailed) as cm:
res = api_interface.set_user_plan(sentinel.api_root,
self.test_user)
the_exception = cm.exception
self.assertEqual(str(the_exception), "DEADBEEF")
