import unittest
from unittest import mock

from blender_asset_tracer.blendfile import dna, dna_io


class StringTest(unittest.TestCase):
    def test_trim_utf8(self):
        fileobj = mock.Mock()
        # Sinhala for 'beer'. This is exactly 15 bytes in UTF-8, so the last
        # character won't fit in the field (due to the 0-byte required).
        dna_io.BigEndianTypes.write_string(fileobj, 'බියර්', 15)

        expect_bytes = ('බියර්'[:-1]).encode('utf8') + b'\0'
        fileobj.write.assert_called_with(expect_bytes)

    def test_utf8(self):
        fileobj = mock.Mock()
        # Sinhala for 'beer'. This is exactly 15 bytes in UTF-8,
        # so with the 0-byte it just fits.
        dna_io.BigEndianTypes.write_string(fileobj, 'බියර්', 16)

        expect_bytes = 'බියර්'.encode('utf8') + b'\0'
        fileobj.write.assert_called_with(expect_bytes)