Advertisement
lenkaseg

final setup_for_my_test

Feb 14th, 2019
114
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.91 KB | None | 0 0
  1. def setup_for_my_test(session, directory):
  2.     # create parent project
  3.     parent = pagure.lib.model.Project(
  4.         user_id=1,  # pingou
  5.         name='vegetable',
  6.         is_fork=False,
  7.         parent_id=None,
  8.         description='vegetable parent',
  9.         hook_token='aaabbbddd' + '',
  10.     )
  11.     parent.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate']
  12.     session.add(parent)
  13.     session.flush()
  14.     session.refresh(parent)
  15.     create_locks(session, parent)
  16.     session.commit()
  17.  
  18.     id_of_parent = pagure.lib.model.Project
  19.     # create fork
  20.     fork = pagure.lib.model.Project(
  21.         user_id=1,  # pingou
  22.         name='vegetable',
  23.         is_fork=True,
  24.         parent_id=parent.id,
  25.         description='vegetable fork',
  26.         hook_token='aaabbbccc' + '',
  27.     )
  28.     fork.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate']
  29.     session.add(fork)
  30.     session.flush()
  31.     create_locks(session, fork)
  32.  
  33.     session.commit()
  34.  
  35.     # create projects git
  36.     repos = []
  37.     for project in ['vegetable.git', os.path.join('forks', 'pingou', 'vegetable.git')]:
  38.         for repo in ['repos', 'requests']:
  39.             repo_path = os.path.join(directory, repo, project)
  40.             repos.append(repo_path)
  41.  
  42.     for repo_path in repos:
  43.         if not os.path.exists(repo_path):
  44.             os.makedirs(repo_path)
  45.         pygit2.init_repository(repo_path, bare=True)
  46.  
  47.     #create tokens
  48.     parent_token = pagure.lib.model.Token(
  49.         id='iamparenttoken',
  50.         user_id=parent.user_id,
  51.         project_id=parent.id,
  52.         expiration=datetime.utcnow() + timedelta(days=30)
  53.     )
  54.     session.add(parent_token)
  55.  
  56.     fork_token = pagure.lib.model.Token(
  57.         id='iamforktoken',
  58.         user_id=fork.user_id,
  59.         project_id=fork.id,
  60.         expiration=datetime.utcnow() + timedelta(days=30)
  61.     )
  62.  
  63.     session.add(fork_token)
  64.  
  65.     session.commit()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement