def test_guest_portal1(session):
'''Check if configuring different auth_methods gives out proper landing pages'''
site1 = Wifisite.query.filter_by(id=1).first()
mac = randomMAC()
ap_mac = randomMAC()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
db.session.commit()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_multi_login_page(url)
site1.auth_method = AUTH_TYPE_SOCIAL
db.session.commit()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_facebook_login_page(url)
site1.auth_method = AUTH_TYPE_SMS
db.session.commit()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_phone_login_page(url)
site1.auth_method = AUTH_TYPE_EMAIL
db.session.commit()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_email_login_page(url)
site1.auth_method = AUTH_TYPE_VOUCHER
db.session.commit()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_voucher_login_page(url)
def test_guest_portal1(session):
"""Check if configuring different auth_methods gives out proper landing pages"""
site1 = Wifisite.query.filter_by(id=1).first()
mac = randomMAC()
ap_mac = randomMAC()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
db.session.commit()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_multi_login_page(url)
site1.auth_method = AUTH_TYPE_SOCIAL
db.session.commit()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_facebook_login_page(url)
site1.auth_method = AUTH_TYPE_SMS
db.session.commit()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_phone_login_page(url)
site1.auth_method = AUTH_TYPE_EMAIL
db.session.commit()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_email_login_page(url)
site1.auth_method = AUTH_TYPE_VOUCHER
db.session.commit()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_voucher_login_page(url)
def test_authorize_guest2(session):
""" authorize_guest check if parameters are correctly configured """
# create a guest visitor
site1 = Wifisite.query.filter_by(id=1).first()
mac = randomMAC()
ap_mac = randomMAC()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
db.session.commit()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_multi_login_page(url)
guest_track = Guesttrack.query.first()
# authorize session
guest_track.state = GUESTRACK_SOCIAL_AUTH
auth_url = url_for("guest.authorize_guest", track_id=guest_track.track_id)
result = current_app.test_client().get(auth_url, follow_redirects=True)
assert "200 OK" == result.status, "authorize_guest getting:%s instead of 200 OK while trying to View URL:%s" % (
result.status,
url,
)
# check if guest_session is created
guest_session = Guestsession.query.first()
assert isinstance(guest_session, Guestsession), "Guestsession is not created when calling authorize_guest"
assert guest_session.state == GUESTRACK_SOCIAL_AUTH, " guest_session state is not GUESTRACK_SOCIAL_AUTH"
def test_guest_portal4(session):
'''User with demo flag set is visiting
'''
site1 = Wifisite.query.filter_by(unifi_id='site1').first()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
mac = randomMAC()
ap_mac1 = randomMAC()
url1 = get_guest_url(site1,mac,ap_mac1,1)
check_multi_login_page(url1)
guest_track1 = Guesttrack.query.filter( and_(Guesttrack.device_mac==mac,Guesttrack.site_id==site1.id)).first()
guest_device1 = Device.query.filter( and_(Device.mac==mac,Device.site_id==site1.id)).first()
assert 1 == guest_track1.demo,'Demo flag is not set for guest_track for URL:%s'%url1
assert 1 == guest_device1.demo,'Demo flag is not set for guest_device for URL:%s'%url1
def test_guest_portal4(session):
"""User with demo flag set is visiting
"""
site1 = Wifisite.query.filter_by(unifi_id="site1").first()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
mac = randomMAC()
ap_mac1 = randomMAC()
url1 = get_guest_url(site1, mac, ap_mac1, 1)
check_multi_login_page(url1)
guest_track1 = Guesttrack.query.filter(and_(Guesttrack.device_mac == mac, Guesttrack.site_id == site1.id)).first()
guest_device1 = Device.query.filter(and_(Device.mac == mac, Device.site_id == site1.id)).first()
assert 1 == guest_track1.demo, "Demo flag is not set for guest_track for URL:%s" % url1
assert 1 == guest_device1.demo, "Demo flag is not set for guest_device for URL:%s" % url1
def test_guest_portal2(session):
'''Check if guest_track creations are done properly or not'''
site1 = Wifisite.query.filter_by(id=1).first()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
db.session.commit()
mac = randomMAC()
ap_mac = randomMAC()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_multi_login_page(url)
# test if the device is created for this user
test_device = Device.query.filter( and_(Device.mac==mac,Device.site_id==site1.id)).first()
assert isinstance(test_device, Device), "Device is not created when a new user visits"
#same user visit multiple times without session expiry, no new sessions should be created
time.sleep(1)
check_multi_login_page(url)
check_multi_login_page(url)
check_multi_login_page(url)
check_multi_login_page(url)
num_device = Device.query.filter( and_(Device.mac==mac,Device.site_id==site1.id)).count()
assert num_device == 1 , "User visiting twice without session expiry: Device shouldn't be created twice"
#
assert 5 == Guesttrack.query.count() , "User visiting without session expiry: Guesttrack should be created eachtime"
def test_guest_portal2(session):
"""Check if guest_track creations are done properly or not"""
site1 = Wifisite.query.filter_by(id=1).first()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
db.session.commit()
mac = randomMAC()
ap_mac = randomMAC()
url = get_guest_url(site1, mac, ap_mac, demo=0)
check_multi_login_page(url)
# test if the device is created for this user
test_device = Device.query.filter(and_(Device.mac == mac, Device.site_id == site1.id)).first()
assert isinstance(test_device, Device), "Device is not created when a new user visits"
# same user visit multiple times without session expiry, no new sessions should be created
time.sleep(1)
check_multi_login_page(url)
check_multi_login_page(url)
check_multi_login_page(url)
check_multi_login_page(url)
num_device = Device.query.filter(and_(Device.mac == mac, Device.site_id == site1.id)).count()
assert num_device == 1, "User visiting twice without session expiry: Device shouldn't be created twice"
#
assert 5 == Guesttrack.query.count(), "User visiting without session expiry: Guesttrack should be created eachtime"
def test_authorize_guest2(session):
''' authorize_guest check if parameters are correctly configured '''
#create a guest visitor
site1 = Wifisite.query.filter_by(id=1).first()
mac = randomMAC()
ap_mac = randomMAC()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
db.session.commit()
url = get_guest_url(site1,mac,ap_mac,demo=0)
check_multi_login_page(url)
guest_track = Guesttrack.query.first()
#authorize session
guest_track.state = GUESTRACK_SOCIAL_AUTH
auth_url = url_for('guest.authorize_guest',track_id =guest_track.track_id)
result = current_app.test_client().get(auth_url,follow_redirects=True)
assert '200 OK' == result.status, 'authorize_guest getting:%s instead of 200 OK while trying to View URL:%s'%(result.status,url)
#check if guest_session is created
guest_session = Guestsession.query.first()
assert isinstance(guest_session,Guestsession) ,'Guestsession is not created when calling authorize_guest'
assert guest_session.state == GUESTRACK_SOCIAL_AUTH, " guest_session state is not GUESTRACK_SOCIAL_AUTH"
def test_guest_portal3(session):
'''Same user visiting multiple sites
'''
site1 = Wifisite.query.filter_by(unifi_id='site1').first()
site2 = Wifisite.query.filter_by(unifi_id='site2').first()
site3 = Wifisite.query.filter_by(unifi_id='site3').first()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
site2.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
site3.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL +AUTH_TYPE_VOUCHER
mac = randomMAC()
ap_mac1 = randomMAC()
ap_mac2 = randomMAC()
ap_mac3 = randomMAC()
url1 = get_guest_url(site1,mac,ap_mac1,demo=0)
url2 = get_guest_url(site2,mac,ap_mac2,demo=0)
url3 = get_guest_url(site3,mac,ap_mac3,demo=0)
check_multi_login_page(url1)
check_multi_login_page(url1)
check_multi_login_page(url2)
check_multi_login_page(url2)
check_multi_login_page(url3)
check_multi_login_page(url3)
assert 2 == Guesttrack.query.filter_by(site_id=site1.id).count(), "Same device visiting 3 sites twice, 2 Guesttrack PER SITE needed"
assert 2 == Guesttrack.query.filter_by(site_id=site2.id).count(), "Same device visiting 3 sites twice, 2 Guesttrack PER SITE needed"
assert 2 == Guesttrack.query.filter_by(site_id=site3.id).count(), "Same device visiting 3 sites twice, 2 Guesttrack PER SITE needed"
assert 3 == Device.query.count(), "Same device visiting 3 sites, 3 Device needed"
def test_guest_portal3(session):
"""Same user visiting multiple sites
"""
site1 = Wifisite.query.filter_by(unifi_id="site1").first()
site2 = Wifisite.query.filter_by(unifi_id="site2").first()
site3 = Wifisite.query.filter_by(unifi_id="site3").first()
site1.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
site2.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
site3.auth_method = AUTH_TYPE_SOCIAL + AUTH_TYPE_SMS + AUTH_TYPE_EMAIL + AUTH_TYPE_VOUCHER
mac = randomMAC()
ap_mac1 = randomMAC()
ap_mac2 = randomMAC()
ap_mac3 = randomMAC()
url1 = get_guest_url(site1, mac, ap_mac1, demo=0)
url2 = get_guest_url(site2, mac, ap_mac2, demo=0)
url3 = get_guest_url(site3, mac, ap_mac3, demo=0)
check_multi_login_page(url1)
check_multi_login_page(url1)
check_multi_login_page(url2)
check_multi_login_page(url2)
check_multi_login_page(url3)
check_multi_login_page(url3)
assert (
2 == Guesttrack.query.filter_by(site_id=site1.id).count()
), "Same device visiting 3 sites twice, 2 Guesttrack PER SITE needed"
assert (
2 == Guesttrack.query.filter_by(site_id=site2.id).count()
), "Same device visiting 3 sites twice, 2 Guesttrack PER SITE needed"
assert (
2 == Guesttrack.query.filter_by(site_id=site3.id).count()
), "Same device visiting 3 sites twice, 2 Guesttrack PER SITE needed"
assert 3 == Device.query.count(), "Same device visiting 3 sites, 3 Device needed"
