Ejemplo n.º 1
0
def test_parse_response(patch_base_class, config):
    stream = ZenloopStream(**config)
    response = MagicMock()
    response.json.return_value = {"answers": [{"id": 123, "name": "John Doe"}]}
    inputs = {"response": response}
    expected_parsed_object = {"answers": [{"id": 123, "name": "John Doe"}]}
    assert next(stream.parse_response(**inputs)) == expected_parsed_object
Ejemplo n.º 2
0
def test_request_headers(patch_base_class, config):
    stream = ZenloopStream(**config)
    inputs = {
        "stream_slice": None,
        "stream_state": None,
        "next_page_token": None
    }
    expected_headers = {}
    assert stream.request_headers(**inputs) == expected_headers
Ejemplo n.º 3
0
def test_next_page_token(patch_base_class, config):
    stream = ZenloopStream(**config)
    inputs = {"response": MagicMock()}
    inputs["response"].json.return_value = {
        "meta": {
            "page": 1,
            "per_page": 12,
            "total": 8
        }
    }
    expected_token = None
    assert stream.next_page_token(**inputs) == expected_token
Ejemplo n.º 4
0
def test_backoff_time(patch_base_class, config):
    response_mock = MagicMock()
    stream = ZenloopStream(**config)
    expected_backoff_time = None
    assert stream.backoff_time(response_mock) == expected_backoff_time
Ejemplo n.º 5
0
def test_should_retry(patch_base_class, config, http_status, should_retry):
    response_mock = MagicMock()
    response_mock.status_code = http_status
    stream = ZenloopStream(**config)
    assert stream.should_retry(response_mock) == should_retry
Ejemplo n.º 6
0
def test_http_method(patch_base_class, config):
    stream = ZenloopStream(**config)
    expected_method = "GET"
    assert stream.http_method == expected_method