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"