class T07_ValidatePR(unittest.TestCase): """ PR validation methods """ def setUp(self): unittest.TestCase.setUp(self) server = gnats.Server('somehost') conn = FakeServerConnectionForDB(server) self.db = Database(server, 'testdb', conn) self.db._validate = self.my_validate self.pr_in = '' self.validate_in = '' def my_validate(self, pr, validate): self.pr_in = pr self.validate_in = validate return 1 def test_01_validate_fields(self): """ validate_fields() """ self.db.validate_fields('pr', change_reasons=False) self.assertEqual(self.pr_in, 'pr') self.assertEqual(self.validate_in, 'fields') def test_02_validate_fields_cr(self): """ validate_fields() with change-reasons """ self.db.validate_fields('pr', change_reasons=True) self.assertEqual(self.pr_in, 'pr') self.assertEqual(self.validate_in, 'fields-cr') def test_03_validate_pr(self): """ validate_pr() """ self.db.validate_pr('pr') self.assertEqual(self.pr_in, 'pr') self.assertEqual(self.validate_in, 'all') def test_04_validate_initial(self): """ validate_initial() """ self.db.validate_initial('pr') self.assertEqual(self.pr_in, 'pr') self.assertEqual(self.validate_in, 'initial')