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
|
import pytest
from globus_sdk.services.gcs import UnpackingGCSResponse
def test_unpacking_response_with_callback(make_response):
def identify_desired_data(d):
return "foo" in d
base_resp = make_response(
json_body={
"data": [
{"x": 1, "y": 2},
{"x": 2, "y": 3, "foo": "bar"},
]
}
)
resp = UnpackingGCSResponse(base_resp, identify_desired_data)
assert resp["x"] == 2
assert resp["y"] == 3
assert resp["foo"] == "bar"
@pytest.mark.parametrize("datatype", ["foo#1.0.1", "foo#1", "foo#0.0.1", "foo#2.0.0.1"])
def test_unpacking_response_matches_datatype(make_response, datatype):
base_resp = make_response(json_body={"data": [{"x": 1, "DATA_TYPE": datatype}]})
resp = UnpackingGCSResponse(base_resp, "foo")
assert "x" in resp
# using some other spec will not match
resp_bar = UnpackingGCSResponse(base_resp, "bar")
assert "x" not in resp_bar
def test_unpacking_response_invalid_spec(make_response):
base_resp = make_response(json_body={"data": [{"x": 1, "DATA_TYPE": "foo#1.0.0"}]})
with pytest.raises(ValueError):
UnpackingGCSResponse(base_resp, "foo 1.0")
def test_unpacking_response_invalid_datatype(make_response):
# we'll never return a match if the DATA_TYPE doesn't appear to be valid
base_resp = make_response(json_body={"data": [{"x": 1, "DATA_TYPE": "foo"}]})
resp = UnpackingGCSResponse(base_resp, "foo")
assert "x" not in resp
|
