def test_get_and_parse_data(self): with app.app_context(): # control data user = User(email=u'*****@*****.**') post1 = Post(title=u'Post 1', content=u'Lorem...', author_id=1) post2 = Post(title=u'Post 2', content=u'Ipsum...', author_id=1) # feed user table self.db.session.add(user) self.db.session.commit() # feed post table self.db.session.add(post1) self.db.session.add(post2) self.db.session.commit() # get data alchemy = AlchemyDumpsDatabase() data = alchemy.get_data() # parse data parsed_user = alchemy.parse_data(data['User']) parsed_posts = alchemy.parse_data(data['Post']) # assert self.assertEqual(len(parsed_user), 1) self.assertEqual(len(parsed_posts), 2) self.assertEqual(user.email, parsed_user[0].email) self.assertEqual(post1.title, parsed_posts[0].title) self.assertEqual(post2.content, parsed_posts[1].content)
def test_mapped_classes(self): with app.app_context(): alchemy = AlchemyDumpsDatabase() classes = alchemy.get_mapped_classes() self.assertIn(User, classes) self.assertIn(Post, classes) self.assertEqual(len(classes), 2)
def test_mapped_classes(self): with app.app_context(): alchemy = AlchemyDumpsDatabase() classes = alchemy.get_mapped_tables() self.assertIn(User, classes) self.assertIn(Post, classes) self.assertEqual(len(classes), 2)