Coverage for product_risk_suite/threat_model/shared_models.py: 100%

1from django.db import models 

2 

3class ThreatModelConnectionName(models.Model): 

4 id = models.AutoField(primary_key=True) 

5 product = models.ForeignKey('product.Product', on_delete=models.CASCADE) 

6 tech_name = models.CharField(max_length=255) 

7 human_name = models.CharField(max_length=255, blank=True, null=True) 

8 

9 def __str__(self): 

10 if self.human_name: 

11 return f"{self.product.title}: {self.human_name} ({self.tech_name})" 

12 return f"{self.product.title}: {self.tech_name}"