def test_create(self): self.assertEqual(len(Section.query.all()), 0) section = Section.create('sports') db.session.add(section) db.session.commit() self.assertEqual(len(Section.query.all()), 1) section = Section.create('computer') db.session.add(section) db.session.commit() self.assertEqual(len(Section.query.all()), 2)
def setUp(self): db.drop_all() db.create_all() user = User.create('*****@*****.**', 'strong_password') admin = User.create('admin', 'admin', admin=True) self.user_id = user.user_id db.session.add(user) db.session.add(admin) db.session.add(Section.create("sport")) db.session.commit() with self.app.test_client() as client: response = client.post('/api/tokens', json={ 'email': '*****@*****.**', 'password': '******', }) self.token = response.json['token'] response = client.post( '/api/posts', json={ 'post_name': "today's sports", 'post_time': datetime.now(), 'section_name': 'sport', 'context': "sport is great!", }, headers={'Authorization': "Token " + self.token}) self.post_id = response.json['post_id']
def test_create(self): self.assertEqual(len(BlockItem.query.all()), 0) user = User.create('*****@*****.**', 'password') section = Section.create('sports') block = BlockItem(user_id=user.user_id) section.blocks.append(block) db.session.add(user) db.session.add(section) db.session.commit() self.assertEqual(len(BlockItem.query.all()), 1)
def test_search(self): user = User.create('*****@*****.**', 'password') user_id = user.user_id section = Section.create('sports') block = BlockItem(user_id=user_id) section.blocks.append(block) db.session.add(user) db.session.add(section) db.session.commit() section = Section.query.filter_by(section_name='sports').first() block = BlockItem.query.with_parent(section).filter_by( user_id=user_id).first() self.assertIsNotNone(block) self.assertEqual(block.user_id, user_id)
def test_create_after_commit(self): self.assertEqual(len(BlockItem.query.all()), 0) user = User.create('admin', 'admin', admin=True) xua = User.create('xua', 'passwd') section = Section.create('sports') db.session.add(section) db.session.add(user) db.session.add(xua) db.session.commit() section = Section.query.filter_by(section_name='sports').first() user = User.query.filter_by(email='xua').first() block = BlockItem(user_id=user.user_id) section.blocks.append(block) db.session.commit() self.assertEqual(len(BlockItem.query.all()), 1)
def setUp(self): db.drop_all() db.create_all() user = User.create('*****@*****.**', 'strong_password') # post = Post.create('sports', datetime.now(), "do some sport!") self.user_id = user.user_id db.session.add(user) db.session.add(Section.create("sport")) db.session.commit() with self.app.test_client() as client: response = client.post('/api/tokens', json={ 'email': '*****@*****.**', 'password': '******', }) self.token = response.json['token']
def test_create(self): user = User.create('xua@wustl', 'password') section = Section.create('sports') time = datetime.now() post = Post.create("today's sports", time, "hey, let's discuss today's sports!") post_id = post.post_id user.posts.append(post) section.posts.append(post) db.session.add(user) db.session.add(section) db.session.commit() post = Post.query.filter_by(post_id=post_id).first() self.assertEqual(post.post_time, time) self.assertEqual(post.post_name, "today's sports") self.assertEqual(post.poster_email, 'xua@wustl') self.assertEqual(post.section_name, 'sports')
def test_create_comment(self): user = User.create('xua@wustl', 'password') section = Section.create('sports') post = Post.create("today's sports", datetime.now(), "hey, let's discuss today's sports!") post_id = post.post_id first = Comment.create(datetime.now(), "it sounds good!") second = Comment.create(datetime.now(), "I don't like sports") post.comments.append(first) post.comments.append(second) user.comments.append(first) user.comments.append(second) user.posts.append(post) section.posts.append(post) db.session.add(user) db.session.add(section) db.session.commit() post = Post.query.filter_by(post_id=post_id).first() self.assertEqual(len(post.comments), 2)
def setUp(self): db.drop_all() db.create_all() user = User.create('*****@*****.**', 'strong_password') # post = Post.create('sports', datetime.now(), "do some sport!") self.user_id = user.user_id db.session.add(user) db.session.add(Section.create("sport")) db.session.commit() with self.app.test_client() as client: response = client.post('/api/tokens', json={ 'email': '*****@*****.**', 'password': '******', }) self.token = response.json['token'] response = client.post('/api/posts', json={ 'post_name': "today's sports", 'post_time': datetime.now(), 'section_name': 'sport', 'context': "sport is great!", }, headers={'Authorization': "Token " + self.token}) self.post_id = response.json['post_id'] client.post('/api/comments', json={ 'post_id': self.post_id, 'comment_time': datetime(2018, 7, 10, 12, 55, 0), 'context': 'this is great!', }, headers={'Authorization': "Token " + self.token}) client.post('/api/comments', json={ 'post_id': self.post_id, 'comment_time': datetime(2018, 8, 10, 12, 55, 0), 'context': 'this is worse!', }, headers={'Authorization': "Token " + self.token}) client.post('/api/comments', json={ 'post_id': self.post_id, 'comment_time': datetime(2018, 9, 10, 12, 55, 0), 'context': 'this is common!', }, headers={'Authorization': "Token " + self.token})