draft filter opt: use bool/None instead of bit flag

This commit is contained in:
Yu Ming Zhu 2023-10-09 22:46:13 +00:00
parent 8a34f7efa0
commit 77a06c98b3
7 changed files with 65 additions and 114 deletions

View file

@ -103,7 +103,7 @@ n-v-r tag owner
self.session.getTag.assert_called_once_with(self.tag, event=self.event_id)
self.session.listTagged.assert_called_once_with(
self.tag, event=self.event_id, inherit=True, latest=True, package=self.pkg,
draft=2)
draft=False)
self.session.listTaggedRPMS.assert_not_called()
self.assert_console_message(stdout, expected)
@ -121,7 +121,7 @@ n-v-r tag owner
self.ensure_connection_mock.assert_called_once_with(self.session, self.options)
self.session.getTag.assert_called_once_with(self.tag, event=None)
self.session.listTagged.assert_called_once_with(
self.tag, inherit=True, latest=True, package=self.pkg, draft=1)
self.tag, inherit=True, latest=True, package=self.pkg, draft=True)
self.session.listTaggedRPMS.assert_not_called()
@mock.patch('sys.stdout', new_callable=six.StringIO)

View file

@ -101,7 +101,7 @@ class TestListBuilds(unittest.TestCase):
package_id = 1
self.get_package_id.return_value = package_id
self.query_executeOne.return_value = None
self.exports.listBuilds(packageID=package, draft=1)
self.exports.listBuilds(packageID=package, draft=True)
self.assertEqual(len(self.queries), 1)
args, kwargs = self.QueryProcessor.call_args
qp = QP(**kwargs)

View file

@ -84,7 +84,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
'package': None, 'packages': self.package_list,
'queryOpts': {'order': '-create_event'}, 'st_complete': 1, 'tables': self.tables,
'tag': self.tag_name, 'tagid': self.tag_name, 'taglist': [self.tag_name],
'type': None, 'draft': 3
'type': None, 'draft': None
}
self.assertEqual(query.tables, self.tables)
self.assertEqual(query.joins, self.joins)
@ -120,7 +120,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
'package': self.pkg_name, 'packages': self.package_list,
'queryOpts': {'order': '-create_event'}, 'st_complete': 1, 'tables': self.tables,
'tag': self.tag_name, 'tagid': self.tag_name, 'taglist': [self.tag_name],
'type': 'maven', 'draft': 3}
'type': 'maven', 'draft': None}
self.assertEqual(query.tables, self.tables)
self.assertEqual(query.joins, joins)
self.assertEqual(set(query.columns), set(columns))
@ -149,7 +149,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
'package': None, 'packages': self.package_list,
'queryOpts': {'order': '-create_event'}, 'st_complete': 1, 'tables': self.tables,
'tag': self.tag_name, 'tagid': self.tag_name, 'taglist': [self.tag_name],
'type': 'win', 'draft': 3}
'type': 'win', 'draft': None}
self.assertEqual(query.tables, self.tables)
self.assertEqual(query.joins, joins)
self.assertEqual(set(query.columns), set(columns))
@ -178,7 +178,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
'package': None, 'packages': self.package_list,
'queryOpts': {'order': '-create_event'}, 'st_complete': 1, 'tables': self.tables,
'tag': self.tag_name, 'tagid': self.tag_name, 'taglist': [self.tag_name],
'type': 'image', 'draft': 3}
'type': 'image', 'draft': None}
self.assertEqual(query.tables, self.tables)
self.assertEqual(query.joins, joins)
self.assertEqual(set(query.columns), set(columns))
@ -213,7 +213,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
'package': None, 'packages': self.package_list,
'queryOpts': {'order': '-create_event'}, 'st_complete': 1, 'tables': self.tables,
'tag': self.tag_name, 'tagid': self.tag_name, 'taglist': [self.tag_name],
'type': type, 'draft': 3}
'type': type, 'draft': None}
self.assertEqual(query.tables, self.tables)
self.assertEqual(query.joins, joins)
self.assertEqual(set(query.columns), set(self.columns))
@ -223,7 +223,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
def test_get_tagged_builds_draft(self):
self.readPackageList.return_value = self.package_list
kojihub.readTaggedBuilds(self.tag_name, draft=koji.DRAFT_FLAG.DRAFT)
kojihub.readTaggedBuilds(self.tag_name, draft=True)
self.assertEqual(len(self.queries), 1)
query = self.queries[0]
@ -236,7 +236,7 @@ class TestReadTaggedBuilds(unittest.TestCase):
'package': None, 'packages': self.package_list,
'queryOpts': {'order': '-create_event'}, 'st_complete': 1, 'tables': self.tables,
'tag': self.tag_name, 'tagid': self.tag_name, 'taglist': [self.tag_name],
'type': None, 'draft': koji.DRAFT_FLAG.DRAFT
'type': None, 'draft': True
}
self.assertEqual(query.tables, self.tables)

View file

@ -105,7 +105,7 @@ class TestReadTaggedRPMS(unittest.TestCase):
def test_get_tagged_rpms_draft(self):
self.readTaggedBuilds.return_value = self.build_list
kojihub.readTaggedRPMS(self.tag_name, draft=2, extra=False)
kojihub.readTaggedRPMS(self.tag_name, draft=False, extra=False)
self.assertEqual(len(self.queries), 1)
query = self.queries[0]