def test_over_commit(self):
e = BaseEncoder()
e.add_encoded_chunk(b'123456789')
self.assertEqual(e.fetch_chunk(), b'123456789')
with self.assertRaises(ValueError):
e.commit(10)
def test_add_chunk_single(self):
e = BaseEncoder()
e.add_encoded_chunk(b'123456789')
self.assertEqual(e.fetch_chunk(), b'123456789')
e.commit(9)
self.assertEqual(e.fetch_chunk(), None)
def test_commit_2(self):
e = BaseEncoder()
e.add_encoded_chunk(b'123456789')
self.assertEqual(e.fetch_chunk(1024), b'123456789')
e.commit(5)
self.assertEqual(e.fetch_chunk(1024), b'6789')
def test_add_empty(self):
e = BaseEncoder()
e.add_encoded_chunk(b'')
self.assertEqual(
e.fetch_chunk(),
b'',
)
self.assertEqual(
e.fetch_chunk(),
None,
)
def test_add_chunk_multiple_add(self):
e = BaseEncoder()
e.add_encoded_chunk(b'1234')
e.add_encoded_chunk(b'5678')
self.assertEqual(e.fetch_chunk(), b'1234')
e.commit(4)
self.assertEqual(e.fetch_chunk(), b'5678')
e.commit(4)
self.assertEqual(e.fetch_chunk(), None)
def test_commit(self):
e = BaseEncoder()
e.add_encoded_chunk(b'123456789')
self.assertEqual(e.fetch_chunk(4), b'1234')
e.commit(4)
self.assertEqual(e.fetch_chunk(2), b'56')
e.commit(2)
self.assertEqual(e.fetch_chunk(), b'789')
e.commit(3)
self.assertEqual(e.fetch_chunk(4), None)
def test_write_socket(self):
s = DummySocket()
e = BaseEncoder()
e.add_encoded_chunk(b'123456789')
s.prepare(5)
n = e.write_to_socket(s)
self.assertEqual(n, 5)
self.assertEqual(s.buff, b'12345')
s.prepare(5)
n = e.write_to_socket(s)
self.assertEqual(n, 4)
self.assertEqual(s.buff, b'123456789')
def test_add_chunk_type_error(self):
e = BaseEncoder()
with self.assertRaises(TypeError):
e.add_encoded_chunk('123456789')