def test_generic_search(self):
""" Test the generic search entity. """
self.save_project_entities()
self.save_address_entity()
self.save_profile_entity()
sleep(0.25)
search = SearchProvider()
address = search.search_entity(AddressEntity.get_index(), state="TX")
self.assertIsNotNone(address)
self.assertEquals(address.city, "Round Rock")
profile = search.search_entity(ProfileEntity.get_index(), user_id=1)
self.assertIsNotNone(profile)
self.assertEquals(profile.last_name, "Blouse")
project = search.search_entity(ProjectEntity.get_index(), task="work")
self.assertIsNotNone(project)
self.assertTrue("Test project description" in project.description)
project_list = search.search_entities(ProjectEntity.get_index(), user_id=1, task="work")
for project in project_list:
self.assertEquals(project.user_id, 1)
self.assertEquals(project.description, "Test project description")
project_list = search.search_entities(ProjectEntity.get_index(), 0, 1, user_id=1, task="work")
self.assertEquals(len(project_list), 1)
def test_entities(self):
""" Test the to_json of models. """
profile = ProfileEntity()
profile.first_name = "Jared"
profile.last_name = "Blouse"
profile.business_phone = "123-456-7890"
profile.home_phone = "123-456-7890"
profile.email = "*****@*****.**"
profile.fax = "123-456-7890"
profile.mobile_phone = "123-456-7890"
project_list = list()
project1 = ProjectEntity()
project1.description = "Test project"
project1.save()
skill = SkillEntity()
skill.name = "C#"
skill.years_experience = 9
skill.level = 9
skill.save()
project_list.append(project1)
profile.projects = project_list
profile.user_id = 5
profile.url = "www.corpseware.com"
self.assertTrue(profile.save())
def save_project_entities(self):
""" Save a test project entity. """
project = ProjectEntity()
project.user_id = 1
project.description = "Test project description"
project.situation = "Needed to work"
project.task = "Work"
project.action = "Completed the work"
project.result = "Awesome job"
self.assertTrue(project.save())
project2 = ProjectEntity()
project2.description = "Test project description2"
project2.situation = "Needed to work2"
project2.task = "Work"
project2.action = "Completed the work2"
project2.result = "Awesome job2"
self.assertTrue(project2.save())
def search_projects(self, **search_criteria):
""" Get the project based on criteria. """
return self._get_entities_by_search(Project.get_index(),
**search_criteria)
