def setUp(self, mock_git):
self.mock_repo = MagicMock()
self.mock_branches = MagicMock()
self.mock_repo.get_branches.return_value = self.mock_branches
mock_git.get_repo.return_value = self.mock_repo
self.instance = Repo(None, branch='', git=mock_git)
self.instance.branch_exists = False
f = open('./tests/fixtures/content.yaml', 'r')
contents = f.read()
mock_file = MagicMock()
mock_file.decoded_content = contents.encode()
self.instance.files.append(mock_file)
self.sandr = SearchAndReplace(
TestSearchAndReplaceTransformation.Args(), self.instance)
class TestSearchAndReplaceTransformation(unittest.TestCase):
class Args(object):
def __init__(self, dry_run=False, search=['iam'], replace=['hello']):
self.dry_run = dry_run
self.search = search
self.replace = replace
@patch('gordian.repo.Github')
def setUp(self, mock_git):
self.mock_repo = MagicMock()
self.mock_branches = MagicMock()
self.mock_repo.get_branches.return_value = self.mock_branches
mock_git.get_repo.return_value = self.mock_repo
self.mock_repo.get_contents.return_value = []
self.instance = Repo(None, branch='', github=mock_git, files=[])
self.instance.branch_exists = False
f = open('./tests/fixtures/content.yaml', 'r')
contents = f.read()
mock_file = MagicMock()
mock_file.decoded_content = contents.encode()
self.instance.files.append(mock_file)
self.sandr = SearchAndReplace(
TestSearchAndReplaceTransformation.Args(), self.instance)
def test_both_false_update_files(self):
assert (self.sandr.run())
self.instance._forked_repo.create_git_ref.assert_called_once()
self.instance._forked_repo.update_file.assert_called_once()
def test_is_word_found(self):
f = open('./tests/fixtures/content.yaml', 'r')
contents = f.read()
self.assertTrue(self.sandr.is_word_found(contents, 'iam'))
self.assertFalse(self.sandr.is_word_found(contents, 'hello'))
def test_pr_false_update_files(self):
forked_repo = MagicMock()
self.instance._forked_repo = forked_repo
self.instance.branch_exists = True
assert (self.sandr.run())
forked_repo.update_file.assert_called_once()
def test_update_files_none(self):
self.sandr.changesets = (('hello', 'iam'), )
self.sandr.run()
self.mock_repo.update_file.assert_not_called()
self.mock_repo.create_pull.assert_not_called()
def test_update_files_empty(self):
self.forked_repo = MagicMock()
self.instance._forked_repo = self.forked_repo
self.instance._forked_repo.files = []
self.sandr.run()
self.mock_repo.update_file.assert_not_called()
self.mock_repo.create_pull.assert_not_called()