Factor out service args into a fixture so other modules can reuse tests.

This commit is contained in:
Phil Hagelberg 2010-03-19 15:41:07 -07:00
parent 7a4b513191
commit 0d0b4e849f
1 changed files with 30 additions and 44 deletions

View File

@ -4,49 +4,39 @@
(:import [org.jclouds.blobstore BlobStoreContextFactory]
[java.io ByteArrayOutputStream]))
(def stub-context (.createContext (BlobStoreContextFactory.) "transient" "" ""))
(def stub-blobstore (.getBlobStore stub-context))
(defn clean-stub-fixture
"This should allow basic tests to easily be run with another service."
[service account key & options]
(fn [f]
(with-blobstore [(apply blobstore service account key options)]
(doseq [container (containers)]
(delete-container (.getName container)))
(f))))
(defn clean-stub-fixture [f]
(with-blobstore [stub-blobstore]
(doseq [container (containers)]
(delete-container (.getName container)))
(f)))
(use-fixtures :each clean-stub-fixture)
(use-fixtures :each (clean-stub-fixture "transient" "" ""))
(deftest blobstore?-test
(is (blobstore? stub-blobstore)))
(deftest blobstore-context?-test
(is (blobstore-context? stub-context)))
(deftest blobstore-context-test
(is (= stub-context (blobstore-context stub-blobstore))))
(is (blobstore? *blobstore*)))
(deftest as-blobstore-test
(is (blobstore? (blobstore "transient" "user" "password")))
(is (blobstore? (as-blobstore stub-blobstore)))
(is (blobstore? (as-blobstore stub-context))))
(deftest with-blobstore-test
(with-blobstore [stub-blobstore]
(is (= stub-blobstore *blobstore*))))
(is (blobstore? (as-blobstore *blobstore*)))
(is (blobstore? (as-blobstore (blobstore-context *blobstore*)))))
(deftest create-existing-container-test
(is (not (container-exists? stub-blobstore "")))
(is (not (container-exists? *blobstore* "")))
(is (not (container-exists? "")))
(is (create-container stub-blobstore "fred"))
(is (container-exists? stub-blobstore "fred")))
(is (create-container *blobstore* "fred"))
(is (container-exists? *blobstore* "fred")))
(deftest create-container-test
(is (create-container stub-blobstore "fred"))
(is (container-exists? stub-blobstore "fred")))
(is (create-container *blobstore* "fred"))
(is (container-exists? *blobstore* "fred")))
(deftest containers-test
(is (empty? (containers stub-blobstore)))
(is (create-container stub-blobstore "fred"))
(is (= 1 (count (containers stub-blobstore)))))
(is (empty? (containers *blobstore*)))
(is (create-container *blobstore* "fred"))
(is (= 1 (count (containers *blobstore*)))))
(deftest list-container-test
(is (create-container "container"))
@ -92,28 +82,24 @@
;; TODO: more tests involving blob-specific functions
(deftest corruption-hunt
(let [service "transient"
account ""
secret-key ""
container-name "test"
(let [container-name "test"
name "work-file"
upload-filename "/home/phil/work-file"
total-downloads 100
threads 10
blob-s (blobstore service account secret-key)]
threads 10]
;; upload
(create-container blob-s container-name)
(when-not (blob-exists? blob-s container-name name)
(create-blob blob-s container-name name
(java.io.File. upload-filename)))
(create-container container-name)
(when-not (blob-exists? container-name name)
(let [data-stream (java.io.ByteArrayOutputStream.)]
(dotimes [i 5000000] (.write data-stream i))
(create-blob container-name name (.toByteArray data-stream))))
;; download
(let [total (atom total-downloads)]
(defn new-agent []
(agent name))
(defn dl-and-restart [file]
(defn dl-and-restart [blob-s file]
(when-not (<= @total 0)
(with-open [baos (java.io.ByteArrayOutputStream.)]
(try
@ -124,14 +110,14 @@
(.write of (.toByteArray baos)))
(throw e))))
(swap! total dec)
(send *agent* dl-and-restart)
(send *agent* (partial dl-and-restart blob-s))
file))
(defn start-agents []
(let [agents (map (fn [_] (new-agent))
(range threads))]
(doseq [a agents]
(send-off a dl-and-restart))
(send-off a (partial dl-and-restart *blobstore*)))
agents))
(let [agents (start-agents)]