Skip to content

C
codejail
Commit d6cf8b68 by Ned Batchelder
Options

Merge pull request #13 from edx/ned/fix-some-tests-for-faster-machine 

Fix some tests for faster machine
parents 5e67e393 bbb5c4e9
Showing with 7 additions and 5 deletions
codejail/tests/test_jail_code.py
...@@ -126,26 +126,28 @@ class TestLimits(JailCodeHelpers, unittest.TestCase): ...@@ -126,26 +126,28 @@ class TestLimits(JailCodeHelpers, unittest.TestCase):
def test_cant_use_too_much_memory(self): def test_cant_use_too_much_memory(self):
# This will fail after setting the limit to 30Mb. # This will fail after setting the limit to 30Mb.
set_limit('VMEM', 30000000) set_limit('VMEM', 30000000)
res = jailpy(code="print len(bytearray(50000000))") res = jailpy(code="print len(bytearray(40000000))")
self.assertEqual(res.stdout, "") self.assertEqual(res.stdout, "")
self.assertNotEqual(res.status, 0) self.assertNotEqual(res.status, 0)
def test_changing_vmem_limit(self): def test_changing_vmem_limit(self):
# Up the limit, it will succeed. # Up the limit, it will succeed.
set_limit('VMEM', 60000000) set_limit('VMEM', 80000000)
res = jailpy(code="print len(bytearray(50000000))") res = jailpy(code="print len(bytearray(40000000))")
self.assertEqual(res.stdout, "50000000\n") self.assertEqual(res.stderr, "")
self.assertEqual(res.stdout, "40000000\n")
self.assertEqual(res.status, 0) self.assertEqual(res.status, 0)
def test_disabling_vmem_limit(self): def test_disabling_vmem_limit(self):
# Disable the limit, it will succeed. # Disable the limit, it will succeed.
set_limit('VMEM', 0) set_limit('VMEM', 0)
res = jailpy(code="print len(bytearray(50000000))") res = jailpy(code="print len(bytearray(50000000))")
self.assertEqual(res.stderr, "")
self.assertEqual(res.stdout, "50000000\n") self.assertEqual(res.stdout, "50000000\n")
self.assertEqual(res.status, 0) self.assertEqual(res.status, 0)
def test_cant_use_too_much_cpu(self): def test_cant_use_too_much_cpu(self):
res = jailpy(code="print sum(xrange(100000000))") res = jailpy(code="print sum(xrange(2**31-1))")
self.assertEqual(res.stdout, "") self.assertEqual(res.stdout, "")
self.assertNotEqual(res.status, 0) self.assertNotEqual(res.status, 0)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment