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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
from tests import ClientHTTPStubber, patch_load_service_model
from tests.functional.test_useragent import (
get_captured_ua_strings,
parse_registered_feature_ids,
)
MOCK_SERVICE_MODEL = {
"version": "1.0",
"documentation": "",
"metadata": {
"apiVersion": "2020-02-02",
"endpointPrefix": "mockservice",
"protocols": ["smithy-rpc-v2-cbor"],
"protocol": "smithy-rpc-v2-cbor",
"serviceFullName": "Mock Service",
"serviceId": "mockservice",
"signatureVersion": "v4",
"signingName": "mockservice",
"targetPrefix": "mockservice",
"uid": "mockservice-2020-02-02",
},
"operations": {
"MockOperation": {
"name": "MockOperation",
"http": {"method": "GET", "requestUri": "/"},
"input": {"shape": "MockOperationRequest"},
"documentation": "",
},
},
"shapes": {
"MockOpParam": {
"type": "string",
},
"MockOperationRequest": {
"type": "structure",
"required": ["MockOpParam"],
"members": {
"MockOpParam": {
"shape": "MockOpParam",
"documentation": "",
"location": "uri",
"locationName": "param",
},
},
},
},
}
MOCK_RULESET = {
"version": "1.0",
"parameters": {},
"rules": [
{
"conditions": [],
"endpoint": {
"url": "https://mockservice.us-west-2.amazonaws.com/"
},
"type": "endpoint",
},
],
}
def test_user_agent_has_cbor_feature_id(patched_session, monkeypatch):
patch_load_service_model(
patched_session, monkeypatch, MOCK_SERVICE_MODEL, MOCK_RULESET
)
client = patched_session.create_client(
'mockservice', region_name='us-west-2'
)
with ClientHTTPStubber(client) as stub_client:
stub_client.add_response()
# The mock CBOR operation registers `'PROTOCOL_RPC_V2_CBOR': 'M'`
client.mock_operation(MockOpParam='mock-op-param-value')
ua_string = get_captured_ua_strings(stub_client)[0]
feature_list = parse_registered_feature_ids(ua_string)
assert 'M' in feature_list
|
