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)
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 )
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
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