def test_create_bad_input(self): factory = PacketFactory() with self.assertRaises(InvalidOpcodeError): factory.create('blahblahblah'.encode()) with self.assertRaises(OpcodeExtractError): factory.create('b'.encode())
def test_create_good_input(self): factory = PacketFactory() p = WRQPacket('file'.encode(), 'netascii'.encode()) self.assertTrue(isinstance(factory.create(p.to_wire()), WRQPacket)) p = DataPacket(4, 'data'.encode()) self.assertTrue(isinstance(factory.create(p.to_wire()), DataPacket)) p = ACKPacket(12) self.assertTrue(isinstance(factory.create(p.to_wire()), ACKPacket)) p = ErrorPacket(1) self.assertTrue(isinstance(factory.create(p.to_wire()), ErrorPacket))