def check_iter_locked(db_path, pre_stuff, iter_stuff):
"""Actual implementation of test_errors_locked, so it can be reused."""
# WAL provides more concurrency; some things won't to block with it enabled.
storage = Storage(db_path, wal_enabled=False)
feed = FeedData('one')
entry = EntryData('one', 'entry', datetime(2010, 1, 1), title='entry')
storage.add_feed(feed.url, datetime(2010, 1, 2))
storage.add_or_update_entry(
EntryUpdateIntent(entry, entry.updated, datetime(2010, 1, 1), 0, 0))
storage.add_feed('two', datetime(2010, 1, 1))
storage.add_or_update_entry(
EntryUpdateIntent(entry._replace(feed_url='two'), entry.updated,
datetime(2010, 1, 1), 0, 0))
storage.set_feed_metadata('two', '1', 1)
storage.set_feed_metadata('two', '2', 2)
storage.add_feed_tag('two', '1')
storage.add_feed_tag('two', '2')
if pre_stuff:
pre_stuff(storage)
rv = iter_stuff(storage)
next(rv)
# shouldn't raise an exception
storage = Storage(db_path, timeout=0, wal_enabled=False)
storage.mark_as_read_unread(feed.url, entry.id, 1)
storage = Storage(db_path, timeout=0)
storage.mark_as_read_unread(feed.url, entry.id, 0)
def check_errors_locked(db_path, pre_stuff, do_stuff, exc_type):
"""Actual implementation of test_errors_locked, so it can be reused."""
# WAL provides more concurrency; some things won't to block with it enabled.
storage = Storage(db_path, wal_enabled=False)
storage.db.execute("PRAGMA busy_timeout = 0;")
feed = FeedData('one')
entry = EntryData('one', 'entry', datetime(2010, 1, 2))
storage.add_feed(feed.url, datetime(2010, 1, 1))
storage.add_or_update_entry(
EntryUpdateIntent(entry, entry.updated, datetime(2010, 1, 1), 0, 0))
in_transaction = threading.Event()
can_return_from_transaction = threading.Event()
def target():
storage = Storage(db_path, wal_enabled=False)
storage.db.isolation_level = None
storage.db.execute("BEGIN EXCLUSIVE;")
in_transaction.set()
can_return_from_transaction.wait()
storage.db.execute("ROLLBACK;")
if pre_stuff:
pre_stuff(storage, feed, entry)
thread = threading.Thread(target=target)
thread.start()
in_transaction.wait()
try:
with pytest.raises(exc_type) as excinfo:
do_stuff(storage, feed, entry)
assert 'locked' in str(excinfo.value.__cause__)
finally:
can_return_from_transaction.set()
thread.join()
