def test_get_branch_after_adding_one(self):

        self.imc.add(FileNode('docs/index.txt',

            content='Documentation\n'))

        self.imc.commit(

            message=u'New branch: foobar',

            author=u'joe',

            branch='foobar',

        )

        self.assertEqual(self.repo.workdir.get_branch(), self.default_branch)

    def test_get_changeset(self):

        old_head = self.repo.get_changeset()

        self.imc.add(FileNode('docs/index.txt',

            content='Documentation\n'))

        head = self.imc.commit(

            message=u'New branch: foobar',

            author=u'joe',

            branch='foobar',

        )

        self.assertEqual(self.repo.workdir.get_branch(), self.default_branch)

        self.repo.workdir.checkout_branch('foobar')

        self.assertEqual(self.repo.workdir.get_changeset(), head)

        self.repo.workdir.checkout_branch(self.default_branch)

        self.assertEqual(self.repo.workdir.get_changeset(), old_head)

    def test_checkout_branch(self):

        from kallithea.lib.vcs.exceptions import BranchDoesNotExistError

        # first, 'foobranch' does not exist.

        self.assertRaises(BranchDoesNotExistError, self.repo.workdir.checkout_branch,

                          branch='foobranch')

        # create new branch 'foobranch'.

        self.imc.add(FileNode('file1', content='blah'))

        self.imc.commit(message=u'asd', author=u'john', branch='foobranch')

        # go back to the default branch

        self.repo.workdir.checkout_branch()

        self.assertEqual(self.repo.workdir.get_branch(), self.backend_class.DEFAULT_BRANCH_NAME)

        # checkout 'foobranch'

        self.repo.workdir.checkout_branch('foobranch')

        self.assertEqual(self.repo.workdir.get_branch(), 'foobranch')

# For each backend create test case class

for alias in SCM_TESTS:

    attrs = {

        'backend_alias': alias,

    }