Merge pull request #526 from awf-dbca/fixture-natural-keys

Added natural key to proposal types via manager which fixtures now use

mooringlicensing/components/proposals/models.py

@@ -146,17 +146,25 @@ class Meta:
         ('ria', 'By Rottnest Island Authority for a mooring allocated by the Authority'),
         )
 
+class ProposalTypeManager(models.Manager):
+    def get_by_natural_key(self, code):
+        return self.get(code=code)
 
 class ProposalType(models.Model):
     code = models.CharField(max_length=30, blank=True, null=True, unique=True)
     description = models.CharField(max_length=200, blank=True, null=True)
 
     def __str__(self):
         return self.description if self.description else ''
+
+    objects = ProposalTypeManager()
 
     class Meta:
         app_label = 'mooringlicensing'
 
+    def natural_key(self):
+        return (self.code,)
+
 
 class ProposalProofOfIdentityDocument(models.Model):
     proof_of_identity_document = models.ForeignKey('ProofOfIdentityDocument', on_delete=models.CASCADE)