From 8e6cd77aa7d1168459e12f098c70bbb53365bff4 Mon Sep 17 00:00:00 2001 From: msoulier Date: Wed, 25 Oct 2006 01:41:03 +0000 Subject: [PATCH] Added testcase for TftpPacketFactory. git-svn-id: https://tftpy.svn.sourceforge.net/svnroot/tftpy/trunk@29 63283fd4-ec1e-0410-9879-cb7f675518da --- t/test.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/t/test.py b/t/test.py index cc990d3..2e730e5 100644 --- a/t/test.py +++ b/t/test.py @@ -116,6 +116,24 @@ class TestTftpy(unittest.TestCase): self.assertEqual(oack.options['blksize'], '4096', "OACK blksize option is correct") + + def testTftpPacketFactory(self): + log.debug("===> Running testcase testTftpPacketFactory") + # Make sure that the correct class is created for the correct opcode. + classes = { + 1: tftpy.TftpPacketRRQ, + 2: tftpy.TftpPacketWRQ, + 3: tftpy.TftpPacketDAT, + 4: tftpy.TftpPacketACK, + 5: tftpy.TftpPacketERR, + 6: tftpy.TftpPacketOACK + } + factory = tftpy.TftpPacketFactory() + for opcode in classes: + self.assert_(isinstance(factory.create(opcode), + classes[opcode]), + "opcode %d returns the correct class" % opcode) + if __name__ == '__main__': unittest.main()