Exemplo n.º 1
0
def test_request_headers(patch_base_class):
    stream = PinterestStream(config=MagicMock())
    inputs = {
        "stream_slice": None,
        "stream_state": None,
        "next_page_token": None
    }
    expected_headers = {}
    assert stream.request_headers(**inputs) == expected_headers
Exemplo n.º 2
0
def test_parse_response(patch_base_class, test_response,
                        test_current_stream_state):
    stream = PinterestStream(config=MagicMock())
    inputs = {
        "response": test_response,
        "stream_state": test_current_stream_state
    }
    expected_parsed_object = {}
    assert next(stream.parse_response(**inputs)) == expected_parsed_object
Exemplo n.º 3
0
def test_backoff_time(patch_base_class):
    response_mock = MagicMock()
    stream = PinterestStream(config=MagicMock())
    expected_backoff_time = None
    assert stream.backoff_time(response_mock) == expected_backoff_time
Exemplo n.º 4
0
def test_should_retry(patch_base_class, http_status, should_retry):
    response_mock = MagicMock()
    response_mock.status_code = http_status
    stream = PinterestStream(config=MagicMock())
    assert stream.should_retry(response_mock) == should_retry
Exemplo n.º 5
0
def test_http_method(patch_base_class):
    stream = PinterestStream(config=MagicMock())
    expected_method = "GET"
    assert stream.http_method == expected_method
Exemplo n.º 6
0
def test_next_page_token(patch_base_class, test_response):
    stream = PinterestStream(config=MagicMock())
    inputs = {"response": test_response}
    expected_token = {"bookmark": "string"}
    assert stream.next_page_token(**inputs) == expected_token