import pytest
from specklepy.api.models import Stream
from specklepy.objects import Base
from specklepy.objects.encoding import ObjectArray
from specklepy.serialization.base_object_serializer import BaseObjectSerializer
from specklepy.transports.sqlite import SQLiteTransport


class TestObject:
    @pytest.fixture(scope="module")
    def stream(self, client):
        stream = Stream(
            name="a sample stream for testing",
            description="a stream created for testing",
            isPublic=True,
        )
        stream.id = client.stream.create(
            stream.name, stream.description, stream.isPublic
        )
        return stream

    def test_object_create(self, client, stream, base):
        transport = SQLiteTransport()
        s = BaseObjectSerializer(write_transports=[transport], read_transport=transport)
        _, base_dict = s.traverse_base(base)
        obj_id = client.object.create(stream_id=stream.id, objects=[base_dict])[0]

        assert isinstance(obj_id, str)
        assert base_dict["@detach"]["speckle_type"] == "reference"
        assert obj_id == base.get_id(True)

    def test_object_get(self, client, stream, base):
        fetched_base = client.object.get(
            stream_id=stream.id, object_id=base.get_id(True)
        )

        assert isinstance(fetched_base, Base)
        assert fetched_base.name == base.name
        assert isinstance(fetched_base.vertices, list)
        # assert fetched_base["@detach"]["speckle_type"] == "reference"

    def test_object_array_decoder(self):
        array = ObjectArray()
        array.data = [5, 1, 1, 1, 1, 1, 4, 1, 1, 1, 1, 3, 1, 1, 1, 2, 1, 1, 1, 1]

        assert array.decode(decoder=sum) == [5, 4, 3, 2, 1]