Example #1
0
async def test_delete_lock_and_base_folder(
    hass,
    mock_osremove,
    mock_osrmdir,
):
    """Test delete_lock_and_base_folder"""
    entry = MockConfigEntry(domain=DOMAIN,
                            title="frontdoor",
                            data=CONFIG_DATA,
                            version=2)

    entry.add_to_hass(hass)
    assert await hass.config_entries.async_setup(entry.entry_id)
    await hass.async_block_till_done()

    delete_lock_and_base_folder(hass, entry)

    assert mock_osrmdir.called
    assert mock_osremove.called

    with patch("custom_components.keymaster.helpers.os",
               autospec=True) as mock_os:
        mock_os.listdir.return_value = False
        delete_lock_and_base_folder(hass, entry)
        mock_os.rmdir.assert_called_once
Example #2
0
async def test_delete_lock_and_base_folder(hass, mock_osrmdir, mock_osremove):
    """Test delete_lock_and_base_folder"""
    entry = MockConfigEntry(domain=DOMAIN,
                            title="frontdoor",
                            data=CONFIG_DATA,
                            version=2)

    entry.add_to_hass(hass)
    assert await hass.config_entries.async_setup(entry.entry_id)
    await hass.async_block_till_done()

    delete_lock_and_base_folder(hass, entry)

    assert mock_osrmdir.called
    assert mock_osremove.called