def test_command_line_update_long(self): with mock.patch( 'sys.argv', [ 'cluster', '-y', 'deploy', '--master', 'master-node', '--slave', 'slave-node', '-d', '-t', '10', 'reponame', 'branch', '--update' ] ): with mock.patch('cluster.cluster.Cluster.deploy') as mo: main() mo.assert_called_once_with( 'reponame', 'branch', master='master-node', slave='slave-node', no_wait=True, timeout=10, ask_user=False, update=True )
def test_command_line(self): node = 'node1' command_line_args = self.fixture_command_line(node) with mock.patch(*command_line_args): with mock.patch('cluster.cluster.Cluster.inspect_node') as mo: main() mo.assert_called_once_with(node)
def test_check_command_lines(self): self.fill_data() with OutputCapture() as output: with mock.patch('sys.argv', [ 'cluster', 'checks', ]): main() output.compare("\n".join([ "Node node-2:", " - Service Service 3:", " - Check (critical): Check Service 3", ]))
def test_command_line_no_update(self): with mock.patch('sys.argv', [ 'cluster', '-y', 'migrate', 'reponame', 'source-branch', 'target-branch', '--target-repo', 'reponame2', '--no-wait', '-t', '5', '-n', ]): with mock.patch('cluster.cluster.Cluster.migrate') as mo: main() mo.assert_called_once_with('reponame', 'source-branch', 'target-branch', target_repo='reponame2', no_wait=True, timeout=5, ask_user=False, no_update=True) with mock.patch('sys.argv', [ 'cluster', '-y', 'migrate', 'reponame', 'source-branch', 'target-branch', '--target-repo', 'reponame2', '--no-wait', '-t', '5', '--no-update', ]): with mock.patch('cluster.cluster.Cluster.migrate') as mo: main() mo.assert_called_once_with('reponame', 'source-branch', 'target-branch', target_repo='reponame2', no_wait=True, timeout=5, ask_user=False, no_update=True)
def test_command_output_empty_node(self): node = 'node5' command_line_args = self.fixture_command_line(node) self.mocked_consul.configure_mock( **{ 'kv.find': lambda state: self._app_kv, }) with OutputCapture() as output: with mock.patch(*command_line_args): main() output.compare("\n".join([ "Master apps of node {node}:".format(node=node), ]))
def test_command_line(self): with mock.patch( 'sys.argv', ['cluster', 'deploy', 'reponame', 'branch', ] ): with mock.patch('cluster.cluster.Cluster.deploy') as mo: main() mo.assert_called_once_with( 'reponame', 'branch', master=None, slave=None, no_wait=False, timeout=cluster.DEFAULT_TIMEOUT, ask_user=True, update=False )
def test_command_line(self): with mock.patch('sys.argv', [ 'cluster', '-y', 'move-masters-from', 'node-1', '--master', 'node-2', '--no-wait', '-t', '5', ]): with mock.patch('cluster.cluster.Cluster.move_masters_from') as mo: main() mo.assert_called_once_with('node-1', master='node-2', no_wait=True, timeout=5, ask_user=False)