benoit@31: from datetime import datetime
benoit@31: 
benoit@31: from tests.utils import TestCase
benoit@31: from farolluz.cvrf import CVRFFullProductName, CVRFGroup, CVRFVulnerability, CVRFThreat
benoit@31: 
benoit@31: class TestGroupIdRename(TestCase):
benoit@31: 
benoit@31:     def testChangeGroupId(self):
benoit@31:         ptree = self.doc.createProductTree()
benoit@31:         prod1 = CVRFFullProductName('1', 'a', ptree)
benoit@31:         ptree.addProduct(prod1)
benoit@31:         prod2 = CVRFFullProductName('2', 'b', ptree)
benoit@31:         ptree.addProduct(prod2)
benoit@31:         grp = CVRFGroup('GRP-1')
benoit@31:         grp.addProductID('1')
benoit@31:         grp.addProductID('2')
benoit@31:         ptree.addGroup(grp)
benoit@31:         vuln = CVRFVulnerability(1)
benoit@31:         th = CVRFThreat('Impact', 'bad !')
benoit@31:         th.addGroupID('GRP-1')
benoit@31:         vuln.addThreat(th)
benoit@31:         self.doc.addVulnerability(vuln)
benoit@31:         self._validate()
benoit@31:         grp._productid = 'GRP-2'
benoit@31:         self.doc.changeProductID('GRP-1', 'GRP-2')
benoit@31:         self._validate()
benoit@31: