def test_trackcontexts(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c', 'd'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'cache_size',
{'outcome': 'created'})
setting = Setting('cache_size', int, ['b', 'c'], 50)
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'cache_size': {
'rules': [{
'context_features': [['b', 'B']],
'value': 100,
'rule_id': 1
}],
'default_value':
100
}
}
})), fake_heksher_service.query_rules.capture_calls(
) as query_calls:
client = ThreadHeksherClient(fake_heksher_service.local_url(), 100000,
['a', 'b', 'c', 'd'])
client.track_contexts(b='B', a=['a0', 'a1'], d=TRACK_ALL)
with client:
assert setting.get(b='B', c='') == 100
query_calls.assert_requested_once_with(
query_params={
'settings': ['cache_size'],
'context_filters': ['a:(a0,a1),b:(B),d:*'],
'include_metadata': ['false']
})
def test_regular_update(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'cache_size',
{'outcome': 'created'})
setting = Setting('cache_size', int, ['b', 'c'], 50)
with ThreadHeksherClient(fake_heksher_service.local_url(), 0.02,
['a', 'b', 'c']):
assert setting.get(b='', c='') == 50
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'cache_size': {
'rules': [{
'context_features': [],
'value': 100,
'rule_id': 1
}],
'default_value':
100
}
}
})):
sleep(0.1)
assert setting.get(b='', c='') == 100
def test_declare_after_main(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'cache_size',
{'outcome': 'created'})
with ThreadHeksherClient(fake_heksher_service.local_url(), 100000,
['a', 'b', 'c']) as client:
setting = Setting('cache_size', int, ['b', 'c'], 50)
assert not client._undeclared.empty()
client._undeclared.join()
assert setting.get(b='', c='') == 50
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'cache_size': {
'rules': [{
'context_features': [],
'value': 100,
'rule_id': 1
}],
'default_value':
100
}
}
})):
client.reload()
assert setting.get(b='', c='') == 100
def test_get_settings(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
settings = {
'settings': [{
'name': 'foo',
'configurable_features': ['a', 'b'],
'type': 'int',
'default_value': None,
'metadata': {},
'aliases': [],
'version': '1.0',
}]
}
with fake_heksher_service.get_settings.patch(JSONResponse(settings)):
with ThreadHeksherClient(fake_heksher_service.local_url(), 1000,
['a', 'b', 'c']) as client:
response = client.get_settings()
assert response == {
"foo":
SettingData(name="foo",
configurable_features=['a', 'b'],
type='int',
default_value=None,
metadata={},
aliases=[],
version='1.0')
}
def test_redundant_defaults(fake_heksher_service, caplog, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'cache_size',
{'outcome': 'created'})
setting = Setting('cache_size', int, ['b', 'c'], 50)
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'cache_size': {
'rules': [{
'context_features': [['b', 'B']],
'value': 100,
'rule_id': 1
}],
'default_value':
100
}
}
})):
with assert_logs(caplog, WARNING):
with ThreadHeksherClient(fake_heksher_service.local_url(), 1000,
['a', 'b', 'c']) as client:
client.set_defaults(b='B', d='im redundant')
assert setting.get(c='') == 100
def test_cf_mismatch(fake_heksher_service, caplog, monkeypatch, expected):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
with assert_logs(caplog, WARNING):
with ThreadHeksherClient(fake_heksher_service.local_url(), 1000,
expected) as client:
assert client._context_features == expected
def test_heksher_unreachable(caplog):
setting = Setting('cache_size', int, ['b', 'c'], 50)
with assert_logs(caplog, ERROR):
with ThreadHeksherClient('http://notreal.fake.notreal', 10000,
['a', 'b', 'c']):
pass
assert setting.get(b='', c='') == 50
def test_switch_main_from_temp(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features', ['a', 'b'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf1',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf2',
{'outcome': 'created'})
setting1 = Setting('conf1', int, ['a'], 74)
client = ThreadHeksherClient(fake_heksher_service.local_url(), 10000000,
['a', 'b'])
client.track_contexts(a=['a', 'b'], b=TRACK_ALL)
client.set_as_main()
setting2 = Setting('conf2', int, ['b'], 26)
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'conf1': {
'rules': [{
'context_features': [],
'value': 5,
'rule_id': 1
}],
'default_value':
74
},
'conf2': {
'rules': [{
'context_features': [],
'value': 4,
'rule_id': 2
}],
'default_value':
26
}
}
})):
client.reload()
assert setting1.get(a='') == 5
assert setting2.get(b='') == 4
client.close()
def test_switch_main_unclosed(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features', ['a', 'b'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf1',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf2',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf3',
{'outcome': 'created'})
setting1 = Setting('conf1', int, ['a'], 74)
client1 = ThreadHeksherClient(fake_heksher_service.local_url(), 10000000,
['a', 'b'])
client1.track_contexts(a=['a', 'b'], b=TRACK_ALL)
client1.set_as_main()
client1.close()
setting2 = Setting('conf2', int, ['b'], 26)
client2 = ThreadHeksherClient(fake_heksher_service.local_url(), 10000000,
['a', 'b'])
client2.track_contexts(a=['a', 'b'], b=TRACK_ALL)
with raises(RuntimeError): # not allowed
client2.set_as_main()
def test_switch_main_different_contexts(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf1',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf2',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf3',
{'outcome': 'created'})
setting1 = Setting('conf1', int, ['a'], 74)
client1 = ThreadHeksherClient(fake_heksher_service.local_url(), 10000000,
['a', 'b'])
client1.track_contexts(a=['a', 'b'], b=TRACK_ALL)
client1.set_as_main()
setting2 = Setting('conf2', int, ['b'], 26)
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'conf1': {
'rules': [{
'context_features': [],
'value': 5,
'rule_id': 1
}],
'default_value':
74
},
'conf2': {
'rules': [{
'context_features': [],
'value': 4,
'rule_id': 2
}],
'default_value':
26
}
}
})):
client1.reload()
assert len(client1._tracked_settings) == 2
assert setting1.get(a='') == 5
assert setting2.get(b='') == 4
client2 = ThreadHeksherClient(fake_heksher_service.local_url(), 10000000,
['b', 'c'])
client2.track_contexts(a=['a', 'b'], b=TRACK_ALL)
client1.close()
with raises(RuntimeError): # not allowed
client2.set_as_main()
def test_switch_main_different_tracking(fake_heksher_service, monkeypatch,
caplog):
monkeypatch.setattr(fake_heksher_service, 'context_features', ['a', 'b'])
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf1',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf2',
{'outcome': 'created'})
monkeypatch.setitem(fake_heksher_service.declare_responses, 'conf3',
{'outcome': 'created'})
setting1 = Setting('conf1', int, ['a'], 74)
client1 = ThreadHeksherClient(fake_heksher_service.local_url(), 10000000,
['a', 'b'])
client1.track_contexts(a=['a', 'b'], b=TRACK_ALL)
client1.set_as_main()
setting2 = Setting('conf2', int, ['b'], 26)
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'conf1': {
'rules': [{
'context_features': [],
'value': 5,
'rule_id': 1
}],
'default_value':
74
},
'conf2': {
'rules': [{
'context_features': [],
'value': 4,
'rule_id': 2
}],
'default_value':
26
}
}
})):
client1.reload()
assert len(client1._tracked_settings) == 2
assert setting1.get(a='') == 5
assert setting2.get(b='') == 4
client2 = ThreadHeksherClient(fake_heksher_service.local_url(),
10000000, ['a', 'b'])
client2.track_contexts(a=['a', 'b', 'c'], b="shoobidoobi")
client1.close()
with assert_logs(
caplog, WARNING
): # it should warn you you're doing bad things, and that your tracking differs
client2.set_as_main()
setting3 = Setting('conf3', int, ['b'], 59)
with fake_heksher_service.query_rules.patch(
JSONResponse({
'settings': {
'conf1': {
'rules': [{
'context_features': [],
'value': 5,
'rule_id': 1
}],
'default_value':
74,
},
'conf2': {
'rules': [{
'context_features': [],
'value': 4,
'rule_id': 2
}],
'default_value':
26,
},
'conf3': {
'rules': [{
'context_features': [],
'value': 3,
'rule_id': 3
}],
'default_value':
59,
}
}
})):
client2.reload()
assert setting1.get(a='') == 5
assert setting2.get(b='') == 4
assert setting3.get(b='') == 3
client2.close()
def test_health_unreachable():
client = ThreadHeksherClient('http://notreal.fake.notreal', 1000,
['a', 'b', 'c'])
with raises(HTTPError):
client.ping()
def test_health_err(fake_heksher_service, monkeypatch):
with fake_heksher_service.health.patch(Response(status_code=403)):
client = ThreadHeksherClient(fake_heksher_service.local_url(), 1000,
['a', 'b', 'c'])
with raises(HTTPError):
client.ping()
def test_health_OK(fake_heksher_service):
client = ThreadHeksherClient(fake_heksher_service.local_url(), 1000,
['a', 'b', 'c'])
client.ping()
def test_init_works(fake_heksher_service, monkeypatch):
monkeypatch.setattr(fake_heksher_service, 'context_features',
['a', 'b', 'c'])
with ThreadHeksherClient(fake_heksher_service.local_url(), 100000,
['a', 'b', 'c']):
pass