test_client.py 文件源码

def test_request(self, api):
        with requests_mock.mock() as m:
            m.get('http://localhost:11000/oozie/v2/endpoint', text='{"result": "pass"}')
            result = api._request('GET', 'endpoint', None, None)
            assert result['result'] == 'pass'

        with requests_mock.mock() as m:
            m.get('http://localhost:11000/oozie/v2/endpoint')
            result = api._request('GET', 'endpoint', None, None)
            assert result is None

        with requests_mock.mock() as m:
            m.get('http://localhost:11000/oozie/v2/endpoint', text='>>> fail <<<')
            with pytest.raises(exceptions.OozieException) as err:
                api._request('GET', 'endpoint', None, None)
            assert 'Invalid response from Oozie server' in str(err)