# -*- coding: utf-8 -*-

import unittest

from kallithea.tests.base import *

from kallithea.tests.fixture import Fixture

from kallithea.model.db import Repository

from kallithea.model.repo import RepoModel

from kallithea.model.user import UserModel

from kallithea.model.meta import Session

fixture = Fixture()

class _BaseTestCase(TestController):

    """

    Write all tests here

    """

    REPO = None

    REPO_TYPE = None

    NEW_REPO = None

    REPO_FORK = None

    def setup_method(self, method):

        self.username = u'forkuser'

        self.password = u'qweqwe'

        self.u1 = fixture.create_user(self.username, password=self.password,

                                      email=u'fork_king@example.com')

        Session().commit()

    def teardown_method(self, method):

        ## run the check page that triggers the flash message

        response = self.app.get(url('repo_check_home', repo_name=fork_name_full))

        #test if we have a message that fork is ok

        self.checkSessionFlash(response,

                'Forked repository %s as <a href="/%s">%s</a>'

                % (repo_name, fork_name_full, fork_name_full))

        #test if the fork was created in the database

        fork_repo = Session().query(Repository) \

            .filter(Repository.repo_name == fork_name_full).one()

        assert fork_repo.repo_name == fork_name_full

        assert fork_repo.fork.repo_name == repo_name

        # test if the repository is visible in the list ?

        response = self.app.get(url('summary_home', repo_name=fork_name_full))

        response.mustcontain(fork_name_full)

        response.mustcontain(self.REPO_TYPE)

        response.mustcontain('Fork of "<a href="/%s">%s</a>"' % (repo_name, repo_name))

        fixture.destroy_repo(fork_name_full)

        fixture.destroy_repo_group(group_id)

    def test_z_fork_create(self):

        self.log_user()

        fork_name = self.REPO_FORK

        description = 'fork of vcs test'

        repo_name = self.REPO

        org_repo = Repository.get_by_repo_name(repo_name)

        creation_args = {

            'repo_name': fork_name,

            'repo_group': u'-1',

            'fork_parent_id': org_repo.repo_id,

            'repo_type': self.REPO_TYPE,

            'description': description,

            'private': 'False',

            'landing_rev': 'rev:tip',

            '_authentication_token': self.authentication_token()}

        self.app.post(url(controller='forks', action='fork_create',

                          repo_name=repo_name), creation_args)

        repo = Repository.get_by_repo_name(self.REPO_FORK)

        assert repo.fork.repo_name == self.REPO

        ## run the check page that triggers the flash message

        response = self.app.get(url('repo_check_home', repo_name=fork_name))

        #test if we have a message that fork is ok

        self.checkSessionFlash(response,