test_GithubClass.py

#!python

import unittest
import nose
import github
import github_class
import logging
import logging.config
import time
import github_token

logging.disable(logging.CRITICAL)


class TestGithubClassMethods(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.g = github_class.GithubClass(github_token.token)
        cls.test_repos = ["repo1", "repo2", "repo3"]
        for test_repo_name in cls.test_repos:
            if test_repo_name not in cls.g.get_names_of_repos():
                cls.g.create_repo(test_repo_name)
        cls.repo = cls.g.get_repo("repo1")
        if "branch1" not in cls.g.get_names_of_branches(cls.repo.name):
            cls.g.create_branch(cls.repo, "branch1")
        new_file = "/requirements/text.txt"
        new_message = "text for test"
        cls.branch_name = "branch1"
        new_content = 'a==1234\nb>2234,<=2235\nc>3234'
        cls.g.create_file(
            cls.repo, path=new_file, message=new_message, content=new_content, branch=cls.branch_name)

    @classmethod
    def tearDownClass(cls):
        time.sleep(10)
        for repo in cls.g.get_repos():
            if repo.name in ["repo1", "repo2", "repo3", "repo4"]:
                cls.g.delete_repo(repo.name)

    def test_get_client(self):
        self.assertTrue(isinstance(self.g.get_client(), github.MainClass.Github))

    def test_create_repo(self):
        if "repo4" not in self.g.get_names_of_repos():
            self.g.create_repo("repo4")
        self.assertTrue("repo4" in self.g.get_names_of_repos())

    def test_delete_repo(self):
        if "repo4" not in self.g.get_names_of_repos():
            self.g.create_repo("repo4")
            time.sleep(10)
        self.g.delete_repo("repo4")
        self.assertTrue("self4" not in self.g.get_names_of_repos())

    def test_get_names_of_repos(self):
        self.assertCountEqual(
                ["repo1", "repo2", "repo3"],
                list(x for x in self.g.get_names_of_repos() if x in self.test_repos))

    def test_get_names_of_branches(self):
        self.assertTrue("branch1" in self.g.get_names_of_branches("repo1"))

    def test_create_branch(self):
        self.g.create_branch(self.repo, "branch2")
        self.assertTrue("branch2" in self.g.get_names_of_branches("repo1"))

    def test_protect_branch(self):
        self.g.protect_branch(self.repo, "branch1")
        branch = self.g.get_protected_branch(self.repo, "branch1")
        self.assertTrue(branch.protected)

    def test_create_file(self):
        expected = ['a==1234', 'b>2234,<=2235', 'c>3234']
        files = self.g.get_branch_dir_contents(self.repo, "requirements", self.branch_name)
        result = self.g.convert_github_files(files)
        self.assertCountEqual(expected, result)