class test_run_command(TestCase):
def setUp(self):
self.short_id = '3ddc8'
self.aws_lambda = Lambda_Package(
'osbot_jupyter.lambdas.execute_python')
self.result = None
self.png_data = None
self.aws_lambda.add_module('osbot_browser')
self.aws_lambda.update_code(
) # use when wanting to update lambda function
def tearDown(self):
if self.result:
Dev.pprint(self.result)
if self.png_data:
png_file = '/tmp/lambda_png_file.png'
with open(png_file, "wb") as fh:
fh.write(base64.decodebytes(self.png_data.encode()))
Dev.pprint("Png data with size {0} saved to {1}".format(
len(self.png_data), png_file))
def test_execute_python(self):
code = '40+3'
target = None
keep_contents = None
payload = {
'short_id': self.short_id,
'python_code': code,
target: target,
keep_contents: keep_contents
}
#self.png_data = self.aws_lambda.invoke(payload)
self.result = self.aws_lambda.invoke(payload)
class test_run_command(TestCase):
def setUp(self):
self.aws_lambda = Lambda_Package('osbot_jupyter.lambdas.browser')
self.result = None
self.png_data = None
self.aws_lambda.add_module('osbot_browser')
self.aws_lambda.update_code() # use when wanting to update lambda function
def tearDown(self):
if self.result:
Dev.pprint(self.result)
if self.png_data:
png_file = '/tmp/lambda_png_file.png'
with open(png_file, "wb") as fh:
fh.write(base64.decodebytes(self.png_data.encode()))
Dev.pprint("Png data with size {0} saved to {1}".format(len(self.png_data), png_file))
def test_invoke_lambda(self):
payload = { 'url' : 'https://www.google.com/abc'}
self.png_data = self.aws_lambda.invoke(payload)