From 7f195c9c9a9b20b95e5f94267c83645de871a2e5 Mon Sep 17 00:00:00 2001
From: meejah <meejah@meejah.ca>
Date: Wed, 28 Nov 2018 02:54:49 -0700
Subject: [PATCH] test of certificates on not-enough-servers causing uploads to
 fail

---
 integration/test_grid_manager.py | 79 ++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git a/integration/test_grid_manager.py b/integration/test_grid_manager.py
index 886bf76e9..39e2fbe9b 100644
--- a/integration/test_grid_manager.py
+++ b/integration/test_grid_manager.py
@@ -7,6 +7,7 @@ from os.path import join, exists, getmtime
 
 from allmydata.util import keyutil
 from allmydata.util import base32
+from allmydata.util import configutil
 
 import util
 
@@ -90,3 +91,81 @@ def test_remove_last_client(reactor):
     )
     # there are no storage servers left at all now
     assert not json.loads(gm_config).has_key('storage_servers')
+
+
+@pytest.inlineCallbacks
+def test_reject_storage_server(reactor, request, alice, storage_nodes):
+    gm_config = yield util.run_tahoe(
+        reactor, "grid-manager", "--config", "-", "create",
+    )
+    privkey_bytes = json.loads(gm_config)['private_key'].encode('ascii')
+    privkey, pubkey_bytes = keyutil.parse_privkey(privkey_bytes)
+    pubkey = keyutil.parse_pubkey(pubkey_bytes)
+
+    # create certificates for first 2 storage-servers
+    for idx, storage in enumerate(storage_nodes[:2]):
+        pubkey_fname = join(storage._node_dir, "node.pubkey")
+        with open(pubkey_fname, 'r') as f:
+            pubkey = f.read().strip()
+
+        gm_config = yield util.run_tahoe(
+            reactor, "grid-manager", "--config", "-", "add",
+            "storage{}".format(idx), pubkey,
+            stdin=gm_config,
+        )
+    assert sorted(json.loads(gm_config)['storage_servers'].keys()) == ['storage0', 'storage1']
+
+    print("inserting certificates")
+    # insert their certificates
+    for idx, storage in enumerate(storage_nodes[:2]):
+        print(idx, storage)
+        cert = yield util.run_tahoe(
+            reactor, "grid-manager", "--config", "-", "sign",
+            "storage{}".format(idx),
+            stdin=gm_config,
+        )
+        with open(join(storage._node_dir, "gridmanager.cert"), "w") as f:
+            f.write(cert)
+        config = configutil.get_config(join(storage._node_dir, "tahoe.cfg"))
+        config.set("storage", "grid_manager_certificate_files", "gridmanager.cert")
+        config.write(open(join(storage._node_dir, "tahoe.cfg"), "w"))
+
+        # re-start this storage server
+        storage.signalProcess('TERM')
+        yield storage._protocol.exited
+        time.sleep(1)
+        storage_nodes[idx] = yield util._run_node(
+            reactor, storage._node_dir, request, None,
+        )
+
+    # now only two storage-servers have certificates .. configure
+    # alice to have the grid-manager certificate
+
+    config = configutil.get_config(join(alice._node_dir, "tahoe.cfg"))
+    print(dir(config))
+    config.add_section("grid_managers")
+    config.set("grid_managers", "test", pubkey_bytes)
+    config.write(open(join(alice._node_dir, "tahoe.cfg"), "w"))
+    alice.signalProcess('TERM')
+    yield alice._protocol.exited
+    time.sleep(1)
+    alice = yield util._run_node(
+        reactor, alice._node_dir, request, None,
+    )
+    time.sleep(5)
+
+    # try to put something into the grid, which should fail (because
+    # alice has happy=3 but should only find storage0, storage1 to be
+    # acceptable to upload to)
+
+    try:
+        yield util.run_tahoe(
+            reactor, "--node-directory", alice._node_dir,
+            "put", "-",
+            stdin="some content" * 200,
+        )
+        assert False, "Should get a failure"
+    except Exception as e:
+        # XXX but how to tell if we failed for the right reason?
+        # (should get UploadUnhappinessError)
+        print("failure expected: {}".format(e))