3446026555
The license.txt file has been replaced with LICENSE for quite a while now. INAL but it didn't seem accurate to say "hey, checkout license.txt although there's no such file". Apart from this, there were inconsistencies in the headers altogether...this change brings consistency.
128 lines
No EOL
4 KiB
Python
128 lines
No EOL
4 KiB
Python
# Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
import unittest
|
|
from frappe.search.full_text_search import FullTextSearch
|
|
|
|
class TestFullTextSearch(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
index = get_index()
|
|
index.build()
|
|
self.index = index
|
|
|
|
def test_search_term(self):
|
|
# Search Wikipedia
|
|
res = self.index.search("multilingual online encyclopedia")
|
|
self.assertEqual(res[0], 'site/wikipedia')
|
|
|
|
res = self.index.search("Linux kernel")
|
|
self.assertEqual(res[0], 'os/linux')
|
|
|
|
res = self.index.search("Enterprise Resource Planning")
|
|
self.assertEqual(res[0], 'sw/erpnext')
|
|
|
|
def test_search_limit(self):
|
|
res = self.index.search("CommonSearchTerm")
|
|
self.assertEqual(len(res), 5)
|
|
|
|
res = self.index.search("CommonSearchTerm", limit=3)
|
|
self.assertEqual(len(res), 3)
|
|
|
|
res = self.index.search("CommonSearchTerm", limit=20)
|
|
self.assertEqual(len(res), 5)
|
|
|
|
def test_search_scope(self):
|
|
# Search outside scope
|
|
res = self.index.search("multilingual online encyclopedia", scope=["os"])
|
|
self.assertEqual(len(res), 0)
|
|
|
|
# Search inside scope
|
|
res = self.index.search("CommonSearchTerm", scope=["os"])
|
|
self.assertEqual(len(res), 2)
|
|
self.assertTrue('os/linux' in res)
|
|
self.assertTrue('os/gnu' in res)
|
|
|
|
def test_remove_document_from_index(self):
|
|
self.index.remove_document_from_index("os/gnu")
|
|
res = self.index.search("GNU")
|
|
self.assertEqual(len(res), 0)
|
|
|
|
def test_update_index(self):
|
|
# Update existing index
|
|
self.index.update_index({
|
|
'name': "sw/erpnext",
|
|
'content': """AwesomeERPNext"""
|
|
})
|
|
|
|
res = self.index.search("CommonSearchTerm")
|
|
self.assertTrue('sw/erpnext' not in res)
|
|
|
|
res = self.index.search("AwesomeERPNext")
|
|
self.assertEqual(res[0], "sw/erpnext")
|
|
|
|
# Update new doc
|
|
self.index.update_index({
|
|
'name': "sw/frappebooks",
|
|
'content': """DesktopAccounting"""
|
|
})
|
|
|
|
res = self.index.search("DesktopAccounting")
|
|
self.assertEqual(res[0], "sw/frappebooks")
|
|
|
|
|
|
|
|
class TestWrapper(FullTextSearch):
|
|
def get_items_to_index(self):
|
|
return get_documents()
|
|
|
|
def get_document_to_index(self, name):
|
|
documents = get_documents()
|
|
for doc in documents:
|
|
if doc["name"] == name:
|
|
return doc
|
|
|
|
def parse_result(self, result):
|
|
return result["name"]
|
|
|
|
|
|
def get_index():
|
|
return TestWrapper("test_frappe_index")
|
|
|
|
def get_documents():
|
|
docs = []
|
|
docs.append({
|
|
'name': "site/wikipedia",
|
|
'content': """Wikipedia is a multilingual online encyclopedia created and maintained
|
|
as an open collaboration project by a community of volunteer editors using a wiki-based editing system.
|
|
It is the largest and most popular general reference work on the World Wide Web. CommonSearchTerm"""
|
|
})
|
|
|
|
docs.append({
|
|
'name': "os/linux",
|
|
'content': """Linux is a family of open source Unix-like operating systems based on the
|
|
Linux kernel, an operating system kernel first released on September 17, 1991, by Linus Torvalds.
|
|
Linux is typically packaged in a Linux distribution. CommonSearchTerm"""
|
|
})
|
|
|
|
docs.append({
|
|
'name': "os/gnu",
|
|
'content': """GNU is an operating system and an extensive collection of computer software.
|
|
GNU is composed wholly of free software, most of which is licensed under the GNU Project's own
|
|
General Public License. GNU is a recursive acronym for "GNU's Not Unix! ",
|
|
chosen because GNU's design is Unix-like, but differs from Unix by being free software and containing no Unix code. CommonSearchTerm"""
|
|
})
|
|
|
|
docs.append({
|
|
'name': "sw/erpnext",
|
|
'content': """ERPNext is a free and open-source integrated Enterprise Resource Planning software developed by
|
|
Frappe Technologies Pvt. Ltd. and is built on MariaDB database system using a Python based server-side framework.
|
|
ERPNext is a generic ERP software used by manufacturers, distributors and services companies. CommonSearchTerm"""
|
|
})
|
|
|
|
docs.append({
|
|
'name': "sw/frappe",
|
|
'content': """Frappe Framework is a full-stack web framework, that includes everything you need to build and
|
|
deploy business applications with Rich Admin Interface. CommonSearchTerm"""
|
|
})
|
|
|
|
return docs |