"""Tests for query_tools."""

from unittest.mock import patch

import query_tools
from tests import helpers


@patch('query_tools.requests')
def test_get_main_instance_query_json(
        requests, get_mock_query_json):
    """Return dict."""
    requests.get.return_value = get_mock_query_json
    result = query_tools.get_main_instance_query_json(1)
    assert result == helpers.query_json


@patch('query_tools.requests')
def test_post_query_to_client_instance(
        requests, get_mock_query_json):
    """Return int."""
    requests.post.return_value = get_mock_query_json
    result = query_tools.post_query_to_client_instance({})
    assert result == 4


@patch('query_tools.get_main_instance_query_json')
@patch('query_tools.post_query_to_client_instance')
def test_sync_query(
        post_query_to_client_instance, get_main_instance_query_json):
    """Assert calls get and post."""
    get_main_instance_query_json.return_value = helpers.query_json
    post_query_to_client_instance.return_value = 4
    query_tools.sync_query(helpers.look_json)

    get_main_instance_query_json.assert_called()
    post_query_to_client_instance.assert_called_with(helpers.query_json)