def test_submit_form_for_new_resource(self):
ard = Add_Resource_DataService("james")
rd = Resource_DAO()
print "before: " + str(rd.get_count_as_int())
ard.submit_form_for_new_resource("test", "testmodel", 1.1, 2.2, 3.5, 2,
"hour", [3, 4, 5], ["aaa", "bbb"])
print "after : " + str(rd.get_count_as_int())
for i in range(0, 50):
User_DAO().insert("muni" + str(i), "muniinfo" + str(i), "pass" + str(i))
UserType_DAO.insert_record_to_municipalities("muni" + str(i), 1000 + i)
for i in range(0, 50):
User_DAO().insert("tomindi" + str(i), "tomindiinfo" + str(i),
"pass" + str(i))
UserType_DAO.insert_record_to_individual("tomindi" + str(i),
"tomindijobtitle" + str(i),
"2016-11-14")
for i in range(1, 50):
usernmae = "tomcompany" + 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"]
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]
def test_get_count(self):
rsd = Resource_DAO()
result = rsd.get_count_as_int()
print result
assert isinstance(result, int)
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()
