def test_more(self): srd = Search_Result_DataService() resd = Resource_DAO() rep = RepairResource_DAO() reqeust = ResourceRequest_DAO() count_for_test = run_sql_string_statement( "select count(*) from Resource") count = int(count_for_test[0][0]) a = count + 1 b = count + 2 resd.insert_resource(a, "123Name", "dummy", "not available", 14.56, 33.12, "james", 120.4, 3, "hour", "tom", "2015-03-02", "2016-07-03") resd.insert_resource(b, "123Name", "dummy", "in repair", 14.56, 33.12, "james", 120.4, 3, "hour", "tom", "2015-03-02", "2016-07-03") rep.insert_records(b, rep.get_count() + 1, "james", "2015-03-02", "2016-07-03") reqeust.insert_record(1, reqeust.get_count() + 1, "2015-03-02", a) ma = srd.populate_table_data("a", "2# aaa", 1000000, "(6) testinci") assert isinstance(ma, list) print ma
def test_insert_record(self): rsd = Resource_DAO() count_for_test = run_sql_string_statement( "select count(*) from Resource") count = int(count_for_test[0][0]) rsd.insert_resource(count + 1, "123Name", "dummy", "available", 14.56, 33.12, "james", 120.4, 3, "hour", "tom", "2015-03-02", "2016-07-03")
unit = "hour" addi = [1, 2, 3, 4, 5, 6, 7] cap = ["capa", "capb"] if count / 5 == 0: status = "in repair" unit = "time" addi = [14, 13, 5, 6, 7] cap = ["capa", "capb", "capc"] else: status = "not available" unit = "km" addi = [11, 12, 8] cap = ["capx"] rsd.insert_resource(count + 1, "ResName" + str(count), "dummymodel", status, 44.44, 55.55, usernmae, 250.24, count % 14 + 1, unit, "tomcompany" + str(i - 1), "2015-03-02", "2016-07-03") Additional_DAO(count + 1).insert_additional_esf_list(addi) Capability_DAO(count + 1).insert_capability(cap) for i in range(1, 50): usernmae = "tomgov" + str(i) rsd = Resource_DAO() count = rsd.get_count_as_int() + 1 if count / 3 == 0: status = "available" unit = "hour" addi = [1, 2, 3, 4, 5, 6, 7] cap = ["capa", "capb"]
class Add_Resource_DataService(): def __init__(self, username): logging.info("start the Add_Resource_DataService") self.resource_dao = Resource_DAO() self.new_generated_resource_id = self.resource_dao.get_count_as_int() + 100000 self.capability_dao = Capability_DAO(self.new_generated_resource_id) self.additional_esf_dao = Additional_DAO(self.new_generated_resource_id) self.username = username self.esf_list_dao = ESF_DAO() self.usertype_dao = UserType_DAO() self.costunitdao = CostUnit_DAO() def get_new_generated_resource_id(self): ''' @rtype: int ''' return self.new_generated_resource_id def submit_form_for_new_resource(self, ResourceName, model, Latitude, Longitude, CostPerValue, EsfID, CostUnit, additionalESFList, capabilitiesList): ''' @rtype: bool ''' assert isinstance(self.new_generated_resource_id, int) assert isinstance(ResourceName, str) assert isinstance(model, str) # assert isinstance(CurrentStatus, str) assert isinstance(Latitude, float) assert isinstance(Longitude, float) # assert isinstance(Owner, str) assert isinstance(CostPerValue, float) assert isinstance(EsfID, int) assert isinstance(CostUnit, str) # assert isinstance(CurrentUser, str) # assert isinstance(StartDate, str) # assert isinstance(DueDate, str) assert isinstance(additionalESFList, list) assert isinstance(capabilitiesList, list) STATUS = ["available", "not available", "in repair"] # insert into the resource table self.resource_dao.insert_resource(self.new_generated_resource_id, ResourceName, model, STATUS[0], Latitude, Longitude, self.username, CostPerValue, EsfID, CostUnit, self.username, time.strftime("%y-%m-%d"), time.strftime("%y-%m-%d")) # insert into the additional esf table add_esf_dao = Additional_DAO(self.new_generated_resource_id) add_esf_dao.insert_additional_esf_list(additionalESFList) # insert into the cap table capdao = Capability_DAO(self.new_generated_resource_id) capdao.insert_capability(capabilitiesList) def populate_owner_title(self): ''' @rtype: str ''' usertype = self.usertype_dao.get_user_type_by_username(self.username) MUNICIPALITIES = "municipalities" GOVERNMENT_AGENCIES = "government_agencies" COMPANIES = "companies" INDIVIDUALS = "individuals" if usertype == MUNICIPALITIES: return self.usertype_dao.get_title_info_from_municipalities(self.username) if usertype == GOVERNMENT_AGENCIES: return self.usertype_dao.get_title_info_from_government(self.username) if usertype == COMPANIES: return self.usertype_dao.get_title_info_from_company(self.username) if usertype == INDIVIDUALS: return self.usertype_dao.get_title_info_from_individuals(self.username)[0] return "" def populate_esf_list(self): ''' @rtype: dict ''' return self.esf_list_dao.get_all_esf_records() def populate_costunit_list(self): return self.costunitdao.get_all_CostUnits()