pass fake state to Annotation ctor

cloverfeed · Oct 31, 2014 · b7e6155 · b7e6155
1 parent 32f31cd
commit b7e6155
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/models.py b/app/models.py
@@ -191,8 +191,12 @@ def generate(fake, doc, user):
         posy = fake.random_int(0, 600)
         width = fake.random_int(50, 300)
         height = fake.random_int(50, 300)
+        if fake.boolean():
+            state = Annotation.STATE_OPEN
+        else:
+            state = Annotation.STATE_CLOSED
         text = fake.text()
-        ann = Annotation(doc, page, posx, posy, width, height, text, user)
+        ann = Annotation(doc, page, posx, posy, width, height, text, user, state)
         return ann
 
     def is_closed(self):