Switched test to simpler put method.

AnIrishDuck · May 6, 2011 · 060c0d7 · 060c0d7
1 parent 729f1d3
commit 060c0d7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/test/test_basic.py b/test/test_basic.py
@@ -51,10 +51,8 @@ def test_ksk(self, client):
         uri = "KSK@" + now.strftime("%Y-%m-%d-%H-%M")
         # First put.
         testdata = "Testing 123..."
-        put = IdentifiedMessage("ClientPut", [("Verbosity", 1), ("URI", uri)])
-        client.sendMessage(put, data=testdata)
-        response = yield client.deferred["PutSuccessful"]
-        self.assertEqual(response["Identifier"], put["Identifier"])
+        response = yield client.put_direct(uri, testdata)
+        self.assertEqual(response["URI"], uri)
         # Then get.
         response = yield client.get_direct(uri)
         # Finally check the data.