From: Brian Warner <warner@allmydata.com>
Date: Fri, 8 Jun 2007 05:20:55 +0000 (-0700)
Subject: consolidate multiple definitions of NotEnoughPeersError
X-Git-Tag: allmydata-tahoe-0.3.0~8
X-Git-Url: https://git.rkrishnan.org/module-simplejson.tests.html?a=commitdiff_plain;h=72bc8627de67577833848128228de230aced670c;p=tahoe-lafs%2Ftahoe-lafs.git

consolidate multiple definitions of NotEnoughPeersError
---

diff --git a/src/allmydata/download.py b/src/allmydata/download.py
index ebbce169..859f9bfc 100644
--- a/src/allmydata/download.py
+++ b/src/allmydata/download.py
@@ -11,9 +11,7 @@ from allmydata import codec, hashtree
 from allmydata.Crypto.Cipher import AES
 from allmydata.uri import unpack_uri
 from allmydata.interfaces import IDownloadTarget, IDownloader
-
-class NotEnoughPeersError(Exception):
-    pass
+from allmydata.encode import NotEnoughPeersError
 
 class HaveAllPeersError(Exception):
     # we use this to jump out of the loop
diff --git a/src/allmydata/test/test_upload.py b/src/allmydata/test/test_upload.py
index d5b20c1c..cf6017a0 100644
--- a/src/allmydata/test/test_upload.py
+++ b/src/allmydata/test/test_upload.py
@@ -3,7 +3,7 @@ from twisted.trial import unittest
 from twisted.python.failure import Failure
 from cStringIO import StringIO
 
-from allmydata import upload
+from allmydata import upload, encode
 from allmydata.uri import unpack_uri
 
 from test_encode import FakePeer
@@ -63,7 +63,7 @@ class FullServer(unittest.TestCase):
         self.u.parent = self.node
 
     def _should_fail(self, f):
-        self.failUnless(isinstance(f, Failure) and f.check(upload.NotEnoughPeersError))
+        self.failUnless(isinstance(f, Failure) and f.check(encode.NotEnoughPeersError))
 
     def testData(self):
         data = "This is some data to upload"
diff --git a/src/allmydata/upload.py b/src/allmydata/upload.py
index 48b03f54..4314c3af 100644
--- a/src/allmydata/upload.py
+++ b/src/allmydata/upload.py
@@ -13,9 +13,6 @@ from allmydata.Crypto.Cipher import AES
 from cStringIO import StringIO
 import collections, random
 
-class NotEnoughPeersError(Exception):
-    pass
-
 class HaveAllPeersError(Exception):
     # we use this to jump out of the loop
     pass
@@ -146,7 +143,7 @@ class FileUploader:
                 # But we allocated places for enough shares.
                 log.msg("%s._locate_all_shareholders() But we allocated places for enough shares.")
                 return self.used_peers
-            raise NotEnoughPeersError
+            raise encode.NotEnoughPeersError
         # we need to keep trying
         return self._locate_more_shareholders()