From 5770b7e360af587f8e478b08e9bc4aa160b85760 Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Dec 21 2021 13:46:15 +0000
Subject: expand get_channel unit tests


---

diff --git a/tests/test_hub/test_get_channel.py b/tests/test_hub/test_get_channel.py
index 23cebf9..880af65 100644
--- a/tests/test_hub/test_get_channel.py
+++ b/tests/test_hub/test_get_channel.py
@@ -6,9 +6,22 @@ import koji
 import kojihub
 
 
+QP = kojihub.QueryProcessor
+
+
 class TestGetChannel(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()
         self.exports = kojihub.RootExports()
 
@@ -29,3 +42,37 @@ class TestGetChannel(unittest.TestCase):
             self.exports.getChannel(channel_info)
         self.assertEqual('Invalid name or id value: %s' % channel_info,
                          str(cm.exception))
+
+    def test_query_by_name(self):
+        self.exports.getChannel('my_channel')
+        self.assertEqual(len(self.queries), 1)
+        query = self.queries[0]
+        clauses = ['(channels.name = %(channels_name)s)']
+        values = {'channels_name': 'my_channel'}
+        self.assertEqual(query.tables, ['channels'])
+        self.assertEqual(query.joins, None)
+        self.assertEqual(set(query.clauses), set(clauses))
+        self.assertEqual(query.values, values)
+
+
+    def test_query_by_id(self):
+        self.exports.getChannel(12345)
+        self.assertEqual(len(self.queries), 1)
+        query = self.queries[0]
+        clauses = ['(channels.id = %(channels_id)s)']
+        values = {'channels_id': 12345}
+        self.assertEqual(query.tables, ['channels'])
+        self.assertEqual(query.joins, None)
+        self.assertEqual(set(query.clauses), set(clauses))
+        self.assertEqual(query.values, values)
+
+    def test_query_by_dict(self):
+        self.exports.getChannel({'id':12345, 'name': 'whatever'})
+        self.assertEqual(len(self.queries), 1)
+        query = self.queries[0]
+        clauses = ['(channels.id = %(channels_id)s)']
+        values = {'channels_id': 12345}
+        self.assertEqual(query.tables, ['channels'])
+        self.assertEqual(query.joins, None)
+        self.assertEqual(set(query.clauses), set(clauses))
+        self.assertEqual(query.values, values)