more testing

1 files changed, 17 insertions, 0 deletions

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\')"}')