saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Cleanup.

Browse source
This commit is contained in:
Gareth J. Greenaway 2018-12-14 09:28:44 -08:00
parent b85c5bf6d9
commit af3e1459c3
No known key found for this signature in database
GPG key ID: 10B62F8A7CAD7A41
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/unit/modules/test_mysql.py
View file

@ -283,17 +283,11 @@ class MySQLTestCase(TestCase, LoaderModuleMockMixin):
self._test_call(mysql.query, 'SELECT * FROM testdb', 'testdb', 'SELECT * FROM testdb')
def test_query_error(self):
def mock_execute(cur, query):
'''
Mock of info method
'''
raise MySQLdb.OperationalError(9999, 'Something Went Wrong')
connect_mock = MagicMock()
with patch.object(mysql, '_connect', connect_mock):
with patch.dict(mysql.__salt__, {'config.option': MagicMock()}):
with patch.object(mysql, '_execute') as execute_mock:
execute_mock.side_effect = mock_execute
side_effect = MySQLdb.OperationalError(9999, 'Something Went Wrong')
with patch.object(mysql, '_execute', MagicMock(side_effect=side_effect)):
mysql.query('testdb', 'SELECT * FROM testdb')
self.assertIn('mysql.error', mysql.__context__)
expected = 'MySQL Error 9999: Something Went Wrong'