test_event_context_formatter.py

from ldclient.context import Context
from ldclient.impl.events.event_context_formatter import EventContextFormatter

def test_simple_context():
    f = EventContextFormatter(False, [])
    c = Context.create('a')
    assert f.format_context(c) == {'kind': 'user', 'key': 'a'}

def test_context_with_more_attributes():
    f = EventContextFormatter(False, [])
    c = Context.builder('a').name('b').anonymous(True).set('c', True).set('d', 2).build()
    assert f.format_context(c) == {
        'kind': 'user',
        'key': 'a',
        'name': 'b',
        'anonymous': True,
        'c': True,
        'd': 2
    }

def test_context_can_redact_anonymous_attributes():
    f = EventContextFormatter(False, [])
    c = Context.builder('a').name('b').anonymous(True).set('c', True).set('d', 2).build()
    assert f.format_context_redact_anonymous(c) == {
        'kind': 'user',
        'key': 'a',
        'anonymous': True,
        '_meta': {
            'redactedAttributes': ['name', 'c', 'd']
        }
    }

def test_multi_kind_context_can_redact_anonymous_attributes():
    f = EventContextFormatter(False, [])
    user = Context.builder('user-key').name('b').anonymous(True).set('c', True).set('d', 2).build()
    org = Context.builder('org-key').kind('org').name('b').set('c', True).set('d', 2).build()
    multi = Context.create_multi(user, org)

    assert f.format_context_redact_anonymous(multi) == {
        'kind': 'multi',
        'user': {
            'key': 'user-key',
            'anonymous': True,
            '_meta': {
                'redactedAttributes': ['name', 'c', 'd']
            }
        },
        'org': {
            'key': 'org-key',
            'name': 'b',
            'c': True,
            'd': 2
        }
    }

def test_multi_context():
    f = EventContextFormatter(False, [])
    c = Context.create_multi(
        Context.create('a'),
        Context.builder('b').kind('c').name('d').build()
    )
    assert f.format_context(c) == {
        'kind': 'multi',
        'user': {
            'key': 'a'
        },
        'c': {
            'key': 'b',
            'name': 'd'
        }
    }

def test_all_private():
    f = EventContextFormatter(True, [])
    c = Context.builder('a').name('b').anonymous(True).set('c', True).set('d', 2).build()
    assert f.format_context(c) == {
        'kind': 'user',
        'key': 'a',
        'anonymous': True,
        '_meta': {'redactedAttributes': ['name', 'c', 'd']}
    }

def test_some_private_global():
    f = EventContextFormatter(False, ['name', 'd'])
    c = Context.builder('a').name('b').anonymous(True).set('c', True).set('d', 2).build()
    assert f.format_context(c) == {
        'kind': 'user',
        'key': 'a',
        'anonymous': True,
        'c': True,
        '_meta': {'redactedAttributes': ['name', 'd']}
    }

def test_some_private_per_context():
    f = EventContextFormatter(False, ['name'])
    c = Context.builder('a').name('b').anonymous(True).set('c', True).set('d', 2).private('d').build()
    assert f.format_context(c) == {
        'kind': 'user',
        'key': 'a',
        'anonymous': True,
        'c': True,
        '_meta': {'redactedAttributes': ['name', 'd']}
    }

def test_private_property_in_object():
    f = EventContextFormatter(False, ['/b/prop1', '/c/prop2/sub1'])
    c = Context.builder('a') \
        .set('b', {'prop1': True, 'prop2': 3}) \
        .set('c', {'prop1': {'sub1': True}, 'prop2': {'sub1': 4, 'sub2': 5}}) \
        .build()
    assert f.format_context(c) == {
        'kind': 'user',
        'key': 'a',
        'b': {'prop2': 3},
        'c': {'prop1': {'sub1': True}, 'prop2': {'sub2': 5}},
        '_meta': {'redactedAttributes': ['/b/prop1', '/c/prop2/sub1']}
    }