def test_is_empty_zero_bytes(self, _m_requests_get):
e = DataUrlElement('some-address')
# simulate no content bytes returned
e.get_bytes = mock.MagicMock(return_value='')
self.assertTrue(e.is_empty())
def test_new_from_internet(self):
e = DataUrlElement(self.EXAMPLE_URL)
self.assertEqual(e.get_bytes(), open(self.EXAMPLE_PTH, 'rb').read())
def test_new_add_missing_scheme(self):
# Construct without scheme header, should add http://
e = DataUrlElement(self.EXAMPLE_URL[8:])
self.assertEqual(e._url, 'http://' + self.EXAMPLE_URL[8:])
self.assertEqual(e.get_bytes(), open(self.EXAMPLE_PTH, 'rb').read())
def test_is_empty_nonzero_bytes(self, _m_requests_get):
e = DataUrlElement('some-address')
# simulate some content bytes returned
e.get_bytes = mock.MagicMock(return_value='some bytes returned')
self.assertFalse(e.is_empty())
def test_get_bytes_from_url(self):
e = DataUrlElement(self.EXAMPLE_URL)
self.assertEqual(e.get_bytes(), open(self.EXAMPLE_PTH, 'rb').read())
self.assertEqual(e.content_type(), 'image/png')
def test_new_add_missing_scheme(self):
# Construct without scheme header, should add http://
e = DataUrlElement(self.EXAMPLE_URL[8:])
self.assertEqual(e._url, 'http://' + self.EXAMPLE_URL[8:])
self.assertEqual(e.get_bytes(), open(self.EXAMPLE_PTH, 'rb').read())
def test_new_from_internet(self):
e = DataUrlElement(self.EXAMPLE_URL)
self.assertEqual(e.get_bytes(), open(self.EXAMPLE_PTH, 'rb').read())
def test_get_bytes_from_url(self):
e = DataUrlElement(self.EXAMPLE_URL)
self.assertEqual(e.get_bytes(), open(self.EXAMPLE_PTH, 'rb').read())
self.assertEqual(e.content_type(), 'image/png')
def test_is_empty_zero_bytes(self, m_requests_get):
e = DataUrlElement('some-address')
# simulate no content bytes returned
e.get_bytes = mock.MagicMock(return_value='')
ntools.assert_true(e.is_empty())
def test_new_from_internet(self):
e = DataUrlElement(self.EXAMPLE_URL)
ntools.assert_equal(e.get_bytes(), open(self.EXAMPLE_PTH).read())
