def test_get_all_data_sends_wrapper_header_without_version():
with start_server() as server:
config = Config(sdk_key='sdk-key',
base_uri=server.uri,
wrapper_name='Flask')
fr = FeatureRequesterImpl(config)
resp_data = {'flags': {}, 'segments': {}}
server.for_path('/sdk/latest-all', JsonResponse(resp_data))
fr.get_all_data()
req = server.require_request()
assert req.headers.get('X-LaunchDarkly-Wrapper') == 'Flask'
def test_get_all_data_sends_headers():
with start_server() as server:
config = Config(sdk_key='sdk-key', base_uri=server.uri)
fr = FeatureRequesterImpl(config)
resp_data = {'flags': {}, 'segments': {}}
server.for_path('/sdk/latest-all', JsonResponse(resp_data))
fr.get_all_data()
req = server.require_request()
assert req.headers['Authorization'] == 'sdk-key'
assert req.headers['User-Agent'] == 'PythonClient/' + VERSION
assert req.headers.get('X-LaunchDarkly-Wrapper') is None
def _verify_https_proxy_is_used(server, config):
fr = FeatureRequesterImpl(config)
resp_data = {'flags': {}, 'segments': {}}
server.setup_json_response(config.base_uri + '/sdk/latest-all', resp_data)
# Our simple stub server implementation can't really do HTTPS proxying, so the request will fail, but
# it can still record that it *got* the request, which proves that the request went to the proxy.
try:
fr.get_all_data()
except:
pass
req = server.require_request()
assert req.method == 'CONNECT'
def _feature_requester_proxy_test(server, config, secure):
resp_data = {'flags': {}, 'segments': {}}
expected_data = {FEATURES: {}, SEGMENTS: {}}
server.for_path(config.base_uri + '/sdk/latest-all',
JsonResponse(resp_data))
fr = FeatureRequesterImpl(config)
if secure:
try:
fr.get_all_data()
except:
pass # we expect this to fail because we don't have a real HTTPS proxy server
else:
result = fr.get_all_data()
assert result == expected_data
def test_get_all_data_can_use_cached_data():
with start_server() as server:
config = Config(sdk_key='sdk-key', base_uri=server.uri)
fr = FeatureRequesterImpl(config)
etag1 = 'my-etag-1'
etag2 = 'my-etag-2'
resp_data1 = {'flags': {}, 'segments': {}}
resp_data2 = {'flags': {'flag1': {'key': 'flag1'}}, 'segments': {}}
expected_data1 = {FEATURES: {}, SEGMENTS: {}}
expected_data2 = {FEATURES: {'flag1': {'key': 'flag1'}}, SEGMENTS: {}}
req_path = '/sdk/latest-all'
server.for_path(req_path, JsonResponse(resp_data1, {'Etag': etag1}))
result = fr.get_all_data()
assert result == expected_data1
req = server.require_request()
assert 'If-None-Match' not in req.headers.keys()
server.for_path(req_path, BasicResponse(304, None, {'Etag': etag1}))
result = fr.get_all_data()
assert result == expected_data1
req = server.require_request()
assert req.headers['If-None-Match'] == etag1
server.for_path(req_path, JsonResponse(resp_data2, {'Etag': etag2}))
result = fr.get_all_data()
assert result == expected_data2
req = server.require_request()
assert req.headers['If-None-Match'] == etag1
server.for_path(req_path, BasicResponse(304, None, {'Etag': etag2}))
result = fr.get_all_data()
assert result == expected_data2
req = server.require_request()
assert req.headers['If-None-Match'] == etag2
def test_get_all_data_returns_data():
with start_server() as server:
config = Config(sdk_key='sdk-key', base_uri=server.uri)
fr = FeatureRequesterImpl(config)
flags = {'flag1': {'key': 'flag1'}}
segments = {'segment1': {'key': 'segment1'}}
resp_data = {'flags': flags, 'segments': segments}
expected_data = {FEATURES: flags, SEGMENTS: segments}
server.for_path('/sdk/latest-all', JsonResponse(resp_data))
result = fr.get_all_data()
assert result == expected_data
def _verify_http_proxy_is_used(server, config):
fr = FeatureRequesterImpl(config)
resp_data = {'flags': {}, 'segments': {}}
expected_data = {FEATURES: {}, SEGMENTS: {}}
server.setup_json_response(config.base_uri + '/sdk/latest-all', resp_data)
# For an insecure proxy request, our stub server behaves enough like the real thing to satisfy the
# HTTP client, so we should be able to see the request go through. Note that the URI path will
# actually be an absolute URI for a proxy request.
result = fr.get_all_data()
assert result == expected_data
req = server.require_request()
assert req.method == 'GET'