FreedomBox/plinth/tests/test_kvstore.py
Sunil Mohan Adapa 97b579c4fc
kvstore: Optionally, don't throw exception when deleting key
Tests:

- Unit tests pass.

Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org>
Reviewed-by: James Valleroy <jvalleroy@mailbox.org>
2023-07-28 21:07:57 -04:00

61 lines
1.5 KiB
Python

# SPDX-License-Identifier: AGPL-3.0-or-later
"""
Test module for key/value store.
"""
import pytest
from plinth import kvstore
from plinth.models import KVStore
pytestmark = pytest.mark.django_db
def test_get_set():
"""Verify that a set value can be retrieved."""
key = 'name'
expected_value = 'Guido'
kvstore.set(key, expected_value)
actual_value = kvstore.get(key)
assert expected_value == actual_value
def test_get_set_complex_structures():
"""Verify that complex structures can be stored and retrieved."""
key = 'compex_structure'
expected_value = {
'k1': 1,
'k2': [2, 3],
'k3': 4.5,
'k4': 'Hello',
'k5': {
'a': 'b'
}
}
kvstore.set(key, expected_value)
actual_value = kvstore.get(key)
assert expected_value == actual_value
def test_get_default():
"""Verify that either a set value or its default can be retrieved."""
expected = 'default'
actual = kvstore.get_default('bad_key', expected)
assert expected == actual
def test_delete():
"""Test that deleting key works."""
with pytest.raises(KVStore.DoesNotExist):
kvstore.delete('nonexistant_key')
with pytest.raises(KVStore.DoesNotExist):
kvstore.delete('nonexistant_key', ignore_missing=False)
kvstore.delete('nonexistant_key', ignore_missing=True)
kvstore.set('test-set-key', 'test-value')
kvstore.delete('test-set-key')
with pytest.raises(KVStore.DoesNotExist):
kvstore.delete('test-set-key')