def test_remove_entry_undo(session): entry = DatabaseEntry() session.add(entry) cmd = RemoveEntry(session, entry) session.commit() cmd() assert session.query(DatabaseEntry).count() == 0 cmd.undo() assert session.query(DatabaseEntry).count() == 1
def test_remove_entry_repr(session): entry = DatabaseEntry(id=3) expected_repr_result = ( '<RemoveEntry(' 'session <sqlalchemy.orm.session.Session object at *>, ' 'entry <DatabaseEntry(id 3)>)>'.format(id(session))) assert fnmatch.fnmatch(repr(RemoveEntry(session, entry)), expected_repr_result)
def test_remove_existing_entry(session): entry = DatabaseEntry() session.add(entry) assert session.query(DatabaseEntry).count() == 1 assert entry.id == 1 RemoveEntry(session, entry)() assert entry in session.deleted assert session.query(DatabaseEntry).count() == 0
def test_add_removed_entry(session): entry = DatabaseEntry() AddEntry(session, entry)() session.commit() RemoveEntry(session, entry)() session.commit() AddEntry(session, entry)() session.commit() assert not session.new assert entry.id == 1
def test_remove_nonexisting_entry(session): with pytest.raises(NoSuchEntryError): RemoveEntry(session, DatabaseEntry())()