ssh_config_utils/tests/tests.py

import os
import sys
import unittest
import tempfile

samples_dir = os.path.join(sys.path[0], "sample_configs")
CONFIG_FILES = dict(
    (file, os.path.join(samples_dir, file)) for file in os.listdir(samples_dir)
)


class ReadWriteTest(unittest.TestCase):
    def setUp(self):
        self.write_file = tempfile.TemporaryFile(mode="r+")

    def tearDown(self):
        self.write_file.close()


class TestTempFile(ReadWriteTest):
    def test_me(self):
        self.write_file.write("hello")
        self.write_file.seek(0)
        self.assertEqual(self.write_file.read(), "hello")


class TestParser(ReadWriteTest):
    pass