Esempi in Python per TRequestsTransport.AuthError

Esempio n. 1

File: test_scheduler_client.py Progetto: theevocater/aurora

 def test_raise_auth_error(self):
   self.mock_thrift_client.killTasks(TaskQuery(), None, None).AndRaise(
       TRequestsTransport.AuthError())
   self.mock_scheduler_client.get_failed_auth_message().AndReturn('failed auth')
   self.mox.ReplayAll()
   with pytest.raises(scheduler_client.SchedulerProxy.AuthError):
     self.make_scheduler_proxy().killTasks(TaskQuery(), None, None)

Esempio n. 2

 def test_raise_auth_error(self):
     self.mock_thrift_client.killTasks(TaskQuery(), None, None,
                                       SESSION).AndRaise(
                                           TRequestsTransport.AuthError())
     self.mox.ReplayAll()
     with pytest.raises(scheduler_client.SchedulerProxy.AuthError):
         self.make_scheduler_proxy().killTasks(TaskQuery(), None, None)