From daeb6f8d402561bddc52ce5a9df19c7441836eba Mon Sep 17 00:00:00 2001
From: Jeff Balogh <jbalogh@mozilla.com>
Date: Tue, 28 Dec 2010 15:10:35 -0500
Subject: [PATCH] detect beta files (bug 619157)

---
 apps/files/fixtures/files/beta-extension.xpi | Bin 0 -> 674 bytes
 apps/files/models.py                         |   2 ++
 apps/files/tests.py                          |   6 ++++++
 3 files changed, 8 insertions(+)
 create mode 100644 apps/files/fixtures/files/beta-extension.xpi

diff --git a/apps/files/fixtures/files/beta-extension.xpi b/apps/files/fixtures/files/beta-extension.xpi
new file mode 100644
index 0000000000000000000000000000000000000000..aeabd7b7db5e5d9b78325e3bf40b4b69ab3b4821
GIT binary patch
literal 674
zcmWIWW@Zs#U|`^2&@G>1Tj;cG#WzL<24fZm25tr!hRnR;lEj=Gy`q$~&=5`rW>rTi
zUl1;>;AUWC0jUIQZVidezHJ~<dw%_eDSJ<S3ZH!Iie~QN1OqPZk_~$`KW)`b@^lM&
zwPf-C+OF9<qGy^nAC!N7r}+HSnAq6&sonDqL@W$#T&!^9`s$lk5B~~&m;a<*Yj(3v
z!Q+_e!MDEGDtz6X5iOPxeA$90m}}CEO|u?LTE(y5^LzKi>lvJlOzD=d6&^%g`n`Mq
z<UMcXvucevTqCDlG+0(xvqZ2upzm$xrs6%HMBCb~C+rlQ``Pcp$GN#*&WYYQ{Q2i`
z`TqFZxqoKgy3J}Np4wQ?w4Z14hA1Jq_MRo(Ta<39u?VT^Bub{7eqvNRbCp8tud;bb
zkCt1oCjS<k#`aL{B~OCO|L2M6A$3yKdQ(c*<aVZbg(Tdyw>>KAz3_~l#|#(mNuOAi
zb7mG51*%R+2|Tk{Yxb8VGg_F|FBFY=G1ql_RKBWlw6&{5u?TyhKx?<JQR@@78JCwG
zxF5OrYr7|Z%<ALq`mE1-KXQJWXpwWyH(<AD*R<#F?5ba^UC|sB5`6X4_J?9W+KM-b
z$$mMvV58bAX@dv3GW!=tDKGQd!un!Dwu6$qi~W~$jfs~$jGo(UxVX1Iq`mRt26c{Q
zPx`GdY?NYiUf$CGR=zx`YeoCCw@(!I6tkbYD6Dtm{jEgpnq@Y(4TGC^zqDF^<XD2_
zhI^Ln<-b49a`~m#_0MI&a^*zR->d=Nj7)OOxDt;9Fi|lu08<FVl12~<B`L8&k`h`n
U3h-uS11V($!ayKx%>?2B07fkvIsgCw

literal 0
HcmV?d00001

diff --git a/apps/files/models.py b/apps/files/models.py
index cf807430ae8..378f581f2a0 100644
--- 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
index 34c5e90f213..89aaf4924ee 100644
--- 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):