From 8253cc047de523fd2097e4d85417316ff9a37ad5 Mon Sep 17 00:00:00 2001 From: Daniel Dunbar Date: Thu, 29 Aug 2013 00:48:39 +0000 Subject: [lit] Change lit.Test.ResultCode to be unique across pickling. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@189549 91177308-0d34-0410-b5e6-96231b3b80d8 --- utils/lit/lit/Test.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'utils/lit') diff --git a/utils/lit/lit/Test.py b/utils/lit/lit/Test.py index 59ca694..05cae99 100644 --- a/utils/lit/lit/Test.py +++ b/utils/lit/lit/Test.py @@ -5,6 +5,17 @@ import os class ResultCode(object): """Test result codes.""" + # We override __new__ and __getnewargs__ to ensure that pickling still + # provides unique ResultCode objects in any particular instance. + _instances = {} + def __new__(cls, name, isFailure): + res = cls._instances.get(name) + if res is None: + cls._instances[name] = res = super(ResultCode, cls).__new__(cls) + return res + def __getnewargs__(self): + return (self.name, self.isFailure) + def __init__(self, name, isFailure): self.name = name self.isFailure = isFailure -- cgit v1.1