如何在python unittest中为此类使用setUp和tearDown

Question

我很难学习测试驱动的开发。

我正在编写一个类，该类将获取文件名或文件描述，并调整输入的大小，并返回文件中数据的大小块。

首先从测试开始，我所能做的就是测试传递的参数是否不是none，并检查参数是否是有效的fileobjects。

我所能做的就是在代码下面，我是应该使用安装和拆卸方法，还是完全错误？我正在创建一个临时文件和在setUp（）中定义为读取的类的实例，我应该在tearDown（）中删除这个对象吗

下面是代码

class Test_FileChunk(unittest.TestCase):
    """
    """
    def setUp(self):
        self.fhandle, self.fname = mkstemp()
        self.fc_obj = FileChunk(filename=self.fname)

    def tearDown(self):
        try:
            os.remove(self.fname)
        except OSError as oserr:
            print(oserr)

    def test_instance_variables(self):
        self.assertIsNotNone(self.fc_obj.filename)
        self.assertIsNone(self.fc_obj.filehandle)
        self.assertEqual(self.fc_obj.chunk_size, 8192)

    def test_check_if_instance_variables_are_valid_file_objects(self):
        handle = open(self.fc_obj.filename
        self.assertEqual(
            hasattr
                (handle, "r"), 'seek'), True,
                    msg="Is not a valid file object")
        handle.close()

我在stackoverflow上浏览了多个TDD问题和建议的教程，但是看起来遵循TDD教程是非常有趣的，但是实际上做TDD是非常困难的。实际上我可以在ReadChunk类中思考我想做的事情，但是我却无法在实际中先找到测试，然后再编写代码。我可以考虑通过TDD检查传递的值是否是有效的文件对象，如果没有TDD进行编码就不会发生这种情况，但是我不确定是否正确使用了unittest。无法了解全局。有谁能建议一下如何处理这个问题，以及上面截取的代码是否正确。

Answer 1

Unlike TemporaryFile(), the user of mkstemp() is responsible for deleting the temporary file when done with it. - Python Documentation

因此，在tearDown中删除文件是正确的解决方案。

我个人会做一些测试功能，比如

def test_filename(self):
    self.assertEqual(self.fc_obj.filename, self.fname)
    self.assertIsTrue(os.path.isfile(self.fc_obj.filename))


def test_handle(self):
    self.assertIsNone(self.fc_obj.filehandle)

def open_file(self):
    # if you implemented an "open" method
    # you can use this method every time you test the opened file
    result = self.fc_obj.open()
    self.assertIsTrue(result, "File open failed")
    self.assertIsNotNone(self.fc_obj.filehandle)

# if you plan to implement a read and a write method you can add the following test
def test_read_write_file(self):
    self.open_file()
    random_data = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(N))
    self.fc_obj.write(random_data)
    self.assertEqual(self.fc_obj.read(), random_data)

等等，为您计划实现的所有内容定义一个测试，实现它并运行测试。到目前为止，您的类看起来还不错，但是正如您可能已经看到的，您应该尽可能具体地进行测试，例如使用

self.assertEqual(self.fc_obj.filename, self.fname)

而不是

self.assertIsNotNone(self.fc_obj.filename)

如果您想对打开的FileChunk对象进行大量测试，还可以添加第二个unittest.TestCase。

class TestOpenFileChunk(unittest.TestCase);
    def setUp(self):
        self.fhandle, self.fname = mkstemp()
        self.fc_obj = FileChunk(filename=self.fname)
        self.fc_obj.open()

    def tearDown(self):
        # if you have this method
        self.fc_object.close()
        # and then
        try:
            os.remove(self.fname)
        except OSError as why:
            print(why)

    def test_read_write(self):
        #...

如果只想创建一次FileChunk对象，还可以使用setUpClass和tearDownClass方法。

class TestOpenFileChunk(unittest.TestCase);
    @classmethod
    def setUpClass(cls):
        cls.fhandle, cls.fname = mkstemp()
        cls.fc_obj = FileChunk(filename=self.fname)
        cls.fc_obj.open()

    @classmethod
    def tearDownClass(cls):
        # if you have this method
        cls.fc_obj.close()
        # and then
        try:
            os.remove(cls.fname)
        except OSError as why:
            print(why)

并像在测试方法中一样使用self.fc_obj。