annotate tests/testProductIdRename.py @ 22:4004b67216a9

Add tests + method to change a ProductID
author Benoît Allard <benoit.allard@greenbone.net>
date Fri, 24 Oct 2014 16:16:50 +0200
parents
children e3ee542a89be
rev   line source
22
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
1 import unittest
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
2
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
3 from datetime import datetime
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
4
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
5 from farolluz.cvrf import CVRF, CVRFPublisher, CVRFTracking, CVRFTrackingID, CVRFRevision, CVRFFullProductName, CVRFVulnerability, CVRFProductStatus, CVRFRelationship
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
6
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
7 class TestProductIdRename(unittest.TestCase):
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
8
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
9 def setUp(self):
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
10 self.doc = CVRF('title', 'type')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
11 self.doc.setPublisher(CVRFPublisher('Other'))
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
12 initial = datetime.now()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
13 current = datetime.now()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
14 track = CVRFTracking(CVRFTrackingID('1234'), 'Draft', (0,0), initial, current)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
15 track.addRevision(CVRFRevision((0,0), current, '1st'))
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
16 self.doc.setTracking(track)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
17 self.doc.validate()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
18
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
19
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
20 def testChangeProductId(self):
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
21 ptree = self.doc.createProductTree()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
22 prod = CVRFFullProductName('1', 'a', ptree)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
23 ptree.addProduct(prod)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
24 vuln = CVRFVulnerability(1)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
25 st = CVRFProductStatus('Fixed')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
26 st.addProductID('1')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
27 vuln.addProductStatus(st)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
28 self.doc.addVulnerability(vuln)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
29 self.doc.validate()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
30 prod._productid = '2'
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
31 self.doc.changeProductID('1', '2')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
32 self.doc.validate()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
33
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
34 def testChangeProductIdRelation(self):
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
35 ptree = self.doc.createProductTree()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
36 prod1 = CVRFFullProductName('1', 'a', ptree)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
37 ptree.addProduct(prod1)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
38 prod2 = CVRFFullProductName('2', 'b', ptree)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
39 ptree.addProduct(prod2)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
40 rel = CVRFRelationship('1', 'Installed On', '2')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
41 ptree.addRelationship(rel)
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
42 self.doc.validate()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
43 prod1._productid = '3'
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
44 self.doc.changeProductID('1', '3')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
45 self.doc.validate()
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
46 prod2._productid = '1'
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
47 self.doc.changeProductID('2', '1')
4004b67216a9 Add tests + method to change a ProductID
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
48 self.doc.validate()
This site is hosted by Intevation GmbH (Datenschutzerklärung und Impressum | Privacy Policy and Imprint)