def test_dropMimeData(self):
self.session.add(model.User(name='John', lastname='Dow'))
self.model_user.refresh()
index = self.model_user.createIndex(0, 2)
self.assertEqual(
self.model_user.dropMimeData('name', Qt.MoveAction, 0, 2, index),
False)
self.assertEqual(
self.model_user.dropMimeData('a', Qt.DropAction, 0, 2, index),
None)
self.assertEqual(self.model_user.data(index, Qt.DisplayRole), 'John')
def test_user_model(self):
"""Test the user model
"""
ed_user = model.User(name='Ed', lastname='Jones')
self.session.add(ed_user)
our_user = self.session.query(model.User).filter_by(name='Ed').first()
self.assertEqual(our_user.name, 'Ed')
self.assertEqual(our_user.lastname, 'Jones')
print(self.session.query(model.User).first())
self.assertEqual(our_user.clubname, None)
self.assertEqual(our_user.agename, None)
self.assertEqual(our_user.bowname, None)
def test_data(self):
"""test data and setData"""
index = QModelIndex()
index.sibling(0, 0)
self.assertEqual(index.isValid(), False)
self.assertEqual(
self.model_user.data(index, Qt.DisplayRole).value(), None)
self.session.add(model.User(name='John', lastname='Dow'))
self.model_user.refresh()
index = self.model_user.createIndex(0, 1)
self.assertEqual(index.isValid(), True)
self.assertEqual(self.model_user.data(index, Qt.DisplayRole), 'Dow')
index = self.model_user.createIndex(0, 2)
self.assertEqual(self.model_user.data(index, Qt.DisplayRole), 'John')
self.assertEqual(self.model_user.setData(index, 'Jonny'), True)
self.assertEqual(self.model_user.data(index, Qt.DisplayRole), 'Jonny')
def test_user_model_with_other(self):
"""Test the user model and add club, bow and age.
"""
self.session.add(model.Club(name='long bow club', short='long'))
self.session.add(model.Bow(name='long bow', short='long'))
self.session.add(model.Age(name='young talented', short='advanced'))
self.session.add(
model.User(name='Ed',
lastname='Jones',
club_id=1,
age_id=1,
bow_id=1))
print(self.session.query(model.User).first())
our_user = self.session.query(model.User).filter_by(name='Ed').first()
self.assertEqual(our_user.clubname, 'long bow club')
self.assertEqual(our_user.agename, 'young talented')
self.assertEqual(our_user.bowname, 'long bow')
def test_refresh(self):
self.session.add(model.User(name='John', lastname='Dow'))
self.model_user.refresh()
self.model_user.sort = (1, 3)
self.assertEqual(self.model_user.sort, (1, 3))
self.model_user.refresh()
def test_rowCount(self):
self.assertEqual(self.model_user.rowCount(), 0)
self.session.add(model.User(name='John', lastname='Dow'))
self.model_user.refresh()
self.assertEqual(self.model_user.rowCount(), 1)
# Set up the session
sm = orm.sessionmaker(bind=engine,
autoflush=True,
autocommit=True,
expire_on_commit=True)
session = orm.scoped_session(sm)
if not session.query(model.Club).all():
session.add(model.Club(name='VfB Stuttgart', short='VfB'))
if not session.query(model.Age).all():
session.add(model.Age(name='Best of All', short='Elite'))
if not session.query(model.Bow).all():
session.add(model.Bow(name='Longbow', short='LB'))
if not session.query(model.User).all():
session.add(
model.User(name='John', lastname='Dow', club_id=1, age_id=1, bow_id=1))
print(session.query(model.Age).first())
print(session.query(model.Bow).first())
print(session.query(model.User).first())
#session.add(model.Club(name='fakename', short='fake'))
#session.add(model.Age(name='fakename', short='fake'))
print(session.query(model.Club).first())
session.query(model.Club).first().short = 'asdydfedf'
#session.commit()
print(session.query(model.Age).all())
print(session.query(model.Club).all())