From c0b6350566f7ce29dca49cd903a418d0cc628b92 Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: May 22 2015 08:49:22 +0000
Subject: Add method to insert a few tokens in the database for the tests


---

diff --git a/tests/__init__.py b/tests/__init__.py
index 4342e5f..6e3df89 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -227,6 +227,35 @@ def create_projects_git(folder, bare=False):
     return repos
 
 
+def create_tokens(session, user_id=1):
+    """ Create some tokens for the project in the database. """
+    item = pagure.lib.model.Token(
+        id='aaabbbcccddd',
+        user_id=user_id,
+        project_id=1,
+        expiration=datetime.utcnow() + timedelta(days=30)
+    )
+    session.add(item)
+
+    item = pagure.lib.model.Token(
+        id='foo_token',
+        user_id=user_id,
+        project_id=1,
+        expiration=datetime.utcnow() + timedelta(days=30)
+    )
+    session.add(item)
+
+    item = pagure.lib.model.Token(
+        id='expired_token',
+        user_id=user_id,
+        project_id=1,
+        expiration=datetime.utcnow() - timedelta(days=1)
+    )
+    session.add(item)
+
+    session.commit()
+
+
 def add_content_git_repo(folder):
     """ Create some content for the specified git repo. """
     if not os.path.exists(folder):