def test_get_multipart_formdata_inputs_with_files(self):
f = io.StringIO(u"test file contents")
g = io.StringIO(u"test file contents")
req = Request.blank('/filesize', POST={'data': ('test', f)})
func = FireflyFunction(dummy)
d = func.get_multipart_formdata_inputs(req)
assert d['data'].read().decode() == g.read()
def test_get_multipart_formdata_inputs_with_no_files(self):
def dummy():
pass
req = Request.blank('/filesize', POST={'abc': 'hi', 'xyz': 1})
func = FireflyFunction(dummy)
d = func.get_multipart_formdata_inputs(req)
assert d['abc'] == 'hi'
assert d['xyz'] == '1'
def test_call_for_bad_request(self):
def sum(a):
return sum(a)
func = FireflyFunction(sum)
request = Request.blank("/sum", POST='{"a": [3 8]}')
response = func(request)
assert response.status == '400 Bad Request'
def test_call_for_internal_function_error(self):
def dummy(a):
raise ValueError("This is a test")
req = Request.blank('/dummy', POST='{"a": 1}')
func = FireflyFunction(dummy)
resp = func(req)
assert resp.status == '500 Internal Server Error'
assert resp.json == {'error': 'ValueError: This is a test'}
def test_call_for_file_inputs(self):
def filesize(data):
return len(data.read())
f = io.StringIO(u"test file contents")
req = Request.blank('/filesize', POST={'data': ('test', f)})
func = FireflyFunction(filesize)
resp = func(req)
assert resp.status == '200 OK'
assert resp.body == b'18'
def test_generate_signature(self):
def sample_function(x, one="hey", two=None, **kwargs):
pass
func = FireflyFunction(sample_function)
assert len(func.sig) == 4
assert func.sig[0]['name'] == 'x'
assert func.sig[0]['kind'] == 'POSITIONAL_OR_KEYWORD'
assert func.sig[1]['name'] == 'one'
assert func.sig[1]['kind'] == 'POSITIONAL_OR_KEYWORD'
assert func.sig[1]['default'] == 'hey'
assert func.sig[2]['default'] == None
assert func.sig[3]['name'] == 'kwargs'
assert func.sig[3]['kind'] == 'VAR_KEYWORD'
def test_generate_signature_py3(self):
# work-around to avoid syntax error in python 2
code = 'def f(x, y=1, *, one="hey", two=None, **kwargs): pass'
env = {}
exec(code, env, env)
f = env['f']
func = FireflyFunction(f)
assert len(func.sig) == 5
assert func.sig[0]['name'] == 'x'
assert func.sig[0]['kind'] == 'POSITIONAL_OR_KEYWORD'
assert func.sig[1]['default'] == 1
assert func.sig[2]['name'] == 'one'
assert func.sig[2]['kind'] == 'KEYWORD_ONLY'
assert func.sig[2]['default'] == 'hey'
assert func.sig[3]['default'] == None
assert func.sig[4]['name'] == 'kwargs'
assert func.sig[4]['kind'] == 'VAR_KEYWORD'
def test_get_content_type_absent(self):
req = Request.blank('/')
func = FireflyFunction(dummy)
content_type = func.get_content_type(req)
assert content_type == 'application/octet-stream'
def test_get_content_type_present(self):
req = Request.blank('/',
headers={'Content-Type': 'multipart/form-data'})
func = FireflyFunction(dummy)
content_type = func.get_content_type(req)
assert content_type == 'multipart/form-data'
def test_call(self):
func = FireflyFunction(square)
request = Request.blank("/square", POST='{"a": 3}')
response = func(request)
assert response.status == '200 OK'
assert response.text == '9'