Finalizing the filebucket client, with test code.

git-svn-id: https://reductivelabs.com/svn/puppet/trunk@2348 980ebf18-57e1-0310-9a29-db15c13687c0

2 files changed, 36 insertions, 1 deletions

diff --git a/test/network/client/dipper.rb b/test/network/client/dipper.rb
new file mode 100755
index 000000000..e91dea2a7
--- /dev/null
+++ b/test/network/client/dipper.rb
@@ -0,0 +1,35 @@
+#!/usr/bin/env ruby
+
+$:.unshift("../../lib") if __FILE__ =~ /\.rb$/
+
+require 'puppettest'
+require 'puppet/network/client/resource'
+
+class TestDipperClient < Test::Unit::TestCase
+    include PuppetTest::ServerTest
+
+    def setup
+        super
+        @dipper = Puppet::Network::Client.dipper.new(:Path => tempfile)
+    end
+
+    # Make sure we can create a new file with 'restore'.
+    def test_restore_to_new_file
+        file = tempfile
+        text = "asdf;lkajseofiqwekj"
+        File.open(file, "w") { |f| f.puts text }
+        md5 = nil
+        assert_nothing_raised("Could not send file") do
+            md5 = @dipper.backup(file)
+        end
+
+        newfile = tempfile
+        assert_nothing_raised("could not restore to new path") do
+            @dipper.restore(newfile, md5)
+        end
+
+        assert_equal(File.read(file), File.read(newfile), "did not restore correctly")
+    end
+end
+
+# $Id$
diff --git a/test/network/handler/bucket.rb b/test/network/handler/bucket.rb
index 9c68afd29..eb488dd33 100755
--- a/test/network/handler/bucket.rb
+++ b/test/network/handler/bucket.rb
@@ -276,7 +276,7 @@ class TestBucket < Test::Unit::TestCase
         md5 = Digest::MD5.hexdigest(text)
 
         dirs = File.join(md5[0..7].split(""))
-        dir = File.join(@bucket, dirs)
+        dir = File.join(@bucket, dirs, md5)
         filedir, contents, paths = bucket.class.paths(@bucket, md5)
 
         assert_equal(dir, filedir, "did not use a deeper file structure")