testresources_quickstart_resource_lifecycle_management_unittest.py

1import unittest
2import testresources
3
4class MyResource(testresources.TestResource):
5    def make(self, dependency_resources):
6        # This is where you would create your expensive resource
7        # (e.g., a database connection, a temporary directory, etc.)
8        return "Resource Data"
9
10    def clean(self, resource):
11        # This is where you would tear down your resource
12        pass
13
14class MyTest(unittest.TestCase, testresources.ResourcedTestCase):
15    # Define the resources this test case requires
16    resources = [('my_res', MyResource())]
17
18    def test_example(self):
19        # The resource is automatically created and attached to the instance
20        self.assertEqual(self.my_res, "Resource Data")
21
22if __name__ == '__main__':
23    unittest.main()