How do I structure my tests with Python unittest module?

Posted by persepolis on Stack Overflow See other posts from Stack Overflow or by persepolis
Published on 2012-09-11T15:37:06Z Indexed on 2012/09/11 15:37 UTC
Read the original article Hit count: 182

Filed under:
|
|

I'm trying to build a test framework for automated webtesting in selenium and unittest, and I want to structure my tests into distinct scripts. So I've organised it as following:

base.py - This will contain, for now, the base selenium test case class for setting up a session.

import unittest
from selenium import webdriver

# Base Selenium Test class from which all test cases inherit.
class BaseSeleniumTest(unittest.TestCase):
    def setUp(self):
        self.browser = webdriver.Firefox()
    def tearDown(self):
        self.browser.close()

main.py - I want this to be the overall test suite from which all the individual tests are run.

import unittest
import test_example

if __name__ == "__main__":
    SeTestSuite = test_example.TitleSpelling()
    unittest.TextTestRunner(verbosity=2).run(SeTestSuite)

test_example.py - An example test case, it might be nice to make these run on their own too.

from base import BaseSeleniumTest

# Test the spelling of the title
class TitleSpelling(BaseSeleniumTest):
    def test_a(self):
        self.assertTrue(False)

    def test_b(self):
        self.assertTrue(True)

The problem is that when I run main.py I get the following error:

Traceback (most recent call last):
  File "H:\Python\testframework\main.py", line 5, in <module>
    SeTestSuite = test_example.TitleSpelling()
  File "C:\Python27\lib\unittest\case.py", line 191, in __init__
    (self.__class__, methodName))
ValueError: no such test method in <class 'test_example.TitleSpelling'>: runTest

I suspect this is due to the very special way in which unittest runs and I must have missed a trick on how the docs expect me to structure my tests. Any pointers?

© Stack Overflow or respective owner

Related posts about python

Related posts about unit-testing