def test_list_channels(self, ensure_connection_mock, stdout):
self.session.listChannels.return_value = [
{'id': 1, 'name': 'default'},
{'id': 2, 'name': 'test'},
]
self.session.multiCall.return_value = [
[[
{'enabled': True, 'ready': True, 'capacity': 2.0, 'task_load': 1.34},
{'enabled': True, 'ready': False, 'capacity': 2.0, 'task_load': 0.0},
{'enabled': True, 'ready': False, 'capacity': 2.0, 'task_load': 0.0},
]],
[[
{'enabled': True, 'ready': True, 'capacity': 2.0, 'task_load': 1.34},
{'enabled': False, 'ready': True, 'capacity': 2.0, 'task_load': 0.34},
{'enabled': True, 'ready': False, 'capacity': 2.0, 'task_load': 0.0},
]],
]
anon_handle_list_channels(self.options, self.session, self.args)
actual = stdout.getvalue()
print(actual)
expected = """\
default 3 1 0 1 6 22%
test 2 2 1 1 6 28%
"""
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session)
def test_list_channels(self, activate_session_mock, stdout):
self.session.listChannels.return_value = [
{
'id': 1,
'name': 'default'
},
{
'id': 2,
'name': 'test'
},
]
self.session.multiCall.return_value = [
[[
{
'enabled': True,
'ready': True
},
{
'enabled': True,
'ready': False
},
{
'enabled': True,
'ready': False
},
]],
[[
{
'enabled': True,
'ready': True
},
{
'enabled': False,
'ready': True
},
{
'enabled': True,
'ready': False
},
]],
]
anon_handle_list_channels(self.options, self.session, self.args)
actual = stdout.getvalue()
expected = 'default 3 1 0\ntest 2 2 1\n'
self.assertMultiLineEqual(actual, expected)
activate_session_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels_with_empty_channels(self, ensure_connection_mock,
stdout):
list_channels = []
list_hosts_mc = [[[]]]
self.session.listChannels.return_value = list_channels
self.session.multiCall.return_value = list_hosts_mc
args = []
anon_handle_list_channels(self.options, self.session, args)
actual = stdout.getvalue()
print(actual)
expected = ''
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels_with_comment(self, ensure_connection_mock, stdout):
self.session.listChannels.return_value = self.list_channels
self.session.multiCall.return_value = self.list_hosts_mc
args = ['--comment']
anon_handle_list_channels(self.options, self.session, args)
actual = stdout.getvalue()
print(actual)
expected = 'default 3 1 0 1 6 22% ' \
'test-comment-1 \n' \
'test [disabled] 2 2 1 1 6 28% ' \
'test-comment-2 \n'
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels_simple_with_quiet(self, ensure_connection_mock,
stdout):
self.session.listChannels.return_value = self.list_channels
self.session.multiCall.return_value = self.list_hosts_mc
args = ['--simple', '--quiet']
anon_handle_list_channels(self.options, self.session, args)
actual = stdout.getvalue()
print(actual)
expected = """default
test [disabled]
"""
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels_with_enabled_hub_without_enabled(
self, ensure_connection_mock, stdout):
self.session.listChannels.side_effect = [
koji.ParameterError, self.list_channels_without_enabled
]
self.session.multiCall.return_value = self.list_hosts_mc
args = ['--enabled']
anon_handle_list_channels(self.options, self.session, args)
actual = stdout.getvalue()
print(actual)
expected = 'default 3 1 0 1 6 22%\n' \
'test 2 2 1 1 6 28%\n'
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels_not_quiet(self, ensure_connection_mock, stdout):
self.session.listChannels.return_value = self.list_channels
self.session.multiCall.return_value = self.list_hosts_mc
args = []
self.options.quiet = False
anon_handle_list_channels(self.options, self.session, args)
actual = stdout.getvalue()
print(actual)
expected = "Channel Enabled Ready Disbld Load Cap Perc \n" \
"default 3 1 0 1 6 22%\n" \
"test [disabled] 2 2 1 1 6 28%\n"
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels(self, activate_session_mock, stdout):
self.session.listChannels.return_value = [
{
'id': 1,
'name': 'default'
},
{
'id': 2,
'name': 'test'
},
]
self.session.multiCall.return_value = [[[1, 2, 3]], [[4, 5]]]
anon_handle_list_channels(self.options, self.session, self.args)
actual = stdout.getvalue()
expected = 'default 3\ntest 2\n'
self.assertMultiLineEqual(actual, expected)
activate_session_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels_with_disabled(self, ensure_connection_mock, stdout):
list_channel_disabled = [
{
'id': 2,
'name': 'test',
'enabled': False,
'comment': 'test-comment-2',
'description': 'test-description-2'
},
]
list_hosts_mc = [[[
{
'enabled': True,
'ready': True,
'capacity': 2.0,
'task_load': 1.34
},
{
'enabled': False,
'ready': True,
'capacity': 2.0,
'task_load': 0.34
},
{
'enabled': True,
'ready': False,
'capacity': 2.0,
'task_load': 0.0
},
]]]
self.session.listChannels.return_value = list_channel_disabled
self.session.multiCall.return_value = list_hosts_mc
args = ['--disabled']
anon_handle_list_channels(self.options, self.session, args)
actual = stdout.getvalue()
print(actual)
expected = 'test [disabled] 2 2 1 1 6 28%\n'
self.assertMultiLineEqual(actual, expected)
ensure_connection_mock.assert_called_once_with(self.session,
self.options)
def test_list_channels(self, activate_session_mock, stdout):
self.session.listChannels.return_value = [
{'id': 1, 'name': 'default'},
{'id': 2, 'name': 'test'},
]
self.session.multiCall.return_value = [
[[
{'enabled': True, 'ready': True},
{'enabled': True, 'ready': False},
{'enabled': True, 'ready': False},
]],
[[
{'enabled': True, 'ready': True},
{'enabled': False, 'ready': True},
{'enabled': True, 'ready': False},
]],
]
anon_handle_list_channels(self.options, self.session, self.args)
actual = stdout.getvalue()
expected = 'default 3 1 0\ntest 2 2 1\n'
self.assertMultiLineEqual(actual, expected)
activate_session_mock.assert_called_once_with(self.session, self.options)