from bulkperformancerights.logic import track


def test_json_for_vendor(monkeypatch):
    def track_mock_method(arg1, arg2, arg3):
        return [{'foo': 'bar'}, {'foo': 'bar2'}]
    monkeypatch.setattr(
        'bulkperformancerights.models.track.tracks_for_template',
        track_mock_method)
    json_for_vendor, row_count = track.tracks_line_json('vendor', 10)
    assert json_for_vendor == '{"foo": "bar"}\n{"foo": "bar2"}'
    assert row_count == 2


def test_json_for_vendor_none(monkeypatch):
    def track_mock_method(arg1, arg2, arg3):
        return []
    monkeypatch.setattr(
        'bulkperformancerights.models.track.tracks_for_template',
        track_mock_method)
    json_for_vendor = track.tracks_line_json('vendor', 10)
    assert json_for_vendor is None