def test_get_product_and_vulnerability_map_by_relationship(self):
products = self.createProducts(name='RD64')
vuls = self.createVuls(code='LEN-2015-07')
pvr = ProdAndVulRelationship(product=products[0], vul=vuls[0])
pvr.save()
pvrs = ProdAndVulRelationship.objects.all()
self.assertEqual(len(pvrs), 1)
self.assertEqual(pvrs[0].product.name, 'RD640')
self.assertEqual(pvrs[0].vul.lenovoCode, 'LEN-2015-070')
def setupPVRelationship(self, products, vuls):
for product in products:
for vul in vuls:
pvr = ProdAndVulRelationship(product=product, vul=vul, status=1, fixedVersion='1.0.0', downloadLink='https://www.lenovo.com')
pvr.save()
