1cfe6538db
Move QueryProcessor, InsertProcessor, UpdateProcessor, BulkInsertProcessor, _applyQueryOpts, get_event, _dml, _fetchMulti, _fetchSingle, _singleValue, _multiRow, _singleRow Update koji-sweep-db script to DB Processors Fixes: https://pagure.io/koji/issue/3466
24 lines
604 B
Python
24 lines
604 B
Python
import mock
|
|
|
|
import unittest
|
|
|
|
import kojihub
|
|
|
|
|
|
class TestSavepoint(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.dml = mock.patch('koji.db._dml').start()
|
|
self.context_db = mock.patch('koji.db.context').start()
|
|
|
|
def tearDown(self):
|
|
mock.patch.stopall()
|
|
|
|
def test_savepoint(self):
|
|
sp = kojihub.Savepoint('some_name')
|
|
self.assertEqual(sp.name, 'some_name')
|
|
self.dml.assert_called_once_with('SAVEPOINT some_name', {})
|
|
|
|
self.dml.reset_mock()
|
|
sp.rollback()
|
|
self.dml.assert_called_once_with('ROLLBACK TO SAVEPOINT some_name', {})
|