measure file size in kilobytes (bug 620864)

jbalogh · Dec 28, 2010 · a2b4353 · a2b4353
1 parent 0d027f3
commit a2b4353
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/apps/files/models.py b/apps/files/models.py
@@ -87,7 +87,7 @@ def from_upload(cls, upload, version, platform, parse_data={}):
         f = cls(version=version, platform=platform)
         upload.path = path.path(nfd_str(upload.path))
         f.filename = f.generate_filename(extension=upload.path.ext)
-        f.size = upload.path.size
+        f.size = int(max(1, round(upload.path.size / 1024, 0)))  # Kilobytes.
         f.jetpack = cls.is_jetpack(upload.path)
         f.hash = upload.hash
         f.no_restart = parse_data.get('no_restart', False)

diff --git a/apps/files/tests.py b/apps/files/tests.py
@@ -372,6 +372,16 @@ def test_utf8(self):
         f = File.from_upload(upload, self.version, self.platform)
         eq_(f.filename, u'jéts-0.1-mac.xpi')
 
+    def test_size(self):
+        upload = self.upload('extension')
+        f = File.from_upload(upload, self.version, self.platform)
+        eq_(f.size, 2)
+
+    def test_size_small(self):
+        upload = self.upload('alt-rdf')
+        f = File.from_upload(upload, self.version, self.platform)
+        eq_(f.size, 1)
+
 
 class TestZip(test_utils.TestCase):