def test_content(self):
post = Post(title='hello world',
text='world',
slug='hello-world',
author_id=1)
self.assertEqual(post.content, '<p>%s</p>\n' % post.text)
post.text = 'new world'
self.assertEqual(post.content, '<p>%s</p>\n' % post.text)
def test_create_post(self):
post = Post(title='hello', text='world', slug='hello', author_id=1)
op = PostOperator(self.session)
op.create_post(post)
self.assertEqual(op.get_post(post.id), post)
# same slug is not allowed
another_post = Post(title='hello',
text='world',
slug='hello',
author_id=1)
self.assertRaises(ModelError, op.create_post, another_post)
def test_date(self):
post = Post(title='hello world', text='world',
slug='hello-world', author_id=1)
self.assertIsNone(post.created_at)
self.assertIsNone(post.updated_at)
self.session.add(post)
self.session.commit()
self.assertIsNotNone(post.created_at)
self.assertIsNone(post.updated_at)
post.text = 'new world'
self.session.commit()
self.assertIsNotNone(post.updated_at)
def test_date(self):
post = Post(title='hello world',
text='world',
slug='hello-world',
author_id=1)
self.assertIsNone(post.created_at)
self.assertIsNone(post.updated_at)
self.session.add(post)
self.session.commit()
self.assertIsNotNone(post.created_at)
self.assertIsNone(post.updated_at)
post.text = 'new world'
self.session.commit()
self.assertIsNotNone(post.updated_at)
def test_query(self):
post = Post(title='a title',
text='the first post',
slug='a-title',
author_id=1)
self.session.add(post)
self.session.commit()
result = self.session.query(Post).filter_by(title='a title').one()
self.assertEqual(result.title, post.title)
post = Post(title='a title',
text='the second post',
slug='a-title',
author_id=1)
self.session.add(post)
self.session.commit()
result = self.session.query(Post).join(PostText)\
.filter(PostText.text=='the second post').one()
self.assertEqual(result.text, post.text)
def test_slug(self):
post = Post(title='hello world',
text='world',
slug='hello-world',
author_id=1)
self.assertEqual(post.slug, 'hello-world')
# invalid slug
def _set_slug(slug):
post.slug = slug
self.assertRaises(ModelError, _set_slug, 'this contains spaces')
self.assertRaises(ModelError, _set_slug, 'this-contains-newline\n')
self.assertRaises(ModelError, _set_slug, 'this-contains-newline\r')
self.assertRaises(ModelError, _set_slug, 'this-contains/slash')
self.assertRaises(ModelError, _set_slug, 'this-contains-?')
self.assertRaises(ModelError, _set_slug, '')
self.assertRaises(ModelError, _set_slug, ' ')
self.assertRaises(ModelError, _set_slug, '\t')
self.assertEqual(
post.permalink, '%d/%d/%s' %
(datetime.utcnow().year, datetime.utcnow().month, post.slug))
def test_tag(self):
clojure = Post(title='clojure lisp',
text='',
slug='clojure-lisp',
author_id=1)
scheme = Post(title='scheme lisp',
text='',
slug='scheme-lisp',
author_id=1)
# post not added to session, raise error
self.assertRaises(RuntimeError, clojure.add_tags, ['clojure'])
self.assertRaises(RuntimeError, clojure.remove_tags, ['clojure'])
self.assertRaises(RuntimeError, clojure.set_tags, ['clojure'])
self.assertRaises(RuntimeError, clojure.clear_tags)
self.session.add(clojure)
self.session.add(scheme)
# add tags
# post tags
# clojure: Clojure, LISP
# scheme: Scheme, LISP
self.assertEqual(clojure.add_tags(['Clojure'])[0].name, 'Clojure')
self.assertEqual(clojure.add_tags(['LISP'])[0].name, 'LISP')
self.assertEqual(set(clojure.tags), {'Clojure', 'LISP'})
self.assertEqual(scheme.add_tags(['Scheme'])[0].name, 'Scheme')
self.assertEqual(scheme.add_tags(['SCHEME']), []) # no new tag added
self.assertEqual(scheme.add_tags(['scheme']), []) # no new tag added
self.assertEqual(scheme.add_tags(['lisp'])[0].name, 'LISP')
self.assertEqual(set(scheme.tags), {'Scheme', 'LISP'})
self.assertEqual(set(clojure.tags), {'Clojure', 'LISP'})
# remove tags
scheme.remove_tags(['SCHEME'])
self.assertIsNone(
self.session.query(Tag).filter_by(name='Scheme').first())
scheme.remove_tags(['lisp'])
self.assertEqual(
self.session.query(Tag).filter_by(name='LISP').first().name,
'LISP')
self.assertEqual(scheme.tags, [])
# clear tags
clojure.clear_tags()
self.assertEqual(clojure.tags, [])
self.assertIsNone(
self.session.query(Tag).filter_by(name='Clojure').first())
self.assertIsNone(self.session.query(Tag).first())
scheme.set_tags(['SCHEME', 'LISP', 'Scheme', 'Lisp'])
self.assertEqual(
set(tag.name for tag in self.session.query(Tag).all()),
{'SCHEME', 'LISP'})
self.assertEqual(scheme.set_tags(['scheme', 'lisp', 'scheme', 'lisp']),
([], [])) # add none, remove none
def test_get_posts(self):
op = PostOperator(self.session)
# create post
post = Post(title='hello',
text='world',
slug='hello-world',
author_id=1)
op.create_post(post)
self.assertEqual(op.get_post(post.id), post)
# get public posts
haskell = Post(title='haskell-2012',
text='world3',
slug='hehe',
author_id=1)
haskell.created_at = datetime(year=2012, month=4, day=29)
op.create_post(haskell)
haskell.add_tags(['haskell', 'fp'])
scheme = Post(title='scheme-2010',
text='world2',
slug='haha',
author_id=1)
scheme.created_at = datetime(year=2010, month=1, day=16)
op.create_post(scheme)
scheme.add_tags(['scheme', 'fp'])
clojure = Post(title='clojure-2009',
text='world1',
slug='haha',
author_id=1)
clojure.created_at = datetime(year=2009, month=12, day=13)
op.create_post(clojure)
clojure.add_tags(['clojure', 'fp'])
posts, more = op.get_public_posts()
self.assertEqual(4, len(posts))
self.assertEqual(posts, [post, haskell, scheme, clojure])
self.assertFalse(more) # no more
self.assertEqual(set([str(tag) for tag in op.get_public_tags()]),
{'clojure', 'fp', 'scheme', 'haskell'})
op.trash_post(post)
posts, more = op.get_public_posts()
self.assertEqual(posts, [haskell, scheme, clojure])
self.assertFalse(more)
# scheme will be removed from public tags
op.trash_post(scheme)
self.assertEqual(set([tag.name for tag in op.get_public_tags()]),
{'clojure', 'fp', 'haskell'})
self.assertEqual(set([str(tag) for tag in op.get_trash_tags()]),
{'scheme', 'fp'})
def test_tag(self):
clojure = Post(title='clojure lisp', text='',
slug='clojure-lisp', author_id=1)
scheme = Post(title='scheme lisp', text='',
slug='scheme-lisp', author_id=1)
# post not added to session, raise error
self.assertRaises(RuntimeError, clojure.add_tags, ['clojure'])
self.assertRaises(RuntimeError, clojure.remove_tags, ['clojure'])
self.assertRaises(RuntimeError, clojure.set_tags, ['clojure'])
self.assertRaises(RuntimeError, clojure.clear_tags)
self.session.add(clojure)
self.session.add(scheme)
# add tags
# post tags
# clojure: Clojure, LISP
# scheme: Scheme, LISP
self.assertEqual(clojure.add_tags(['Clojure'])[0].name, 'Clojure')
self.assertEqual(clojure.add_tags(['LISP'])[0].name, 'LISP')
self.assertEqual(set(clojure.tags), {'Clojure', 'LISP'})
self.assertEqual(scheme.add_tags(['Scheme'])[0].name, 'Scheme')
self.assertEqual(scheme.add_tags(['SCHEME']), []) # no new tag added
self.assertEqual(scheme.add_tags(['scheme']), []) # no new tag added
self.assertEqual(scheme.add_tags(['lisp'])[0].name, 'LISP')
self.assertEqual(set(scheme.tags), {'Scheme', 'LISP'})
self.assertEqual(set(clojure.tags), {'Clojure', 'LISP'})
# remove tags
scheme.remove_tags(['SCHEME'])
self.assertIsNone(self.session.query(Tag).filter_by(name='Scheme').first())
scheme.remove_tags(['lisp'])
self.assertEqual(self.session.query(Tag).filter_by(name='LISP').first().name, 'LISP')
self.assertEqual(scheme.tags, [])
# clear tags
clojure.clear_tags()
self.assertEqual(clojure.tags, [])
self.assertIsNone(self.session.query(Tag).filter_by(name='Clojure').first())
self.assertIsNone(self.session.query(Tag).first())
scheme.set_tags(['SCHEME', 'LISP', 'Scheme', 'Lisp'])
self.assertEqual(set(tag.name for tag in self.session.query(Tag).all()), {'SCHEME', 'LISP'})
self.assertEqual(scheme.set_tags(['scheme', 'lisp', 'scheme', 'lisp']), ([], [])) # add none, remove none
def test_get_posts(self):
op = PostOperator(self.session)
# create post
post = Post(title='hello', text='world',
slug='hello-world', author_id=1)
op.create_post(post)
self.assertEqual(op.get_post(post.id), post)
# get public posts
haskell = Post(title='haskell-2012', text='world3',
slug='hehe', author_id=1)
haskell.created_at = datetime(year=2012, month=4, day=29)
op.create_post(haskell)
haskell.add_tags(['haskell', 'fp'])
scheme = Post(title='scheme-2010', text='world2',
slug='haha', author_id=1)
scheme.created_at = datetime(year=2010, month=1, day=16)
op.create_post(scheme)
scheme.add_tags(['scheme', 'fp'])
clojure = Post(title='clojure-2009', text='world1',
slug='haha', author_id=1)
clojure.created_at = datetime(year=2009, month=12, day=13)
op.create_post(clojure)
clojure.add_tags(['clojure', 'fp'])
posts, more = op.get_public_posts()
self.assertEqual(4, len(posts))
self.assertEqual(posts, [post, haskell, scheme, clojure])
self.assertFalse(more) # no more
self.assertEqual(set([str(tag) for tag in op.get_public_tags()]),
{'clojure', 'fp', 'scheme', 'haskell'})
op.trash_post(post)
posts, more = op.get_public_posts()
self.assertEqual(posts, [haskell, scheme, clojure])
self.assertFalse(more)
# scheme will be removed from public tags
op.trash_post(scheme)
self.assertEqual(set([tag.name for tag in op.get_public_tags()]),
{'clojure', 'fp', 'haskell'})
self.assertEqual(set([str(tag) for tag in op.get_trash_tags()]),
{'scheme', 'fp'})
def test_content(self):
post = Post(title='hello world', text='world',
slug='hello-world', author_id=1)
self.assertEqual(post.content, '<p>%s</p>\n' % post.text)
post.text = 'new world'
self.assertEqual(post.content, '<p>%s</p>\n' % post.text)