Skip to content

Instantly share code, notes, and snippets.

@allieus
Last active August 29, 2015 14:06
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save allieus/d249ecc257a848c323b6 to your computer and use it in GitHub Desktop.
Save allieus/d249ecc257a848c323b6 to your computer and use it in GitHub Desktop.
1) 명시한 순서대로 테스트가 수행되도록 함. 2) 모든 테스트를 수행한 후에 cleanup 수행토록 함.
import unittest
class SampleTestCase(unittest.TestCase):
def setUp(self):
pass
def setDriver(self, driver):
self.driver = driver
def test_default_size(self):
print '[1] test default size', self.driver
self.assertEqual(1, 1)
def test_z(self):
print '[2] test z', self.driver
def test_resize(self):
print '[3] test resize', self.driver
self.assertEqual(2, 2)
class TestRunner(unittest.TextTestRunner):
def run(self, test, cleanup=None):
super(TestRunner, self).run(test)
if callable(cleanup):
cleanup()
def main():
method_names = ['test_default_size', 'test_z', 'test_resize']
driver = 'this is driver' # FIXME
test_cases = []
for method_name in method_names:
test_case = SampleTestCase(method_name)
test_case.setDriver(driver)
test_cases.append(test_case)
runner = TestRunner()
def cleanup():
print 'driver, cleanup'
driver = None # FIXME
runner.run(unittest.TestSuite(test_cases), cleanup)
if __name__ == '__main__':
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment