particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add tests for group and countOnly for QueryProcessor

Browse source
This commit is contained in:
Yu Ming Zhu 2019-11-28 12:20:13 +00:00 • committed by Tomas Kopecek
parent b92143370e
commit f63d94cbad
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/test_hub/test_query_processor.py
View file

@ -75,6 +75,7 @@ class TestQueryProcessor(unittest.TestCase):
" ORDER BY something OFFSET 10 LIMIT 3"
self.assertEqual(actual, expected)
@mock.patch('kojihub.context')
def test_simple_with_execution(self, context):
cursor = mock.MagicMock()
@ -95,6 +96,18 @@ class TestQueryProcessor(unittest.TestCase):
cursor.execute.assert_called_once_with('\nSELECT count(*)\n FROM awesome\n\n\n \n \n\n \n', {})
self.assertEqual(results, 'some count')
cursor.reset_mock()
args['opts']['group'] = 'id'
args['enable_group'] = True
proc = kojihub.QueryProcessor(**args)
results = proc.execute()
cursor.execute.assert_called_once_with(
'SELECT count(*)\nFROM (\nSELECT 1\n'
' FROM awesome\n\n\n GROUP BY id\n \n\n \n) numrows', {})
self.assertEqual(results, 'some count')
@mock.patch('kojihub.context')
def test_simple_execution_with_iterate(self, context):
cursor = mock.MagicMock()