Python _init_optional_base Examples

Example #1

File: test_init.py Project: codingjoe/OpenAlchemy

def test_init_optional_base_none_call(
        mocked_init_model_factory: mock.MagicMock,
        mocked_declarative_base: mock.MagicMock):
    """
    GIVEN mocked init_model_factory and declarative_base
    WHEN _init_optional_base is called with none base
    THEN init_model_factory is called with declarative_base return value as base.
    """
    # pylint: disable=protected-access
    spec = mock.MagicMock()

    open_alchemy._init_optional_base(base=None, spec=spec, define_all=True)

    mocked_init_model_factory.assert_called_once_with(
        base=mocked_declarative_base.return_value, spec=spec, define_all=True)

Example #2

File: test_init.py Project: SkynetRTN/OpenAlchemy

def test_init_optional_base_def_call(mocked_init_model_factory: mock.MagicMock):
    """
    GIVEN mocked init_model_factory and mock base
    WHEN _init_optional_base is called with the base
    THEN init_model_factory is called with base.
    """
    # pylint: disable=protected-access
    spec = mock.MagicMock()
    base = mock.MagicMock()

    open_alchemy._init_optional_base(base=base, spec=spec, define_all=True)

    mocked_init_model_factory.assert_called_once_with(
        base=base, spec=spec, define_all=True, models_filename=None, spec_path=None
    )

Example #3

def test_init_optional_base_def_return(
        _mocked_init_model_factory: mock.MagicMock):
    """
    GIVEN mocked init_model_factory and and mock base
    WHEN _init_optional_base is called with the base
    THEN the base is returned.
    """
    # pylint: disable=protected-access
    spec = mock.MagicMock()
    base = mock.MagicMock()

    returned_base, _ = open_alchemy._init_optional_base(base=base, spec=spec)

    assert returned_base == base

Example #4

def test_init_optional_base_none_return(
        _mocked_init_model_factory: mock.MagicMock,
        mocked_declarative_base: mock.MagicMock):
    """
    GIVEN mocked init_model_factory and declarative_base
    WHEN _init_optional_base is called with none base
    THEN the declarative_base return value is returned as base.
    """
    # pylint: disable=protected-access
    spec = mock.MagicMock()

    base, _ = open_alchemy._init_optional_base(base=None, spec=spec)

    assert base == mocked_declarative_base.return_value