mitogen/tests/buildah_test.py

import os

import mitogen

import unittest2

import testlib


class ConstructorTest(testlib.RouterMixin, testlib.TestCase):
    def test_okay(self):
        buildah_path = testlib.data_path('stubs/stub-buildah.py')
        context = self.router.buildah(
            container='container_name',
            buildah_path=buildah_path,
        )
        stream = self.router.stream_by_id(context.context_id)

        argv = eval(context.call(os.getenv, 'ORIGINAL_ARGV'))
        self.assertEquals(argv[0], buildah_path)
        self.assertEquals(argv[1], 'run')
        self.assertEquals(argv[2], '--')
        self.assertEquals(argv[3], 'container_name')
        self.assertEquals(argv[4], stream.python_path)


if __name__ == '__main__':
    unittest2.main()