consolidate multiple definitions of NotEnoughPeersError

This commit is contained in:
Brian Warner 2007-06-07 22:20:55 -07:00
parent f62a544b93
commit 72bc8627de
3 changed files with 4 additions and 9 deletions

View File

@ -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

View File

@ -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"

View File

@ -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()