storage: test

circle/storage/tests/test_models.py

@@ -2,7 +2,7 @@ from datetime import timedelta
 
 
 from django.test import TestCase
 from django.test import TestCase
 from django.utils import timezone
 from django.utils import timezone
-from mock import MagicMock
+from mock import MagicMock, Mock
 
 
 from ..models import Disk, DataStore
 from ..models import Disk, DataStore
 
 
@@ -81,3 +81,8 @@ class DiskTestCase(TestCase):
         d.ready = False
         d.ready = False
         with self.assertRaises(MockException):
         with self.assertRaises(MockException):
             Disk.save_as(d)
             Disk.save_as(d)
+
+    def test_download_percentage_no_download(self):
+        d = MagicMock(spec=Disk)
+        d.is_downloading = Mock(return_value=False)
+        assert Disk.get_download_percentage(d) is None