def test_document_markdown(self):
user = User(username='******', password='******')
user.save()
document1 = Document(title='abcd',
slug='a',
user=user,
content='abcd')
document1.save()
saved_document1 = Document.objects.first()
self.assertEqual(saved_document1.markdown(), '<p>abcd</p>')
def test_document_category(self):
user = User(username='******', password='******')
user.save()
category = Category(name='Book', slug='book')
category.save()
document1 = Document(title='abcd',
slug='a',
user=user,
category=category,
content='abcd')
document1.save()
saved_document1 = Document.objects.first()
self.assertEqual(saved_document1.category.name, 'Book')
def test_document_tags(self):
user = User(username='******', password='******')
user.save()
tag1 = Tag(name='book', slug='book')
tag1.save()
tag2 = Tag(name='programming', slug='programming')
tag2.save()
document = Document(title='abcd',
slug='a',
user=user,
content='abcd')
document.save()
document.tags.add(tag1)
document.tags.add(tag2)
saved_document = Document.objects.first()
self.assertEqual(len(saved_document.tags.all()), 2)
def test_saving_and_retrieving_documents(self):
user = User(username='******', password='******')
user.save()
document1 = Document(title='abcd', slug='a', user=user)
document1.save()
document2 = Document(title='abcde', slug='b', user=user)
document2.save()
saved_documets = Document.objects.all()
self.assertEqual(saved_documets.count(), 2)
saved_document1 = saved_documets[0]
saved_document2 = saved_documets[1]
self.assertEqual(saved_document1.title, 'abcd')
self.assertEqual(saved_document2.title, 'abcde')
