def test_atualizar_livro(self):
db.drop_all()
db.create_all()
db.session.add(models.Autor('Jose'))
db.session.add(models.Editora('Alfa'))
autor = models.Autor.query.filter_by(nome='Jose').first()
editora = models.Editora.query.filter_by(nome='Alfa').first()
livro = models.Livro(
123456, 'A volta dos que não foram', 'Drama', 1, 2007,
'Esse livro é um drama sobra a volta dos que não foram', 2,
editora.id, autor.id)
db.session.add(livro)
db.session.commit()
livroTest = models.Livro(
123456, 'A volta', 'Drama', 1, 2007,
'Esse livro é um drama sobra a volta dos que não foram', 2,
editora.id, autor.id)
modelsDAO.atualizar_livro(livro.id, livroTest.titulo)
self.assertEqual(compareObjects(livro, livroTest), True)
def test_obter_livro(self):
db.drop_all()
db.create_all()
db.session.add(models.Autor('Jose'))
db.session.add(models.Editora('Alfa'))
autor = models.Autor.query.filter_by(nome='Jose').first()
editora = models.Editora.query.filter_by(nome='Alfa').first()
db.session.add(
models.Livro(
123456, 'A volta dos que não foram', 'Drama', 1, 2007,
'Esse livro é um drama sobra a volta dos que não foram', 2,
editora.id, autor.id))
db.session.commit()
livro = modelsDAO.obter_livros()
self.assertEqual(livro, models.Livro.query.all())