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