From c0a2af5d69bbf63d244f3f17ab80f0e16de0fe0d Mon Sep 17 00:00:00 2001
From: Daira Hopwood <daira@jacaranda.org>
Date: Sun, 23 Nov 2014 05:46:33 +0000
Subject: [PATCH] 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 | 30 +++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 src/allmydata/test/test_storage_client.py

diff --git a/src/allmydata/test/test_storage_client.py b/src/allmydata/test/test_storage_client.py
new file mode 100644
index 00000000..592e40e3
--- /dev/null
+++ b/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)
+
-- 
2.45.2