notifications-admin/tests/app/models/test_base_model.py

import pytest

from app.models import JSONModel


def test_looks_up_from_dict():

    class Custom(JSONModel):
        ALLOWED_PROPERTIES = {'foo'}

    assert Custom({'foo': 'bar'}).foo == 'bar'


def test_raises_when_overriding_custom_properties():

    class Custom(JSONModel):

        ALLOWED_PROPERTIES = {'foo'}

        @property
        def foo(self):
            pass

    with pytest.raises(AttributeError) as e:
        Custom({'foo': 'NOPE'})

    assert str(e.value) == "can't set attribute"


@pytest.mark.parametrize('json_response', (
    {},
    {'foo': 'bar'},  # Should still raise an exception
))
def test_model_raises_for_unknown_attributes(json_response):

    class Custom(JSONModel):
        ALLOWED_PROPERTIES = set()

    model = Custom(json_response)
    assert model.ALLOWED_PROPERTIES == set()

    with pytest.raises(AttributeError) as e:
        model.foo

    assert str(e.value) == "'Custom' object has no attribute 'foo'"


def test_model_raises_keyerror_if_item_missing_from_dict():

    class Custom(JSONModel):
        ALLOWED_PROPERTIES = {'foo'}

    with pytest.raises(AttributeError) as e:
        Custom({}).foo

    assert str(e.value) == "'Custom' object has no attribute 'foo'"


@pytest.mark.parametrize('json_response', (
    {},
    {'foo': 'bar'},  # Should be ignored
))
def test_model_doesnt_swallow_attribute_errors(json_response):

    class Custom(JSONModel):
        ALLOWED_PROPERTIES = set()
        @property
        def foo(self):
            raise AttributeError('Something has gone wrong')

    with pytest.raises(AttributeError) as e:
        Custom(json_response).foo

    assert str(e.value) == 'Something has gone wrong'


def test_dynamic_properties_are_introspectable():

    class Custom(JSONModel):
        ALLOWED_PROPERTIES = {'foo', 'bar', 'baz'}

    model = Custom({'foo': None, 'bar': None, 'baz': None})

    assert dir(model)[-3:] == ['bar', 'baz', 'foo']