def test_remote_origin_read_session_started_rollover(self):
# if a rollover is done, we reach blocknum 0 again. But this time
# session is already started.
# Here we test the case where rollover has already happened
data_datagram = DATADatagram(1, self.test_data[:5])
data_datagram.opcode = OP_ACK
data_datagram.blocknum = 0
self.rs.session.block_size = 5
self.rs.startProtocol()
self.clock.pump((1, ) * 3)
self.assertEqual(self.rs._datagramReceived(data_datagram),
self.rs.session.datagramReceived(data_datagram))
self.addCleanup(self.rs.cancel)
def test_remote_origin_read_session_started_rollover(self):
# if a rollover is done, we reach blocknum 0 again. But this time
# session is already started.
# Here we test the case where rollover has already happened
data_datagram = DATADatagram(1, self.test_data[:5])
data_datagram.opcode = OP_ACK
data_datagram.blocknum = 0
self.rs.session.block_size = 5
self.rs.startProtocol()
self.clock.pump((1,)*3)
self.assertEqual(
self.rs._datagramReceived(data_datagram),
self.rs.session.datagramReceived(data_datagram))
self.addCleanup(self.rs.cancel)
def test_remote_origin_read_session_not_started_rollover(self):
# if a rollover is done, we reach blocknum 0 again. But this time
# session is already started.
# Here we test the case where rollover has not happened yet
data_datagram = DATADatagram(1, self.test_data[:5])
data_datagram.opcode = OP_ACK
data_datagram.blocknum = 0
self.rs.session.block_size = 5
self.clock.pump((1,)*3)
self.rs.session.transport = b"Fake"
self.rs.session.started = False
self.rs._datagramReceived(data_datagram)
self.assertEqual(self.rs.session.started, True)
self.assertEqual(self.rs.session.transport, self.rs.transport)
self.addCleanup(self.rs.cancel)
def test_remote_origin_read_session_not_started_rollover(self):
# if a rollover is done, we reach blocknum 0 again. But this time
# session is already started.
# Here we test the case where rollover has not happened yet
data_datagram = DATADatagram(1, self.test_data[:5])
data_datagram.opcode = OP_ACK
data_datagram.blocknum = 0
self.rs.session.block_size = 5
self.clock.pump((1, ) * 3)
self.rs.session.transport = b"Fake"
self.rs.session.started = False
self.rs._datagramReceived(data_datagram)
self.assertEqual(self.rs.session.started, True)
self.assertEqual(self.rs.session.transport, self.rs.transport)
self.addCleanup(self.rs.cancel)
