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