dart-sdk/tools/make_bundle_unittest.py

#!/usr/bin/env python
#
# Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
# for details. All rights reserved. Use of this source code is governed by a
# BSD-style license that can be found in the LICENSE file.
#
"""Unit tests for make_bundle.py."""

import os
from os import path
import shutil
import subprocess
import sys
import tempfile
import unittest

import make_bundle


class BundleMakerTest(unittest.TestCase):
    """Unit test class for BundleMaker."""

    def setUp(self):
        self._tempdir = tempfile.mkdtemp()
        self._top_dir = path.normpath(
            path.join(path.dirname(sys.argv[0]), os.pardir))
        self._dest = path.join(self._tempdir, 'new_bundle')

    def tearDown(self):
        shutil.rmtree(self._tempdir)

    def testBuildOptions(self):
        op = make_bundle.BundleMaker.BuildOptions()
        op.parse_args(args=[])

    def testCheckOptions(self):
        op = make_bundle.BundleMaker.BuildOptions()
        options = make_bundle.BundleMaker.CheckOptions(op, self._top_dir,
                                                       ['--dest', self._dest])
        self.failUnless(path.exists(self._dest))
        os.rmdir(self._dest)
        self.assertEquals(self._dest, options['dest'])
        self.assertEquals(self._top_dir, options['top_dir'])
        self.failIf(options['verbose'])
        self.failIf(options['skip_build'])
        options = make_bundle.BundleMaker.CheckOptions(
            op, self._top_dir, ['--dest', self._dest, '-v'])
        self.failUnless(path.exists(self._dest))
        self.assertEquals(self._dest, options['dest'])
        self.assertEquals(self._top_dir, options['top_dir'])
        self.failUnless(options['verbose'])
        self.failIf(options['skip_build'])

    def _RunCommand(self, *args):
        proc = subprocess.Popen(
            args,
            cwd=self._dest,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT)
        stdout = proc.communicate()[0]
        self.assertEqual(
            0, proc.wait(), msg='%s\n%s' % (' '.join(args), stdout))

    def testMakeBundle(self):
        os.mkdir(self._dest)
        maker = make_bundle.BundleMaker(dest=self._dest, top_dir=self._top_dir)
        self.assertEquals(0, maker.MakeBundle())
        commands = [
            './dart samples/hello.dart',
            './dart samples/deltablue.dart',
            './dart samples/mandelbrot.dart',
            './dart samples/towers.dart',
        ]
        for command in commands:
            args = command.split(' ')
            self._RunCommand(*args)
            args.append('--arch=dartc')
            self._RunCommand(*args)


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