python测试驱动开发
来源:岁月联盟
时间:2012-07-16
import unittest
from main import Sample
class SampleTest(unittest.TestCase):
def setUp(self):
print "create a new Sample"
self._sample = Sample("b64e5843ca7db8199c405be565fa7f57")
def tearDown(self):
print "Destory the sample"
self._sample = None www.2cto.com
def test_GetVirusNameFromVT(self):
"this md5 has the VT info"
aSample = Sample("b64e5843ca7db8199c405be565fa7f57")
dict_virusName = aSample._GetVirusNameFromVT()
self.assertTrue(dict_virusName!=None)
def test_GetVirusNameFromVT2(self):
"this md5 has not the VT info"
aSample = Sample("2b666ffe98e465523e514d2b93b7666a")
dict_virusName = aSample._GetVirusNameFromVT ()
self.assertTrue(len(dict_virusName) == 0)
if __name__=="__main__":
#unittest.main()
suite = unittest.TestLoader().loadTestsFromTestCase(SampleTest)
unittest.TextTestRunner(verbosity=2).run(suite)
作者:xiaocaiju