46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
import os
|
|
|
|
import mitogen
|
|
import mitogen.lxc
|
|
|
|
try:
|
|
any
|
|
except NameError:
|
|
from mitogen.core import any
|
|
|
|
import unittest2
|
|
|
|
import testlib
|
|
|
|
|
|
def has_subseq(seq, subseq):
|
|
return any(seq[x:x+len(subseq)] == subseq for x in range(0, len(seq)))
|
|
|
|
|
|
class ConstructorTest(testlib.RouterMixin, testlib.TestCase):
|
|
lxc_attach_path = testlib.data_path('stubs/stub-lxc-attach.py')
|
|
|
|
def test_okay(self):
|
|
context = self.router.lxc(
|
|
container='container_name',
|
|
lxc_attach_path=self.lxc_attach_path,
|
|
)
|
|
|
|
argv = eval(context.call(os.getenv, 'ORIGINAL_ARGV'))
|
|
self.assertEquals(argv[0], self.lxc_attach_path)
|
|
self.assertTrue('--clear-env' in argv)
|
|
self.assertTrue(has_subseq(argv, ['--name', 'container_name']))
|
|
|
|
def test_eof(self):
|
|
e = self.assertRaises(mitogen.parent.EofError,
|
|
lambda: self.router.lxc(
|
|
container='container_name',
|
|
lxc_attach_path='true',
|
|
)
|
|
)
|
|
self.assertTrue(str(e).endswith(mitogen.lxc.Stream.eof_error_hint))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest2.main()
|