# Suites of tests should have classes named Test<SuiteName>.# For something to be run before each test, define a function named setUp.# For something to be run after each test, define a function named tearDown.# Individual units should have the name <unit>_test.from bandage.tests import *from bandage import modelclass TestUser(TestController): def setUp(self): self.user = model.User() model.meta.Session.add(self.user) model.meta.Session.commit() def existance_test(self): assert self.user.id != None def get_by_id_test(self): found_user = model.User.get_by_id(self.user.id) assert found_user == self.user def tearDown(self): model.meta.Session.delete(self.user) model.meta.Session.commit()class TestSubmission(TestController): def setUp(self): self.submission = model.Submission() self.submission.name = "test" model.meta.Session.add(self.submission) model.meta.Session.commit() def existance_test(self): assert self.submission._id != None def get_by_id_test(self): found_submission = model.Submission.get_by_id(self.submission._id) assert found_submission == self.submission def duplicate_test(self): self.submission2 = model.Submission() self.submission2.name = "test" model.meta.Session.add(self.submission2) model.meta.Session.commit() assert self.submission._id != self.submission2._id assert self.submission.name != self.submission2.name model.meta.Session.delete(self.submission2) model.meta.Session.commit() def tearDown(self): model.meta.Session.delete(self.submission) model.meta.Session.commit()