From 50a8254df8de22af0efea3c4a99ee2cb53bac2f5 Mon Sep 17 00:00:00 2001 From: Tomas Kopecek Date: Wed, 21 Feb 2018 14:40:35 +0100 Subject: [PATCH] tests for host_channels --- tests/test_hub/test_add_host_to_channel.py | 135 ++++++++++++++++++ tests/test_hub/test_list_channels.py | 79 ++++++++++ .../test_hub/test_remove_host_from_channel.py | 103 +++++++++++++ 3 files changed, 317 insertions(+) create mode 100644 tests/test_hub/test_add_host_to_channel.py create mode 100644 tests/test_hub/test_list_channels.py create mode 100644 tests/test_hub/test_remove_host_from_channel.py diff --git a/tests/test_hub/test_add_host_to_channel.py b/tests/test_hub/test_add_host_to_channel.py new file mode 100644 index 00000000..7a7792da --- /dev/null +++ b/tests/test_hub/test_add_host_to_channel.py @@ -0,0 +1,135 @@ +import unittest +import mock + +import koji +import kojihub + +IP = kojihub.InsertProcessor + + +class TestAddHostToChannel(unittest.TestCase): + def getInsert(self, *args, **kwargs): + insert = IP(*args, **kwargs) + insert.execute = mock.MagicMock() + self.inserts.append(insert) + return insert + + def setUp(self): + self.InsertProcessor = mock.patch('kojihub.InsertProcessor', + side_effect=self.getInsert).start() + self.inserts = [] + self.context = mock.patch('kojihub.context').start() + # It seems MagicMock will not automatically handle attributes that + # start with "assert" + self.context.session.assertLogin = mock.MagicMock() + self.context.session.assertPerm = mock.MagicMock() + self.context.event_id = 42 + self.context.session.user_id = 23 + self.context.opts = {'HostPrincipalFormat': '-%s-'} + self.exports = kojihub.RootExports() + + def tearDown(self): + mock.patch.stopall() + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_valid(self, get_host, get_channel_id, list_channels): + name = 'hostname' + cname = 'channel_name' + get_host.return_value = {'id': 123, 'name': name} + get_channel_id.return_value = 456 + list_channels.return_value = [{'id': 1, 'name': 'default'}] + + kojihub.add_host_to_channel(name, cname, create=False) + + get_host.assert_called_once_with(name) + get_channel_id.assert_called_once_with(cname, create=False) + list_channels.assert_called_once_with(123) + + self.assertEqual(len(self.inserts), 1) + insert = self.inserts[0] + data = { + 'host_id': 123, + 'channel_id': 456, + 'creator_id': 23, + 'create_event': 42, + } + self.assertEqual(insert.table, 'host_channels') + self.assertEqual(insert.data, data) + self.assertEqual(insert.rawdata, {}) + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_no_host(self, get_host, get_channel_id, list_channels): + name = 'hostname' + cname = 'channel_name' + get_host.return_value = None + + with self.assertRaises(koji.GenericError): + kojihub.add_host_to_channel(name, cname, create=False) + + get_host.assert_called_once_with(name) + self.assertEqual(len(self.inserts), 0) + + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_no_channel(self, get_host, get_channel_id): + name = 'hostname' + cname = 'channel_name' + get_host.return_value = {'id': 123, 'name': name} + get_channel_id.return_value = None + + with self.assertRaises(koji.GenericError): + kojihub.add_host_to_channel(name, cname, create=False) + + get_host.assert_called_once_with(name) + get_channel_id.assert_called_once_with(cname, create=False) + self.assertEqual(len(self.inserts), 0) + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_no_channel_create(self, get_host, get_channel_id, list_channels): + name = 'hostname' + cname = 'channel_name' + get_host.return_value = {'id': 123, 'name': name} + get_channel_id.return_value = 456 + list_channels.return_value = [{'id': 1, 'name': 'default'}] + + kojihub.add_host_to_channel(name, cname, create=True) + + get_host.assert_called_once_with(name) + get_channel_id.assert_called_once_with(cname, create=True) + list_channels.assert_called_once_with(123) + + self.assertEqual(len(self.inserts), 1) + insert = self.inserts[0] + data = { + 'host_id': 123, + 'channel_id': 456, + 'creator_id': 23, + 'create_event': 42, + } + self.assertEqual(insert.table, 'host_channels') + self.assertEqual(insert.data, data) + self.assertEqual(insert.rawdata, {}) + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_exists(self, get_host, get_channel_id, list_channels): + name = 'hostname' + cname = 'channel_name' + get_host.return_value = {'id': 123, 'name': name} + get_channel_id.return_value = 456 + list_channels.return_value = [{'id': 456, 'name': cname}] + + with self.assertRaises(koji.GenericError): + kojihub.add_host_to_channel(name, cname, create=False) + + get_host.assert_called_once_with(name) + get_channel_id.assert_called_once_with(cname, create=False) + list_channels.assert_called_once_with(123) + self.assertEqual(len(self.inserts), 0) diff --git a/tests/test_hub/test_list_channels.py b/tests/test_hub/test_list_channels.py new file mode 100644 index 00000000..ee4e68f0 --- /dev/null +++ b/tests/test_hub/test_list_channels.py @@ -0,0 +1,79 @@ +import unittest +import mock + +import koji +import kojihub + +QP = kojihub.QueryProcessor + + +class TestListChannels(unittest.TestCase): + def getQuery(self, *args, **kwargs): + query = QP(*args, **kwargs) + query.execute = mock.MagicMock() + query.executeOne = mock.MagicMock() + self.queries.append(query) + return query + + def setUp(self): + self.QueryProcessor = mock.patch('kojihub.QueryProcessor', + side_effect=self.getQuery).start() + self.queries = [] + self.context = mock.patch('kojihub.context').start() + # It seems MagicMock will not automatically handle attributes that + # start with "assert" + self.exports = kojihub.RootExports() + + def tearDown(self): + mock.patch.stopall() + + + def test_all(self): + kojihub.list_channels() + self.assertEqual(len(self.queries), 1) + query = self.queries[0] + self.assertEqual(query.tables, ['channels']) + self.assertEqual(query.aliases, ('name', 'id')) + self.assertEqual(query.joins, None) + self.assertEqual(query.values, {}) + self.assertEqual(query.columns, ('channels.name', 'channels.id')) + self.assertEqual(query.clauses, None) + + def test_host(self): + kojihub.list_channels(hostID=1234) + + self.assertEqual(len(self.queries), 1) + query = self.queries[0] + joins = ['channels ON channels.id = host_channels.channel_id'] + clauses = [ + '(host_channels.active = TRUE)', + 'host_channels.host_id = %(host_id)s' + ] + self.assertEqual(query.tables, ['host_channels']) + self.assertEqual(query.aliases, ('name', 'id')) + self.assertEqual(query.joins, joins) + self.assertEqual(query.values, {'host_id': 1234}) + self.assertEqual(query.columns, ('channels.name', 'channels.id')) + self.assertEqual(query.clauses, clauses) + + def test_host_and_event(self): + kojihub.list_channels(hostID=1234, event=2345) + + self.assertEqual(len(self.queries), 1) + query = self.queries[0] + joins = ['channels ON channels.id = host_channels.channel_id'] + clauses = [ + '(host_channels.create_event <= 2345 AND ( host_channels.revoke_event IS NULL OR 2345 < host_channels.revoke_event ))', + 'host_channels.host_id = %(host_id)s', + ] + self.assertEqual(query.tables, ['host_channels']) + self.assertEqual(query.aliases, ('name', 'id')) + self.assertEqual(query.joins, joins) + self.assertEqual(query.values, {'host_id': 1234}) + self.assertEqual(query.columns, ('channels.name', 'channels.id')) + self.assertEqual(query.clauses, clauses) + + def test_event_only(self): + with self.assertRaises(koji.GenericError): + kojihub.list_channels(event=1234) + self.assertEqual(len(self.queries), 0) diff --git a/tests/test_hub/test_remove_host_from_channel.py b/tests/test_hub/test_remove_host_from_channel.py new file mode 100644 index 00000000..f2f31afd --- /dev/null +++ b/tests/test_hub/test_remove_host_from_channel.py @@ -0,0 +1,103 @@ +import unittest +import mock + +import koji +import kojihub + +UP = kojihub.UpdateProcessor + + +class TestRemoveHostFromChannel(unittest.TestCase): + def getUpdate(self, *args, **kwargs): + update = UP(*args, **kwargs) + update.execute = mock.MagicMock() + self.updates.append(update) + return update + + def setUp(self): + self.UpdateProcessor = mock.patch('kojihub.UpdateProcessor', + side_effect=self.getUpdate).start() + self.updates = [] + self.context = mock.patch('kojihub.context').start() + # It seems MagicMock will not automatically handle attributes that + # start with "assert" + self.context.session.assertLogin = mock.MagicMock() + self.context.session.assertPerm = mock.MagicMock() + self.context.event_id = 42 + self.context.session.user_id = 23 + self.context.opts = {'HostPrincipalFormat': '-%s-'} + self.exports = kojihub.RootExports() + + def tearDown(self): + mock.patch.stopall() + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_valid(self, get_host, get_channel_id, list_channels): + get_host.return_value = {'id': 123, 'name': 'hostname'} + get_channel_id.return_value = 234 + list_channels.return_value = [{'id': 234, 'name': 'channelname'}] + + kojihub.remove_host_from_channel('hostname', 'channelname') + + get_host.assert_called_once_with('hostname') + get_channel_id.assert_called_once_with('channelname') + list_channels.assert_called_once_with(123) + + self.assertEqual(len(self.updates), 1) + update = self.updates[0] + values = { + 'host_id': 123, + 'channel_id': 234, + } + clauses = [ + 'host_id = %(host_id)i AND channel_id = %(channel_id)i', + 'active = TRUE', + ] + self.assertEqual(update.table, 'host_channels') + self.assertEqual(update.values, values) + self.assertEqual(update.clauses, clauses) + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_wrong_host(self, get_host, get_channel_id, list_channels): + get_host.return_value = None + + with self.assertRaises(koji.GenericError): + kojihub.remove_host_from_channel('hostname', 'channelname') + + get_host.assert_called_once_with('hostname') + self.assertEqual(len(self.updates), 0) + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_wrong_channel(self, get_host, get_channel_id, list_channels): + get_host.return_value = {'id': 123, 'name': 'hostname'} + get_channel_id.return_value = None + list_channels.return_value = [{'id': 234, 'name': 'channelname'}] + + with self.assertRaises(koji.GenericError): + kojihub.remove_host_from_channel('hostname', 'channelname') + + get_host.assert_called_once_with('hostname') + get_channel_id.assert_called_once_with('channelname') + self.assertEqual(len(self.updates), 0) + + @mock.patch('kojihub.list_channels') + @mock.patch('kojihub.get_channel_id') + @mock.patch('kojihub.get_host') + def test_missing_record(self, get_host, get_channel_id, list_channels): + get_host.return_value = {'id': 123, 'name': 'hostname'} + get_channel_id.return_value = 234 + list_channels.return_value = [] + + with self.assertRaises(koji.GenericError): + kojihub.remove_host_from_channel('hostname', 'channelname') + + get_host.assert_called_once_with('hostname') + get_channel_id.assert_called_once_with('channelname') + list_channels.assert_called_once_with(123) + self.assertEqual(len(self.updates), 0)