def test_module_func_imported():
node = extract_node('''
from requests import get as custom_name
custom_name
''')
assert is_requests_func(node)
def test_module_func_simple(func_name):
node = extract_node(f'''
import requests
requests.{func_name}
''')
assert is_requests_func(node)
def test_session_method_in_function():
node = extract_node('''
import requests
def request_api(sess: requests.Session):
sess.get #@
''')
assert is_requests_func(node)
def test_session_method_by_annotation():
node = extract_node('''
from requests import Session
session: Session = get_session_from_somewhere()
session.get
''')
assert is_requests_func(node)
def test_session_method_simple(method_name):
node = extract_node(f'''
import requests
session = requests.Session()
session.{method_name}
''')
assert is_requests_func(node)
def test_module_func_assigned():
node = extract_node('''
import requests
custom_name = requests.get
custom_name
''')
assert is_requests_func(node)
def test_session_method_from_constructor_parameter():
node = extract_node('''
import requests
class MyAPIClient:
def __init__(self, http: requests.Session):
self.http = http
def test_method(self):
self.http.get #@
''')
assert is_requests_func(node)
def test_session_method_in_class():
node = extract_node('''
import requests
class MyAPIClient:
def __init__(self):
self.http = requests.Session()
def test_method(self):
self.http.get #@
''')
assert is_requests_func(node)
def test_some_other_session_class():
node = extract_node('''
class Session:
def get(self, *args, **kwargs):
pass
class MyAPIClient:
def __init__(self, session: Session):
self.session = session
def test_method(self):
self.session.get #@
''')
assert not is_requests_func(node)
def test_not_requests():
node = extract_node('''
requests = {}
requests.get
''')
assert not is_requests_func(node)
