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)