ssato / python-anyconfig

tests/lib.py

#
# Copyright (C) 2015 Satoru SATOH <ssato @ redhat.com>
# License: MIT
#
# pylint: disable=missing-docstring
import os.path
import subprocess
import unittest

import tests.common


SCRIPT_TO_USE_ANYCONFIG = """\
#! /usr/bin/env python
import anyconfig

c = anyconfig.load("/") or {}
anyconfig.dump(c, "/dev/null", "yaml")
"""


def check_output(cmd):
    devnull = open('/dev/null', 'w')
    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=devnull)
    return proc.communicate()[0]


class Test(unittest.TestCase):

This code seems to be duplicated in your project.

    def setUp(self):
        self.workdir = tests.common.setup_workdir()
        self.script = os.path.join(self.workdir, "a.py")

    def tearDown(self):
        tests.common.cleanup_workdir(self.workdir)

    def test_00_run_script(self):
        with open(self.script, 'w') as fileobj:
            fileobj.write(SCRIPT_TO_USE_ANYCONFIG)

            out = check_output(["python", self.script])
            self.assertTrue(out in (b'', ''))

# vim:sw=4:ts=4:et: