"""
open/DurusWorks/durus/test/utest_serialize.py

"""
from durus.connection import ROOT_DURUS_ID
from durus.persistent import Persistent, ConnectionBase
from durus.serialize import ObjectWriter, ObjectReader, pack_record
from durus.serialize import unpack_record, split_durus_ids
from durus.utils import join_bytes, as_bytes
from sancho.utest import UTest, raises


class Test (UTest):

    def check_object_writer(self):
        class FakeConnection(ConnectionBase):
            def new_durus_id(self):
                return ROOT_DURUS_ID
            def note_access(self, obj):
                pass
        connection = FakeConnection()
        self.s=s=ObjectWriter(connection)
        x = Persistent()
        assert x._p_connection == None
        x._p_durus_id = ROOT_DURUS_ID
        x._p_connection = connection
        assert s._persistent_id(x) == (ROOT_DURUS_ID, Persistent)
        x._p_connection = FakeConnection()
        # connection of x no longer matches connection of s.
        raises(ValueError, s._persistent_id, x)
        x.a = Persistent()
        assert s.get_state(x), (
            '\x80\x02cdurus.persistent\nPersistent\nq\x01.\x80\x02}q\x02U'
            '\x01aU\x08\x00\x00\x00\x00\x00\x00\x00\x00q\x03h\x01\x86Qs.',
            '\x00\x00\x00\x00\x00\x00\x00\x00')
        assert list(s.gen_new_objects(x)) == [x, x.a]
        # gen_new_objects() can only be called once.
        raises(RuntimeError, s.gen_new_objects, 3)
        s.close()

    def check_object_reader(self):
        class FakeConnection:
            pass
        self.r = r = ObjectReader(FakeConnection())
        root = ('\x80\x02cdurus.persistent_dict\nPersistentDict\nq\x01.'
            '\x80\x02}q\x02U\x04dataq\x03}q\x04s.\x00\x00\x00\x00')
        root = as_bytes(root)
        assert r.get_ghost(root)._p_is_ghost()

    def check_record_pack_unpack(self):
        durus_id = as_bytes('0'*8)
        data = as_bytes('sample')
        reflist = ['1'*8, '2'*8]
        reflist =  list(map(as_bytes, reflist))
        refs = join_bytes(reflist)
        result=unpack_record(pack_record(durus_id, data, refs))
        assert result[0] == durus_id
        assert result[1] == data
        assert split_durus_ids(result[2]) == reflist
        assert split_durus_ids('') == []

if __name__ == "__main__":
    Test()