[or-cvs] [torflow/master 65/92] HTMLTest httpcode dict maintenance. Calling the right add_/remove_target

[mikeperry at torproject.org]

Author: John M. Schanck <john at anomos.info>
Date: Sat, 7 Aug 2010 18:30:40 -0400
Subject: HTMLTest httpcode dict maintenance. Calling the right add_/remove_target
Commit: c3141a0b44116a3cc543f0ac40cea5d9972264f1

---
 NetworkScanners/ExitAuthority/soat.py |   16 ++++++++++++++++
 1 files changed, 16 insertions(+), 0 deletions(-)

diff --git a/NetworkScanners/ExitAuthority/soat.py b/NetworkScanners/ExitAuthority/soat.py
index 74f8c36..af37e60 100755
--- a/NetworkScanners/ExitAuthority/soat.py
+++ b/NetworkScanners/ExitAuthority/soat.py
@@ -1386,6 +1386,22 @@ class BaseHTMLTest(BaseHTTPTest):
     self.recurse_filetypes = recurse_filetypes
     self.fetch_queue = []
 
+  def _reset(self):
+    self.httpcode_fails = {}
+    self.httpcode_fails_per_exit = {}
+    Test._reset(self)
+
+  def depickle_upgrade(self):
+    if self._pickle_revision < 7:
+      self.httpcode_fails_per_exit = {}
+    Test.depickle_upgrade(self)
+
+  def add_target(self, target):
+    Test.add_target(self, target)
+
+  def remove_target(self, target, reason="None"):
+    Test.remove_target(self, target, reason)
+
   def run_test(self):
     # A single test should have a single cookie jar
     self.tor_cookie_jar = cookielib.MozillaCookieJar()
-- 
1.7.1