def test_check_fail_set_cache():
checker = MagicMock(return_value=False)
with patch.object(cache, 'set') as cache_set_mock:
check_db(checker, 'default', 10)
checker.assert_called_once_with(connections['default'])
cache_set_mock.assert_called_once_with('%s:MagicMock:default' % hostname, 'dead', 10)
def test_check_success_no_cache():
checker = MagicMock(return_value=True)
with patch.object(cache, 'set') as cache_set_mock:
check_db(checker, 'default', 10)
checker.assert_called_once_with(connections['default'])
cache_set_mock.assert_not_called()
def test_check_success_no_cache():
checker = MagicMock(return_value=True)
with patch.object(cache, 'set') as cache_set_mock:
check_db(checker, 'default', 10)
checker.assert_called_once_with(connections['default'])
cache_set_mock.assert_not_called()
def test_check_fail_set_cache():
checker = MagicMock(return_value=False)
with patch.object(cache, 'set') as cache_set_mock:
check_db(checker, 'default', 10)
checker.assert_called_once_with(connections['default'])
cache_set_mock.assert_called_once_with(
'%s:MagicMock:default' % hostname, 'dead', 10)
def test_check_fail_get_cache_force():
checker = MagicMock(return_value=False)
with patch.object(cache, 'get') as cache_get_mock:
cache_get_mock.return_value = 'dead'
check_db(checker, 'default', 10, force=True)
cache_get_mock.assert_not_called()
checker.assert_called_once_with(connections['default'])
def test_check_fail_get_cache():
checker = MagicMock(return_value=False)
with patch.object(cache, 'get') as cache_get_mock:
cache_get_mock.return_value = 'dead'
check_db(checker, 'default', 10)
cache_get_mock.assert_called_once_with('%s:MagicMock:default' % hostname)
checker.assert_not_called()
def test_check_fail_get_cache_force():
checker = MagicMock(return_value=False)
with patch.object(cache, 'get') as cache_get_mock:
cache_get_mock.return_value = 'dead'
check_db(checker, 'default', 10, force=True)
cache_get_mock.assert_not_called()
checker.assert_called_once_with(connections['default'])
def test_check_fail_get_cache():
checker = MagicMock(return_value=False)
with patch.object(cache, 'get') as cache_get_mock:
cache_get_mock.return_value = 'dead'
check_db(checker, 'default', 10)
cache_get_mock.assert_called_once_with('%s:MagicMock:default' %
hostname)
checker.assert_not_called()
def test_check_retry():
checker = MagicMock(return_value=False)
check_db(checker, 'default', number_of_tries=3)
checker.assert_has_calls([call(connections['default']) for _ in range(3)])
def test_check_fail():
assert check_db(MagicMock(return_value=False), 'default') == False
def test_check_success():
assert check_db(MagicMock(return_value=True), 'default') == True
def test_check_retry():
checker = MagicMock(return_value=False)
check_db(checker, 'default', number_of_tries=3)
checker.assert_has_calls([call(connections['default']) for _ in range(3)])
def test_check_fail():
assert check_db(MagicMock(return_value=False), 'default') is False
def test_check_success():
assert check_db(MagicMock(return_value=True), 'default') is True
