27 lines
669 B
Python
27 lines
669 B
Python
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
import unittest
|
|
|
|
import kojihub
|
|
|
|
|
|
class TestSavepoint(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.dml = mock.patch('kojihub.db._dml').start()
|
|
self.context_db = mock.patch('kojihub.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', {})
|