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)
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)
