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()