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)