Files
git-upstream/git_upstream/tests/test_commands.py
Darragh Bailey d8bb479880 Ensure parent options retained for exec'ed finish
Retrieve parent parser arguments set by user, and ensure they are made
available as part of the default script cmdline. This allows the
logging from the exec'ed finish command called as the last step by
git-rebase to log to the same file automatically.

Change-Id: Ia79a54afe8ace7ade2aa5a29ad16855c9883bd54
Closes-Bug: #1625864
2016-09-29 16:54:55 +01:00

133 lines
4.3 KiB
Python

# Copyright (c) 2012, 2013, 2014 Hewlett-Packard Development Company, L.P.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests for the 'commands' module"""
from argparse import ArgumentParser
import os
import subprocess
import tempfile
import mock
import testtools
from testtools import content
from testtools import matchers
from git_upstream import commands as c
from git_upstream import main
from git_upstream.tests import base
class TestGetSubcommands(testtools.TestCase):
"""Test case for get_subcommands function"""
_available_subcommands = ('help', 'import', 'supersede', 'drop')
def test_available_subcommands(self):
"""Test available subcommands"""
parser = ArgumentParser()
subcommands = c.get_subcommands(parser)
self.assertEqual(len(TestGetSubcommands._available_subcommands),
len(subcommands.keys()))
for command in subcommands.keys():
self.assertIn(command, TestGetSubcommands._available_subcommands)
class TestMainParser(testtools.TestCase):
"""Test cases for main parser"""
def test_logging_options_kept(self):
argv = ["-v", "--log-level", "debug", "import"]
with mock.patch(
'git_upstream.commands.GitUpstreamCommand.run') as mock_import:
main.main(argv)
self.assertThat(mock_import.call_args[0][0].script_cmdline[2:],
matchers.Equals(argv[:3]))
class TestLoggingOptions(base.BaseTestCase):
"""Test cases for logging options behaviour"""
def setUp(self):
_, self.parser = main.build_parsers()
script_cmdline = self.parser.get_default('script_cmdline')
script_cmdline[-1] = os.path.join(os.getcwd(), main.__file__)
self.parser.set_defaults(script_cmdline=script_cmdline)
# builds the tree to be tested
super(TestLoggingOptions, self).setUp()
def test_logfile_contains_finish(self):
"""Confirm that logger calls in 'finish' phase recorded
Repository layout being checked
B---C local/master
/
A---D---E upstream/master
"""
tree = [
('A', []),
('B', ['A']),
('C', ['B']),
('D', ['A']),
('E', ['D']),
]
branches = {
'head': ('master', 'C'),
'upstream': ('upstream/master', 'E'),
}
self.gittree = base.BuildTree(self.testrepo, tree, branches.values())
cmdline = self.parser.get_default('script_cmdline')
tfile = None
try:
tfile = tempfile.NamedTemporaryFile(delete=False)
# need to close to allow reopen to write
tfile.close()
cmdline.extend(['-v', '--log-file', tfile.name, '--log-level',
'debug', 'import', '--into', 'master',
'upstream/master'])
try:
output = subprocess.check_output(cmdline,
stderr=subprocess.STDOUT)
except subprocess.CalledProcessError as cpe:
self.addDetail(
'subprocess-output',
content.text_content(cpe.output.decode('utf-8')))
raise
self.addDetail(
'subprocess-output',
content.text_content(output.decode('utf-8')))
logfile_contents = open(tfile.name, 'r').read()
finally:
if tfile and os.path.exists(tfile.name):
os.remove(tfile.name)
self.assertThat(
logfile_contents,
matchers.Contains("Merging by inverting the 'ours' strategy"))
self.assertThat(
logfile_contents,
matchers.Contains("Replacing tree contents with those from"))