Add test for scheme initialization.

diff --git a/scheme.py b/scheme.py
index 9eebdb8745f5bafe4efc9612dcde4332418c712b..20f6521d96e527096a3809e5421c06424a187019 100644 (file)
--- a/scheme.py
+++ b/scheme.py
@@ -130,7 +130,7 @@ class ListBranchingScheme(BranchingScheme):
         :param branch_list: List of know branch locations.
         """
         if isinstance(branch_list, basestring):
-            branch_list = bz2.decompress(urlsafe_b64decode(branch_list).replace(".", "=")).splitlines()
+            branch_list = bz2.decompress(urlsafe_b64decode(branch_list.replace(".", "="))).splitlines()
         self.branch_list = [p.strip("/") for p in branch_list]
         self.split_branch_list = [p.split("/") for p in self.branch_list]
 

diff --git a/tests/test_scheme.py b/tests/test_scheme.py
index 5c57323e8e3656bb8e04e5719ddc588cac1b45c8..b7d9aff6403071943f7a6ae8cd02e78d3ae11ca1 100644 (file)
--- a/tests/test_scheme.py
+++ b/tests/test_scheme.py
@@ -138,6 +138,10 @@ class ListScheme(TestCase):
     def setUp(self):
         self.scheme = ListBranchingScheme(["foo", "bar/bloe"])
 
+    def test_create_from_string(self):
+        self.scheme = ListBranchingScheme('QlpoOTFBWSZTWXb2s-UAAADBAAAQAQCgACGYGYQYXckU4UJB29rPlA..')
+        self.assertEquals(["foo"], self.scheme.branch_list)
+
     def test_is_tag_empty(self):
         self.assertFalse(self.scheme.is_tag(""))