test_storage_client.py: add test_get_available_space_{old,new}.

Signed-off-by: Daira Hopwood <daira@jacaranda.org>

src/allmydata/test/test_storage_client.py

@@ -0,0 +1,30 @@
+
+from twisted.trial import unittest
+from allmydata.storage_client import NativeStorageServer
+
+
+class NativeStorageServerWithVersion(NativeStorageServer):
+    def __init__(self,version):
+        self.version=version
+    def get_version(self):
+        return self.version
+
+
+class TestNativeStorageServer(unittest.TestCase):
+    def test_get_available_space_new(self):
+        nss = NativeStorageServerWithVersion(
+            { "http://allmydata.org/tahoe/protocols/storage/v1":
+                { "maximum-immutable-share-size": 111,
+                  "available-space": 222,
+                }
+            })
+        self.failUnlessEqual(nss.get_available_space(), 222)
+
+    def test_get_available_space_old(self):
+        nss = NativeStorageServerWithVersion(
+            { "http://allmydata.org/tahoe/protocols/storage/v1":
+                { "maximum-immutable-share-size": 111,
+                }
+            })
+        self.failUnlessEqual(nss.get_available_space(), 111)
+