http_client_test.py

import json
from unittest import mock

from imhotep.http_client import BasicAuthRequester


def test_auth():
    ghr = BasicAuthRequester("user", "pass")
    auth = ghr.get_auth()
    assert auth.username == "user"
    assert auth.password == "pass"


def test_empty_auth():
    ghr = BasicAuthRequester("", "")
    auth = ghr.get_auth()
    assert auth is None


def test_get():
    ghr = BasicAuthRequester("user", "pass")
    with mock.patch("requests.get") as g:
        g.return_value.status_code = 200
        ghr.get("url")
        g.assert_called_with("url", auth=mock.ANY)


def test_delete():
    ghr = BasicAuthRequester("user", "pass")
    with mock.patch("requests.delete") as g:
        ghr.delete("url")
        g.assert_called_with("url", auth=mock.ANY)


def test_post():
    ghr = BasicAuthRequester("user", "pass")
    with mock.patch("requests.post") as g:
        g.return_value.status_code = 200
        payload = {"a": 2}
        ghr.post("url", payload)
        g.assert_called_with("url", data=json.dumps(payload), auth=mock.ANY)