python - 如何在 Python 中删除(拆解)单例实例装饰器以进行单元测试？

Question

我是 Python 新手。我有一个单例装饰器定义为:

def singleton(cls):
    instances = {}

    def getinstance():
        if cls not in instances:
            instances[cls] = cls()
        return instances[cls]
    return getinstance

还有我的单例类:

@singleton
class MyClass:

    def __init__(self, **kwargs):
        self.config = None
        ...

我想为 MyClass 编写一些单元测试，但我需要在 tearDown() 方法上销毁它，以便测试正常运行。我该怎么做？

非常感谢，

Answer 1

我认为可以将问题重新问为“我可以访问闭包外部的闭包变量吗”以清除“instances”指令。

我没有对此进行测试，但这种方法可以在您将实例数据公开为函数属性(基本上使其等于模块级变量)的情况下工作:

def singleton(cls):

    def getinstance():
        if cls not in singleton.instances:
            singleton.instances[cls] = cls()
        return singleton.instances[cls]
    return getinstance

singleton.instances = {}


# Then clear
def tearDown(self):
    singleton.instances = {}

但是我不确定你是否想经历所有的魔法只是为了拥有一个单例。例如。另一种不那么神奇的延迟加载模式可以是:

class MyClass:

     @classmethod
     def get_instance(cls):
         if not hasattr(cls, "_instance"):
             cls._instance = cls()
         return cls._instance

instance = MyClass.get_instance()

或者在没有延迟加载的情况下更简单:

class MyClass:
    ...


MyClass.instance = MyClass()