upsolver-sdk-python/upsolver/tests/test_qodo.py at 4c59f336a95fcf37f076d7a5187c4e7314530e4d · Upsolver/upsolver-sdk-python · 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

# Generated by Qodo Gen

# Dependencies:
# pip install pytest-mock
import pytest

class Test__Init__:

    # Initialize RestQueryApi with valid Requester and ResponsePollerBuilder instances
    def test_init_with_valid_parameters(self, mocker):
        from upsolver.client.requester import Requester
        from upsolver.client.poller import ResponsePollerBuilder
        from upsolver.client.query import RestQueryApi

        mock_requester = mocker.Mock(spec=Requester)
        mock_poller_builder = mocker.Mock(spec=ResponsePollerBuilder)

        api = RestQueryApi(requester=mock_requester, poller_builder=mock_poller_builder)

        assert api.requester == mock_requester
        assert api.poller_builder == mock_poller_builder


    # Initialize with None as poller_builder parameter
    def test_init_with_none_poller_builder(self, mocker):
        from upsolver.client.requester import Requester
        from upsolver.client.query import RestQueryApi

        mock_requester = mocker.Mock(spec=Requester)

        api = RestQueryApi(requester=mock_requester, poller_builder=None)

        assert api.requester == mock_requester
        assert api.poller_builder is None

    # Verify requester attribute is correctly set after initialization
    def test_requester_attribute_initialization(self, mocker):
        from upsolver.client.requester import Requester
        from upsolver.client.poller import ResponsePollerBuilder
        from upsolver.client.query import RestQueryApi

        mock_requester = mocker.Mock(spec=Requester)
        mock_poller_builder = mocker.Mock(spec=ResponsePollerBuilder)

        api = RestQueryApi(requester=mock_requester, poller_builder=mock_poller_builder)

        assert api.requester == mock_requester

    def test_connection_rollback_not_supported(mocker):
        from upsolver.dbapi.connection import Connection
        from upsolver.client.exceptions import NotSupportedError

        conn = Connection(token="dummy_token", api_url="http://dummy_url")
        with pytest.raises(NotSupportedError):
            conn.rollback()


    def test_connection_commit_not_supported(mocker):
        from upsolver.dbapi.connection import Connection
        from upsolver.client.exceptions import NotSupportedError

        conn = Connection(token="dummy_token", api_url="http://dummy_url")
        with pytest.raises(NotSupportedError):
            conn.commit()


    def test_connection_exit_no_error(mocker):
        from upsolver.dbapi.connection import Connection

        conn = Connection(token="dummy_token", api_url="http://dummy_url")
        assert conn.__exit__(None, None, None) is True