detect beta files (bug 619157)

jbalogh · Dec 28, 2010 · daeb6f8 · daeb6f8
1 parent 72973bd
commit daeb6f8
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/apps/files/fixtures/files/beta-extension.xpi b/apps/files/fixtures/files/beta-extension.xpi
diff --git a/apps/files/models.py b/apps/files/models.py
@@ -91,6 +91,8 @@ def from_upload(cls, upload, version, platform, parse_data={}):
         f.jetpack = cls.is_jetpack(upload.path)
         f.hash = upload.hash
         f.no_restart = parse_data.get('no_restart', False)
+        if amo.VERSION_BETA.search(parse_data.get('version', '')):
+            f.status = amo.STATUS_BETA
         f.save()
         log.debug('New file: %r from %r' % (f, upload))
         # Move the uploaded file from the temp location.

diff --git a/apps/files/tests.py b/apps/files/tests.py
@@ -387,6 +387,12 @@ def test_size_small(self):
         f = File.from_upload(upload, self.version, self.platform)
         eq_(f.size, 1)
 
+    def test_beta_version(self):
+        upload = self.upload('beta-extension')
+        data = parse_addon(upload.path)
+        f = File.from_upload(upload, self.version, self.platform, data)
+        eq_(f.status, amo.STATUS_BETA)
+
 
 class TestZip(test_utils.TestCase):