def test_search_tickets_with_organizations(self):
searchOptions = createSearchOptions('Tickets', '_id', '1')
db = TinyDbDAO()
db.ticketTable.search = MagicMock(return_value=[{"hi": "there"}])
db.organizationTable.search = MagicMock(return_value=[{"name": "abc"}])
targetData = db.search(searchOptions)
assert targetData[0]['organizations'] == [{"name": "abc"}]
def test_load_data(self):
searchOptions = createSearchOptions('Organizations', '_id', '101')
db = TinyDbDAO("/tests/testTinyDb.json")
db.loadData()
targetData = db.search(searchOptions)
assert targetData is not None
currentDir = os.getcwd()
os.remove(currentDir + "/tests/testTinyDb.json")
def test_get_search_result(self):
searchOptions = createSearchOptions('Users', 'external_id', '74341f74-9c79-49d5-9611-87ef9b6eb75f')
mockDb = TinyDbDAO()
mockDb.search = MagicMock(return_value=[(0, "fake row", 0.0)])
search = Search(mockDb)
searchResult = search.getSearchResult(searchOptions)
assert searchResult == [(0, "fake row", 0.0)]
def test_list_search_fields(self):
db = TinyDbDAO()
db.organizationTable.all = MagicMock(return_value=[{"hi": "there"}])
db.userTable.all = MagicMock(return_value=[{"hi": "there"}])
db.ticketTable.all = MagicMock(return_value=[{"hi": "there"}])
search = Search(db)
fieldDict = search.listSearchFields()
assert fieldDict["organization"] == ["hi"]
def test_search_users(self):
searchOptions = createSearchOptions('Users', '_id', '1')
db = TinyDbDAO()
db.userTable.search = MagicMock(return_value=[{"hi": "there"}])
targetData = db.search(searchOptions)
assert targetData[0]['users'] == {"hi": "there"}