From ab2bb78ec8c5979816a9536e96396b2f024c3415 Mon Sep 17 00:00:00 2001 From: Christopher Baklid Date: Thu, 31 May 2018 16:36:03 +0200 Subject: more testing --- tests/test_snekbox.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'tests/test_snekbox.py') diff --git a/tests/test_snekbox.py b/tests/test_snekbox.py index 4d86ecf..d2f81fe 100644 --- a/tests/test_snekbox.py +++ b/tests/test_snekbox.py @@ -1,12 +1,18 @@ import unittest import pytest import os +import json from snekbox import Snekbox +from rmq import Rmq + +r = Rmq() + python_binary = os.environ.get('PYTHONEXECUTABLE', '/usr/bin/python3.6') nsjail = os.sep.join([os.getcwd(), f'binaries{os.sep}nsjail2.6-ubuntu-x86_64']) snek = Snekbox(nsjail_binary=nsjail, python_binary=python_binary) + class SnekTests(unittest.TestCase): def test_nsjail(self): result = snek.python3('print("test")') @@ -39,3 +45,14 @@ class SnekTests(unittest.TestCase): self.assertIn('ModuleNotFoundError', result.strip()) else: self.assertIn('returned non-zero exit status 1.', result.strip()) + + +class RMQTests(unittest.TestCase): + def test_a_publish(self): + message = json.dumps({"snekid": "test", "message": "print('test')"}) + result = r.publish(message) + self.assertTrue(result) + + def test_b_consume(self): + result = r.consume(callback=snek.message_handler, queue='input', run_once=True) + self.assertEquals(result[2], b'{"snekid": "test", "message": "print(\'test\')"}') -- cgit v1.2.3