[tor-commits] [bridgedb/develop] Add missing unittest import

[isis at torproject.org]

commit 0ea71dd97c292063247c25943ed0c4bc11bc93ac
Author: Matthew Finkel <Matthew.Finkel at gmail.com>
Date:   Wed Aug 20 06:15:55 2014 +0000

    Add missing unittest import
---
 lib/bridgedb/test/test_Bucket.py |    4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/bridgedb/test/test_Bucket.py b/lib/bridgedb/test/test_Bucket.py
index 8893dcd..82424b6 100644
--- a/lib/bridgedb/test/test_Bucket.py
+++ b/lib/bridgedb/test/test_Bucket.py
@@ -19,6 +19,8 @@ from sure import this
 from sure import the
 from sure import expect
 
+from twisted.trial import unittest
+
 from bridgedb import Bucket
 
 
@@ -31,7 +33,7 @@ class BucketDataTest(unittest.TestCase):
         distname = "test-distributor"
         bucket = Bucket.BucketData(distname, alloc)
         this(distname).should.be.equal(bucket.name)
-        this(alloc).should.be.equal(bucket.needed)
+        this(Bucket.BUCKET_MAX_BRIDGES).should.be.equal(bucket.needed)
 
     def test_alloc_all_the_bridges(self):
         """Set the needed number of bridges to the default"""