Python perodic_db_cleanups示例

示例#1

文件： test_util.py 项目： xuefer/home-assistant-core

def test_perodic_db_cleanups(hass_recorder):
    """Test perodic db cleanups."""
    hass = hass_recorder()
    with patch.object(hass.data[DATA_INSTANCE].engine,
                      "execute") as execute_mock:
        util.perodic_db_cleanups(hass.data[DATA_INSTANCE])
    assert execute_mock.call_args[0][0] == "PRAGMA wal_checkpoint(TRUNCATE);"

示例#2

def test_perodic_db_cleanups(hass_recorder):
    """Test perodic db cleanups."""
    hass = hass_recorder()
    with patch.object(hass.data[DATA_INSTANCE].engine,
                      "connect") as connect_mock:
        util.perodic_db_cleanups(hass.data[DATA_INSTANCE])

    text_obj = connect_mock.return_value.__enter__.return_value.execute.mock_calls[
        0][1][0]
    assert isinstance(text_obj, TextClause)
    assert str(text_obj) == "PRAGMA wal_checkpoint(TRUNCATE);"