def set_test_values(cls):
"""
This will set the global branches, wallace_branch_name,
wallace_branch_commits, TEST_wallace_branch_model,
TEST_master_branch_model.
"""
dummy_model = QGitModel(cls.TEST_dir)
branches = dummy_model.get_branches()
cls.TEST_master_branch = [branch for branch in branches
if branch.name == 'master'][0]
master_model = cls.TEST_master_branch_model = QGitModel(cls.TEST_dir)
master_model.set_current_branch(cls.TEST_master_branch)
master_model.populate()
cls.TEST_wallace_branch = [branch for branch in branches
if branch.name == 'wallace_branch'][0]
wallace_model = cls.TEST_wallace_branch_model = QGitModel(cls.TEST_dir)
wallace_model.set_current_branch(cls.TEST_wallace_branch)
wallace_model.populate()
cls.run_command("git checkout master")
cls.TEST_master_branch_commits = cls.read_commits_from_log()
cls.run_command("git checkout wallace_branch")
cls.TEST_wallace_branch_commits = cls.read_commits_from_log()
stdout, stderr = cls.run_command("git branch")
cls.TEST_branches = []
for line in stdout:
branch_name = line.split(' ')[-1].strip()
if line[0] == '*':
cls.TEST_current_branch_name = branch_name
cls.TEST_branches.append(branch_name)
def set_current_directory(self, directory, reset_all=False):
"""
Sets the current directory.
:param directory:
The git directory.
"""
self._models = {}
a_model = QGitModel(directory)
self.current_branch = a_model.get_current_branch()
for branch in a_model.get_branches():
model = QEditableGitModel(self._models, directory=directory,
parent=self)
model.set_current_branch(branch)
model.setMerge(False)
model.enable_option("filters")
model.populate()
self._models[branch] = model
QObject.connect(model, SIGNAL("newHistoryEvent"),
self.new_history_event)
if reset_all:
self.rebase_main_class.reset_interface(self._models)
self.filter_main_class.reset_interface(self._models)
