robotframework-requests/utests/test_RequestsOnSessionKeywords.py at c9add73de9241b2a3cf8f5e51b488f0528f078fd · MarketSquare/robotframework-requests · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import os

from RequestsLibrary import RequestsLibrary
from utests import SCRIPT_DIR
from utests import mock


def build_mocked_session_common_request(alias='alias', url='http://mocking.rules',
                                        verify=None, cookies={}, timeout=None):
    keywords = RequestsLibrary()
    session = keywords.create_session(alias, url, verify=verify, cookies=cookies, timeout=timeout)
    # this prevents a real network call from being executed
    session.request = mock.MagicMock()
    return session, keywords._common_request


def test_common_request_file_descriptor_closing():
    session, m_common_request = build_mocked_session_common_request()
    with open(os.path.join(SCRIPT_DIR, '../atests/randombytes.bin'), 'rb') as f:
        m_common_request('get', session,
                         'http://mocking.rules', data=f)
        assert f.closed is True


def test_common_request_files_descriptor_closing_when_passed_as_files_param():
    session, m_common_request = build_mocked_session_common_request()
    with open(os.path.join(SCRIPT_DIR, '../atests/randombytes.bin'), 'rb') as f1:
        with open(os.path.join(SCRIPT_DIR, '../atests/data.json'), 'rb') as f2:
            m_common_request('get', session,
                             'http://mocking.rules', files={'randombytes': f1, 'data': f2})
            assert f1.closed is True
            assert f2.closed is True


def test_common_request_files_descriptor_closing_when_passed_as_file_tuple():
    session, m_common_request = build_mocked_session_common_request()
    with open(os.path.join(SCRIPT_DIR, '../atests/randombytes.bin'), 'rb') as f:
        m_common_request(
            'get',
            session,
            'http://mocking.rules',
            files={'randombytes': ('randombytes.bin', f)},
        )
        assert f.closed is True


def test_common_request_verify_override_true():
    session, m_common_request = build_mocked_session_common_request(verify=False)
    m_common_request('get', session, '/', verify=True)
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={}, verify=True)
    assert session.verify is False


def test_common_request_verify_override_false():
    session, m_common_request = build_mocked_session_common_request(verify=True)
    m_common_request('get', session, '/', verify=False)
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={}, verify=False)
    assert session.verify is True


def test_common_request_verify_true_default():
    session, m_common_request = build_mocked_session_common_request(verify=True)
    m_common_request('get', session, '/')
    assert session.verify is True
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={})


def test_common_request_verify_false_default():
    session, m_common_request = build_mocked_session_common_request(verify=False)
    m_common_request('get', session, '/')
    assert session.verify is False
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={})


def test_common_request_with_cookies_override():
    session, m_common_request = build_mocked_session_common_request()
    m_common_request('get', session, '/', cookies={'a': 1, 'b': 2})
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={'a': 1, 'b': 2})


def test_common_request_with_cookies_override_default():
    session, m_common_request = build_mocked_session_common_request(cookies={'a': 1, 'b': 2})
    m_common_request('get', session, '/', cookies={'a': 3, 'b': 4})
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={'a': 3, 'b': 4})


def test_common_request_with_cookies_default_only():
    session, m_common_request = build_mocked_session_common_request(cookies={'a': 1, 'b': 2})
    m_common_request('get', session, '/')
    session.request.assert_called_with('get','http://mocking.rules/', timeout=None, cookies={'a': 1, 'b': 2})


def test_common_request_with_float_timeout():
    session, m_common_request = build_mocked_session_common_request(timeout=123.4)
    m_common_request('get', session, '/')
    session.request.assert_called_with('get','http://mocking.rules/', timeout=123.4, cookies={})


def test_common_request_with_float_timeout_override():
    session, m_common_request = build_mocked_session_common_request(timeout=None)
    m_common_request('get', session, '/', timeout=123.4)
    session.request.assert_called_with('get','http://mocking.rules/', timeout=123.4, cookies={})


def test_common_request_with_touple_timeout():
    session, m_common_request = build_mocked_session_common_request(timeout=(123.4, 432.1))
    m_common_request('get', session, '/')
    session.request.assert_called_with('get','http://mocking.rules/', timeout=(123.4, 432.1), cookies={})


def test_common_request_with_touple_timeout_override():
    session, m_common_request = build_mocked_session_common_request(timeout=None)
    m_common_request('get', session, '/', timeout=(123.4, 432.1))
    session.request.assert_called_with('get','http://mocking.rules/', timeout=(123.4, 432.1), cookies={})