def test_reset(self): Base = declarative_base() class A(Base): __tablename__ = 'A' id = Column(Integer, primary_key=True) id_values = [id_value1, id_value2] = [12, 34] # Each time it uses different value def initial(db): db.add(A(id=id_values.pop(0))) engine = create_engine('sqlite://') cli = Sqla(orm.sessionmaker(bind=engine), metadata=Base.metadata, initial=initial) with mock.patch.object(sys.stdin, 'readline', return_value='y'): cli.command_reset() query = cli.session.query(A) self.assertEqual(query.count(), 1) a = query.one() self.assertEqual(a.id, id_value1) with mock.patch.object(sys.stdin, 'readline', return_value='y'): cli.command_reset() query = cli.session.query(A) self.assertEqual(query.count(), 1) a = query.one() self.assertEqual(a.id, id_value2)