from json import loads as json_loads, dumps as json_dumps
from sanic import Sanic
from sanic.response import json, text
from sanic.utils import sanic_endpoint_test

# ------------------------------------------------------------ #
#  GET
# ------------------------------------------------------------ #

def test_sync():
	app = Sanic('test_text')

	@app.route('/')
	def handler(request):
		return text('Hello')

	request, response = sanic_endpoint_test(app)

	assert response.text == 'Hello'

def test_text():
	app = Sanic('test_text')

	@app.route('/')
	async def handler(request):
		return text('Hello')

	request, response = sanic_endpoint_test(app)

	assert response.text == 'Hello'


def test_json():
	app = Sanic('test_json')

	@app.route('/')
	async def handler(request):
		return json({"test":True})

	request, response = sanic_endpoint_test(app)

	try:
		results = json_loads(response.text)
	except:
		raise ValueError("Expected JSON response but got '{}'".format(response))

	assert results.get('test') == True


def test_query_string():
	app = Sanic('test_query_string')

	@app.route('/')
	async def handler(request):
		return text('OK')

	request, response = sanic_endpoint_test(app, params=[("test1", 1), ("test2", "false"), ("test2", "true")])

	assert request.args.get('test1') == '1'
	assert request.args.get('test2') == 'false'

# ------------------------------------------------------------ #
#  POST
# ------------------------------------------------------------ #

def test_post_json():
	app = Sanic('test_post_json')

	@app.route('/')
	async def handler(request):
		return text('OK')

	payload = {'test': 'OK'}
	headers = {'content-type': 'application/json'}

	request, response = sanic_endpoint_test(app, data=json_dumps(payload), headers=headers)

	assert request.json.get('test') == 'OK'
	assert response.text == 'OK'