add relation between response and project

andrewsmedina · Jul 30, 2011 · 73fdb33 · 73fdb33
1 parent 8df2627
commit 73fdb33
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/startdust_website/responses/models.py b/startdust_website/responses/models.py
@@ -1,9 +1,11 @@
 from django.db import models
 from datetime import datetime
+from projects.models import Project
 
 
 class Response(models.Model):
 
     time = models.IntegerField()
     url = models.URLField(verify_exists=False)
     date = models.DateTimeField(default=datetime.now, editable=False)
+    project = models.ForeignKey(Project, null=True, editable=False) #will be change this
diff --git a/startdust_website/responses/tests.py b/startdust_website/responses/tests.py
@@ -22,6 +22,12 @@ def test_response_model_should_have_date_attribute(self):
         '''
         self.assertFieldIn('date', Response._meta.fields)
 
+    def test_response_should_be_related_with_project(self):
+        '''
+        Response should be related with Project
+        '''
+        self.assertFieldIn('project', Response._meta.fields)
+
     def assertFieldIn(self, expected_field, field_list):
         '''
         assert if field in a field list