from unittest.mock import patch

from bonobo.contrib.opendatasoft import OpenDataSoftAPI
from bonobo.util.objects import ValueHolder


class ResponseMock:
    def __init__(self, json_value):
        self.json_value = json_value
        self.count = 0

    def json(self):
        if self.count:
            return {}
        else:
            self.count += 1
            return {"records": self.json_value}


def test_read_from_opendatasoft_api():
    extract = OpenDataSoftAPI(dataset="test-a-set")
    with patch("requests.get", return_value=ResponseMock([{"fields": {"foo": "bar"}}, {"fields": {"foo": "zab"}}])):
        for line in extract("http://example.com/", ValueHolder(0)):
            assert "foo" in line