mirror of
https://git.notmuchmail.org/git/notmuch
synced 2024-11-23 19:38:07 +01:00
60 lines
2 KiB
Python
60 lines
2 KiB
Python
import collections.abc
|
|
|
|
import pytest
|
|
|
|
import notmuch2._database as dbmod
|
|
|
|
import notmuch2._config as config
|
|
|
|
|
|
class TestIter:
|
|
|
|
@pytest.fixture
|
|
def db(self, maildir):
|
|
with dbmod.Database.create(maildir.path) as db:
|
|
yield db
|
|
|
|
def test_type(self, db):
|
|
assert isinstance(db.config, collections.abc.MutableMapping)
|
|
assert isinstance(db.config, config.ConfigMapping)
|
|
|
|
def test_alive(self, db):
|
|
assert db.config.alive
|
|
|
|
def test_set_get(self, maildir):
|
|
# Ensure get-set works from different db objects
|
|
with dbmod.Database.create(maildir.path, config=dbmod.Database.CONFIG.EMPTY) as db0:
|
|
db0.config['spam'] = 'ham'
|
|
with dbmod.Database(maildir.path, config=dbmod.Database.CONFIG.EMPTY) as db1:
|
|
assert db1.config['spam'] == 'ham'
|
|
|
|
def test_get_keyerror(self, db):
|
|
with pytest.raises(KeyError):
|
|
val = db.config['not-a-key']
|
|
print(repr(val))
|
|
|
|
def test_iter(self, db):
|
|
def has_prefix(x):
|
|
return x.startswith('TEST.')
|
|
|
|
assert [ x for x in db.config if has_prefix(x) ] == []
|
|
db.config['TEST.spam'] = 'TEST.ham'
|
|
db.config['TEST.eggs'] = 'TEST.bacon'
|
|
assert { x for x in db.config if has_prefix(x) } == {'TEST.spam', 'TEST.eggs'}
|
|
assert { x for x in db.config.keys() if has_prefix(x) } == {'TEST.spam', 'TEST.eggs'}
|
|
assert { x for x in db.config.values() if has_prefix(x) } == {'TEST.ham', 'TEST.bacon'}
|
|
assert { (x, y) for (x,y) in db.config.items() if has_prefix(x) } == \
|
|
{('TEST.spam', 'TEST.ham'), ('TEST.eggs', 'TEST.bacon')}
|
|
|
|
def test_len(self, db):
|
|
defaults = len(db.config)
|
|
db.config['spam'] = 'ham'
|
|
assert len(db.config) == defaults + 1
|
|
db.config['eggs'] = 'bacon'
|
|
assert len(db.config) == defaults + 2
|
|
|
|
def test_del(self, db):
|
|
db.config['spam'] = 'ham'
|
|
assert db.config.get('spam') == 'ham'
|
|
del db.config['spam']
|
|
assert db.config.get('spam') is None
|