def test_from_uri_base64_header_empty_data(self) -> None:
e = DataMemoryElement.from_uri('base64://')
self.assertIsInstance(e, DataMemoryElement)
# no base64 data, which should decode to no bytes
self.assertEqual(e.get_bytes(), b"")
def test_from_uri_empty_string(self) -> None:
# Should return an element with no byte data
e = DataMemoryElement.from_uri('')
self.assertIsInstance(e, DataMemoryElement)
# no base64 data, which should decode to no bytes
self.assertEqual(e.get_bytes(), b"")
def test_from_uri_data_format_empty_data(self) -> None:
e = DataMemoryElement.from_uri('data:text/plain;base64,')
self.assertIsInstance(e, DataMemoryElement)
# no base64 data, which should decode to no bytes
self.assertEqual(e.get_bytes(), b"")
self.assertEqual(e.content_type(), 'text/plain')
def test_from_uri_data_format(self) -> None:
e = DataMemoryElement.from_uri(self.VALID_DATA_URI)
self.assertIsInstance(e, DataMemoryElement)
self.assertEqual(e.get_bytes(), self.EXPECTED_BYTES)
self.assertEqual(e.content_type(), self.EXPECTED_CT)
def test_from_uri_base64_header(self) -> None:
e = DataMemoryElement.from_uri(self.VALID_B64_URI)
self.assertIsInstance(e, DataMemoryElement)
self.assertEqual(e.get_bytes(), self.EXPECTED_BYTES)
# No content type info in base64 format
self.assertEqual(e.content_type(), None)