def setUp(self):
welcomeNote = Note()
welcomeNote.name = "Welcome to Note Space!"
welcomeNote.desc = "Introduction and Quick guide"
welcomeNote.owner = "Note Space Team"
welcomeNote.save()
econNote = Note()
econNote.name = "Basic Economics"
econNote.subject = "Economics"
econNote.desc = "This note is about demands, suplies and how market works."
econNote.owner = "Susan"
econNote.save()
img1 = Image()
img1.index = 1
img1.note = econNote
img1.image = SimpleUploadedFile(
name='IMG_0809.JPG',
content=open(
"C:/Users/B/Desktop/NoteSpace/static/TestNotes/Econ/IMG_0809.JPG",
'rb').read(),
content_type='image/jpeg')
img1.save()
img2 = Image()
img2.index = 2
img2.note = econNote
img2.image = SimpleUploadedFile(
name='IMG_0810.JPG',
content=open(
"C:/Users/B/Desktop/NoteSpace/static/TestNotes/Econ/IMG_0810.JPG",
'rb').read(),
content_type='image/jpeg')
img2.save()
img3 = Image()
img3.index = 3
img3.note = econNote
img3.image = SimpleUploadedFile(
name='IMG_0811.JPG',
content=open(
"C:/Users/B/Desktop/NoteSpace/static/TestNotes/Econ/IMG_0811.JPG",
'rb').read(),
content_type='image/jpeg')
img3.save()
img4 = Image()
img4.index = 4
img4.note = econNote
img4.image = SimpleUploadedFile(
name='IMG_0812.JPG',
content=open(
"C:/Users/B/Desktop/NoteSpace/static/TestNotes/Econ/IMG_0812.JPG",
'rb').read(),
content_type='image/jpeg')
img4.save()
User.objects.create_user("smith", "*****@*****.**", "1234")
self.browser = webdriver.Firefox()
def test_database_can_search_by_similar(self):
from django.db.models import Q
note1_namesearch = Note()
note1_namesearch.name = "django"
note1_namesearch.save()
note2_descsearch = Note()
note2_descsearch.name = "writing unit test"
note2_descsearch.desc = "for django"
note2_descsearch.save()
Tag_django = Tag()
Tag_django.title = "django"
Tag_django.save()
note3_tagsearch = Note()
note3_tagsearch.name = "Html Template Tags"
note3_tagsearch.save()
note3_tagsearch.tags.add(Tag_django)
note3_tagsearch.save()
note4_ownersearch = Note()
note4_ownersearch.name = "FAQ"
note4_ownersearch.owner = "Django official"
note4_ownersearch.save()
search_result = list(
Note.objects.filter(
Q(name__icontains="django") | Q(desc__icontains="django")
| Q(tags__title__icontains="django")
| Q(owner__icontains="django")))
print(search_result)
self.assertGreaterEqual(len(search_result), 4)
self.assertIn(note1_namesearch, search_result)
self.assertIn(note2_descsearch, search_result)
self.assertIn(note3_tagsearch, search_result)
self.assertIn(note4_ownersearch, search_result)
search_result = Note.objects.filter(
name__trigram_similar="django").filter(
desc__trigram_similar="django")
