from dataclasses import dataclass

import pytest

from src.backend.connectors.youtube_apis import models


class TestBaseObject:
    @dataclass
    class BaseObjectImplementation(models._BaseModel):
        field_a: str
        field_b: str

    @pytest.fixture(scope="class")
    def instance(self):
        return self.BaseObjectImplementation("a", "b")

    def test_fields(self, instance):
        assert instance.fields() == ["field_a", "field_b"]

    def test_json(self, instance):
        assert instance.json() == {"field_a": "a", "field_b": "b"}