49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Copyright: (c) 2020, Abhijeet Kasurde <akasurde@redhat.com>
|
||
|
|
# Copyright: (c) 2020, Ansible Project
|
||
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||
|
|
|
||
|
|
from __future__ import (absolute_import, division, print_function)
|
||
|
|
__metaclass__ = type
|
||
|
|
|
||
|
|
|
||
|
|
from ansible.module_utils.api import rate_limit, retry
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
class TestRateLimit:
|
||
|
|
|
||
|
|
def test_ratelimit(self):
|
||
|
|
@rate_limit(rate=1, rate_limit=1)
|
||
|
|
def login_database():
|
||
|
|
return "success"
|
||
|
|
r = login_database()
|
||
|
|
|
||
|
|
assert r == 'success'
|
||
|
|
|
||
|
|
|
||
|
|
class TestRetry:
|
||
|
|
|
||
|
|
def test_no_retry_required(self):
|
||
|
|
self.counter = 0
|
||
|
|
|
||
|
|
@retry(retries=4, retry_pause=2)
|
||
|
|
def login_database():
|
||
|
|
self.counter += 1
|
||
|
|
return 'success'
|
||
|
|
|
||
|
|
r = login_database()
|
||
|
|
|
||
|
|
assert r == 'success'
|
||
|
|
assert self.counter == 1
|
||
|
|
|
||
|
|
def test_catch_exception(self):
|
||
|
|
|
||
|
|
@retry(retries=1)
|
||
|
|
def login_database():
|
||
|
|
return 'success'
|
||
|
|
|
||
|
|
with pytest.raises(Exception):
|
||
|
|
login_database()
|