tests for host_channels
This commit is contained in:
Tomas Kopecek
Mike McLean
parent
b50479abb0
commit
50a8254df8
3 changed files with 317 additions and 0 deletions
135
tests/test_hub/test_add_host_to_channel.py
Normal file
135
tests/test_hub/test_add_host_to_channel.py
Normal file
|
|
@ -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)
|
||||
79
tests/test_hub/test_list_channels.py
Normal file
79
tests/test_hub/test_list_channels.py
Normal file
|
|
@ -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)
|
||||
103
tests/test_hub/test_remove_host_from_channel.py
Normal file
103
tests/test_hub/test_remove_host_from_channel.py
Normal file
|
|
@ -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)
|
||||
Loading…
Add table
Add a link
Reference in a new issue