def test_facilities():
Database.initialize()
# add company
status = Facilities.is_company_exist('google')
assert status != Facilities.add_company('google', 'matam')
status = Facilities.is_company_exist('yahoo')
assert status != Facilities.add_company('yahoo', 'matam')
status = Facilities.is_company_exist('microsoft')
assert status != Facilities.add_company('microsoft', 'matam')
status = Facilities.is_company_exist('vmware')
assert status != Facilities.add_company('vmware', 'matam')
status = Facilities.is_company_exist('intel')
assert status != Facilities.add_company('intel', 'matam')
# add facilities to each company
status = Facilities.is_facility_exist('google', 'carmel')
assert status != Facilities.add_facility('google', 'carmel')
status = Facilities.is_facility_exist('yahoo', 'carmel')
assert status != Facilities.add_facility('yahoo', 'carmel')
status = Facilities.is_facility_exist('microsoft', 'carmel')
assert status != Facilities.add_facility('microsoft', 'carmel')
status = Facilities.is_facility_exist('vmware', 'carmel')
assert status != Facilities.add_facility('vmware', 'carmel')
status = Facilities.is_facility_exist('intel', 'carmel')
assert status != Facilities.add_facility('intel', 'carmel')
assert 'carmel' in Facilities.get_facilities('intel')
assert 'matam' in Facilities.get_facilities('intel')
assert 'carmel' in Facilities.get_facilities('vmware')
assert 'matam' in Facilities.get_facilities('vmware')
Facilities.remove_facility('vmware', 'carmel')
assert 'carmel' not in Facilities.get_facilities('vmware')
def user_register(cls, email, password, username, _id, role, permission,
company, facility):
if not Facilities.is_company_exist(company):
return False, "company dose'nt exist"
user = User.get_by_email(email)
print(email)
if not cls.check_id(_id):
return False, "bad number ID"
if user is None:
# User dose'nt exist, create new user
new_user = User(email, username, password, _id, role, permission,
company, facility)
try:
new_user.save_to_mongodb()
except Exception as e:
return False, str(e)
return True, ''
else:
# User already exist
return False, "user email already exist"