diff --git a/tests/test_distgit.py b/tests/test_distgit.py index b3726c8..ee12cfc 100644 --- a/tests/test_distgit.py +++ b/tests/test_distgit.py @@ -158,3 +158,23 @@ class TestDistgitC6DistroBranch(unittest.TestCase): def test_target_gets_set(self): self.assertEqual(self.d.target, 'c6') + +class TestDistgitC6PlusDistroBranch(unittest.TestCase): + def setUp(self): + self.branchstring = 'c6-plus' + self.d = DistGitDirectory(self.branchstring) + + def test_signame_gets_set(self): + self.assertEqual(self.d.signame, 'centos') + + def test_centosversion_gets_set(self): + self.assertEqual(self.d.centosversion, '6') + + def test_projectname_gets_set(self): + self.assertEqual(self.d.projectname, 'plus') + + def test_releasename_gets_set(self): + self.assertEqual(self.d.releasename, None) + + def test_target_gets_set(self): + self.assertEqual(self.d.target, 'c6-plus')