Python perodic_db_cleanups Examples

Example #1

File: test_util.py Project: 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);"

Example #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);"