from itsdangerous.url_safe import URLSafeSerializer
from app import app as flask_app
ts = URLSafeSerializer(flask_app.config["SECRET_KEY"])

data2check_visitors = {
	"/index": {
	"code": 200, "data": b"Nice Tagline"
	},
	"/": {
	"code": 200, "data": b"Nice Tagline"
	},
	"/ContactUs":{
	"code": 200, "data": b"send us a message."
	},
	"/doesnotexists":{
	"code": 404, "data": b"Page Not Found"
	},
	"/logout":{
	"code": 200, "data": b"You have been logged out."
	},
	"/dashboard":{
	"code":401,"data":b"You need to be logged in to access this resource"
	},
	"/signup":{
	"code":200,"data":b"Register your account."
	},
	"/signin":{
	"code":200,"data":b"Sign in to your account."
	},
	"/Simulate500":{
	"code":500,"data":b"Server Could Not Process This."
	},
	"/admin/user/":{
	"code":403,"data":b"Forbidden"
	},
	"/confirm":{
	"code":200,"data":b"Token not provided in URL Parameter"
	},
	"/confirm?confirmation_token=123":{
	"code":200,"data":b"Bad Token Provided"
	}
}

def test_visitors(app, client):
	for page in data2check_visitors:
		res = client.get(page)
		print("Testing %s",page)
		assert res.status_code == data2check_visitors[page]["code"]
		assert data2check_visitors[page]["data"] in res.data 

def test_user_auth_flow(app, client):
	res = client.post("/signup",data=dict(
		email="test@example.com",
		first_name="John",
		password="testpassword",
		), follow_redirects=True)

	assert res.status_code == 200
	assert b"confirm your email" in res.data

	res = client.post("/signin",data=dict(
		email="test@example.com",
		password="testpassword"),
		follow_redirects=True)
	assert res.status_code == 200
	assert b"Please Confirm Your Email First." in res.data

	confirmation_token = ts.dumps("test@example.com",salt="email-confirm-key")
	res = client.get("/confirm?confirmation_token={}".format(confirmation_token),
		follow_redirects=True)
	print(res.data)
	assert b"Succesfully Verified" in res.data	

	res = client.post("/signin",data=dict(
		email="test@example.com",
		password="testpassword"),
		follow_redirects=True)
	assert res.status_code == 200
	assert b"Hi John" in res.data

	res = client.get("/logout", follow_redirects=True)
	assert res.status_code == 200
	assert b"You have been logged out." in res.data

	res = client.post("/signin",data=dict(
		email="test@example.com",
		password="testpassword"),
		follow_redirects=True)
	assert res.status_code == 200
	assert b"Hi John" in res.data

	res = client.get("/signin",follow_redirects=True)
	assert res.status_code == 200
	assert b"Hi John" in res.data

	res = client.get("/signup",follow_redirects=True)
	assert res.status_code == 200
	assert b"Hi John" in res.data

	res = client.get("/admin/user/")
	assert res.status_code == 403 

	res = client.get("/logout")
	res = client.post("/signin",data=dict(
		email="testtest@example.com",
		password="123456"),follow_redirects=True)
	assert res.status_code == 200
	assert b"Incorrect Email" in res.data
	res = client.post("/signin",data=dict(
		email="test@example.com",
		password="incorrectpassword"),
		follow_redirects = True)
	assert res.status_code == 200
	assert b"Incorrect Password" in res.data	

	res = client.post("/signup",data=dict(
		email="test@example.com",
		first_name="John",
		password="testpassword",
		), follow_redirects=True)
	assert res.status_code == 200
	assert b"Oops! An account with that email already exists" in res.data