37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
import unittest
|
|
|
|
import kojihub
|
|
|
|
|
|
class TestUpdateProcessor(unittest.TestCase):
|
|
|
|
def test_basic_instantiation(self):
|
|
kojihub.UpdateProcessor('sometable') # No exception!
|
|
|
|
def test_to_string_with_data(self):
|
|
proc = kojihub.UpdateProcessor('sometable', data={'foo': 'bar'})
|
|
actual = str(proc)
|
|
expected = 'UPDATE sometable SET foo = %(data.foo)s'
|
|
self.assertEqual(actual, expected)
|
|
|
|
def test_to_values_from_data(self):
|
|
proc = kojihub.UpdateProcessor('sometable', data={'foo': 'bar'})
|
|
actual = proc.get_values()
|
|
expected = {'data.foo': 'bar'}
|
|
self.assertEqual(actual, expected)
|
|
|
|
@mock.patch('kojihub.db.context')
|
|
def test_simple_execution_with_iterate(self, context_db):
|
|
cursor = mock.MagicMock()
|
|
context_db.cnx.cursor.return_value = cursor
|
|
proc = kojihub.UpdateProcessor('sometable', data={'foo': 'bar'})
|
|
proc.execute()
|
|
cursor.execute.assert_called_once_with(
|
|
'UPDATE sometable SET foo = %(data.foo)s',
|
|
{'data.foo': 'bar'},
|
|
log_errors=True,
|
|
)
|