fix sleep/sleepsegment __str__

orcasgit · Dec 18, 2014 · 285db38 · 285db38
1 parent fe26ec0
commit 285db38
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/misfitapp/models.py b/misfitapp/models.py
@@ -109,7 +109,7 @@ class Sleep(models.Model):
     duration = models.IntegerField()
 
     def __str__(self):
-        '%s %s' % (start_time, duration)
+        return '%s %s' % (self.start_time, self.duration)
 
 
 @python_2_unicode_compatible
@@ -124,4 +124,4 @@ class SleepSegment(models.Model):
     sleep_type = models.SmallIntegerField(choices=SLEEP_TYPES)
 
     def __str__(self):
-        '%s %s' % (time, sleep_type)
+        return '%s %s' % (self.time, self.sleep_type)
diff --git a/misfitapp/tests/test_models.py b/misfitapp/tests/test_models.py
@@ -101,10 +101,13 @@ def test_sleep(self):
         }
         sleep = Sleep(**data)
         sleep.save()
+        self.assertEqual(
+            '%s' % sleep, '%s %s' % (sleep.start_time, sleep.duration))
 
         seg_data = {'sleep': sleep,
                     'time': data['start_time'],
                     'sleep_type': SleepSegment.SLEEP,
         }
         seg = SleepSegment(**seg_data)
         seg.save()
+        self.assertEqual('%s' % seg, '%s %s' % (seg.time, seg.sleep_type))