Дразнящие экземпляры ConfigObj

Question

Используя ConfigObj, я хочу протестировать некоторый код создания раздела:

def create_section(config, section):
    config.reload()
    if section not in config:
         config[session] = {}
         logging.info("Created new section %s.", section)
    else:
         logging.debug("Section %s already exists.", section)

Я хотел бы написать несколько юнит-тестов, но столкнулся с проблемой. Например,

def test_create_section_created():
    config = Mock(spec=ConfigObj)  # ← This is not right…
    create_section(config, 'ook')
    assert 'ook' in config
    config.reload.assert_called_once_with()

Ясно, что метод испытаний не удастся из-за TypeError в качестве аргумента типа "Mock" не повторяется.

Как я могу определить config объект как макет?

Answer 1

И вот почему вы никогда не должны публиковать сообщения до того, как проснулись:

def test_create_section_created():
    logger.info = Mock()
    config = MagicMock(spec=ConfigObj)  # ← This IS right…
    config.__contains__.return_value = False  # Negates the next assert.
    create_section(config, 'ook')
    # assert 'ook' in config  ← This is now a pointless assert!
    config.reload.assert_called_once_with()
    logger.info.assert_called_once_with("Created new section ook.")

Я оставлю этот ответ / вопрос здесь для потомков на случай, если у кого-то еще случится мозговой сбой...