def test_geolocate_request(httpx_mock: HTTPXMock):
httpx_mock.add_response(
method="POST", url=f"{SuggestClient.BASE_URL}geolocate/address", json={"suggestions": []},
)
dadata.geolocate(name="address", lat=55.8782557, lon=37.65372, radius_meters=200, count=5)
body = b'{"lat": 55.8782557, "lon": 37.65372, "radius_meters": 200, "count": 5}'
request = httpx_mock.get_request()
assert request.read() == body
def test_find_by_id_request(httpx_mock: HTTPXMock):
httpx_mock.add_response(
method="POST", url=f"{SuggestClient.BASE_URL}findById/party", json={"suggestions": []},
)
dadata.find_by_id(name="party", query="7719402047", count=5)
body = b'{"query": "7719402047", "count": 5}'
request = httpx_mock.get_request()
assert request.read() == body
def test_suggest_request(httpx_mock: HTTPXMock):
httpx_mock.add_response(
method="POST", url=f"{SuggestClient.BASE_URL}suggest/address", json={"suggestions": []},
)
dadata.suggest(name="address", query="samara", to_bound={"value": "city"})
body = b'{"query": "samara", "count": 10, "to_bound": {"value": "city"}}'
request = httpx_mock.get_request()
assert request.read() == body
def test_default_request_retrieval(httpx_mock: HTTPXMock):
httpx_mock.add_response()
with httpx.Client() as client:
client.post("http://test_url", headers={"X-TEST": "test header 1"})
request = httpx_mock.get_request()
assert request.headers["x-test"] == "test header 1"
def test_request_retrieval_on_same_method(httpx_mock: HTTPXMock):
httpx_mock.add_response()
with httpx.Client() as client:
client.get("http://test_url", headers={"X-TEST": "test header 1"})
client.post("http://test_url", headers={"X-TEST": "test header 2"})
request = httpx_mock.get_request(method="GET")
assert request.headers["x-test"] == "test header 1"
def test_request_with_pattern_in_url(httpx_mock: HTTPXMock):
httpx_mock.add_response(url="http://test_url")
httpx_mock.add_response(url="http://unmatched")
with httpx.Client() as client:
client.get("http://unmatched")
client.get("http://test_url", headers={"X-Test": "1"})
assert httpx_mock.get_request(url=re.compile(".*test.*")).headers["x-test"] == "1"
async def test_find_affiliated_request(httpx_mock: HTTPXMock):
httpx_mock.add_response(
method="POST",
url=f"{SuggestClient.BASE_URL}findAffiliated/party",
json={"suggestions": []},
)
await dadata.find_affiliated("7736207543", count=5)
body = b'{"query": "7736207543", "count": 5}'
request = httpx_mock.get_request()
assert request.read() == body
async def test_request_retrieval_on_same_url(httpx_mock: HTTPXMock):
httpx_mock.add_response()
async with httpx.AsyncClient() as client:
await client.get("http://test_url",
headers={"X-TEST": "test header 1"})
await client.get("http://test_url2",
headers={"X-TEST": "test header 2"})
request = httpx_mock.get_request(url=httpx.URL("http://test_url"))
assert request.headers["x-test"] == "test header 1"
def test_requests_retrieval(httpx_mock: HTTPXMock):
httpx_mock.add_response(url="http://test_url", method="GET", data=b"test content 1")
httpx_mock.add_response(
url="http://test_url", method="POST", data=b"test content 2"
)
httpx_mock.add_response(url="http://test_url", method="PUT", data=b"test content 3")
httpx_mock.add_response(
url="http://test_url", method="DELETE", data=b"test content 4"
)
httpx_mock.add_response(
url="http://test_url", method="PATCH", data=b"test content 5"
)
httpx_mock.add_response(
url="http://test_url", method="HEAD", data=b"test content 6"
)
with httpx.Client() as client:
client.post("http://test_url", data=b"sent content 2")
client.get("http://test_url", headers={"X-TEST": "test header 1"})
client.put("http://test_url", data=b"sent content 3")
client.head("http://test_url")
client.patch("http://test_url", data=b"sent content 5")
client.delete("http://test_url", headers={"X-Test": "test header 4"})
assert (
httpx_mock.get_request(url=httpx.URL("http://test_url"), method="PATCH").read()
== b"sent content 5"
)
assert (
httpx_mock.get_request(url=httpx.URL("http://test_url"), method="HEAD").read()
== b""
)
assert (
httpx_mock.get_request(url=httpx.URL("http://test_url"), method="PUT").read()
== b"sent content 3"
)
assert (
httpx_mock.get_request(url=httpx.URL("http://test_url"), method="GET").headers[
"x-test"
]
== "test header 1"
)
assert (
httpx_mock.get_request(url=httpx.URL("http://test_url"), method="POST").read()
== b"sent content 2"
)
assert (
httpx_mock.get_request(
url=httpx.URL("http://test_url"), method="DELETE"
).headers["x-test"]
== "test header 4"
)
def test_aws_auth_root_path(httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(url="http://authorized_only")
httpx.post("http://authorized_only/", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=ce708380ee69b1a9558b9b0dddd4d15f35a2a5e5ea3534b541247f1a746626db"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_without_path(httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(url="http://authorized_only")
httpx.get("http://authorized_only", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=e3411118ac098a820690144b8b273aa64a3366d899fa68fd64a1ab950c982b4b"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_path_normalize(httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(url="http://authorized_only/stuff//more/")
httpx.post("http://authorized_only/./test/../stuff//more/", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=e49fb885d30c9e74901071748b783fabe8ba7a979aa20420ac76af1dda1edd03"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_query_parameters(httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(url="http://authorized_only?param1¶m2=blah*")
httpx.post("http://authorized_only?param1¶m2=blah*", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=f2b8a73e388dc04586b5bcc208c6e50d92f04a1296e561229cd88811ad2494e9"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_with_content_in_request(httpx_mock: HTTPXMock,
mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(url="http://authorized_only")
httpx.post("http://authorized_only", json=[{"key": "value"}], auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"fb65c1441d6743274738fe3b3042a73167ba1fb2d34679d8dd16433473758f97")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=content-type;host;x-amz-content-sha256;x-amz-date, Signature=5f4f832a19fc834d4f34047289ad67d96da25bd414a70f02ce6b85aef9ab8068"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_path_quoting(httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(
url="http://authorized_only/test/hello-*.&%5E%7E+%7B%7D!$%C2%A3_%20")
httpx.post("http://authorized_only/test/hello-*.&^~+{}!$£_ ", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=98dd3cdd2a603907495164f08fe7197fb405bf8c556ddf7b88d7e15341a9588a"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_path_percent_encode_s3(httpx_mock: HTTPXMock,
mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="s3",
)
httpx_mock.add_response(
url="http://authorized_only/test/%252a%252b%2525/%7E-_%5E&%20%25%25")
httpx.post("http://authorized_only/test/%2a%2b%25/~-_^& %%", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/s3/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=2fc7c2f27151e18348862bab0bbe90c4a9f29d7863a33e725d7b1ec96709fdd6"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_path_percent_encode_non_s3(httpx_mock: HTTPXMock,
mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
)
httpx_mock.add_response(
url="http://authorized_only/test/%252a%252b%2525/%7E-_%5E&%20%25%25")
httpx.post("http://authorized_only/test/%2a%2b%25/~-_^& %%", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=1da6c689b7a20044144a9f265ddecc38b1b884902846fbe4dc8049595f25565f"
)
assert headers["x-amz-date"] == "20181011T150505Z"
def test_aws_auth_with_security_token_and_without_content_in_request(
httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
security_token="security_token",
)
httpx_mock.add_response(url="http://authorized_only")
httpx.post("http://authorized_only", auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date;x-amz-security-token, Signature=2ae27ce5e8dcc005736c97ff857e4f44401fc3a33d8358b1d67c079f0f5a8b3e"
)
assert headers["x-amz-date"] == "20181011T150505Z"
assert headers["x-amz-security-token"] == "security_token"
def test_aws_auth_with_security_token_and_content_in_request(
httpx_mock: HTTPXMock, mock_aws_datetime):
auth = httpx_auth.AWS4Auth(
access_id="access_id",
secret_key="wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
region="us-east-1",
service="iam",
security_token="security_token",
)
httpx_mock.add_response(url="http://authorized_only")
httpx.post("http://authorized_only", json=[{"key": "value"}], auth=auth)
headers = httpx_mock.get_request().headers
assert (headers["x-amz-content-sha256"] ==
"fb65c1441d6743274738fe3b3042a73167ba1fb2d34679d8dd16433473758f97")
assert (
headers["Authorization"] ==
"AWS4-HMAC-SHA256 Credential=access_id/20181011/us-east-1/iam/aws4_request, SignedHeaders=content-type;host;x-amz-content-sha256;x-amz-date;x-amz-security-token, Signature=e02c4733589cf6e80361f6905564da6d0c23a0829bb3c3899b328e43b2f7b581"
)
assert headers["x-amz-date"] == "20181011T150505Z"
assert headers["x-amz-security-token"] == "security_token"