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 = list(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")