PageRenderTime 38ms CodeModel.GetById 9ms RepoModel.GetById 0ms app.codeStats 0ms

/test/clojurewerkz/welle/test/indices_test.clj

http://github.com/michaelklishin/welle
Clojure | 91 lines | 81 code | 10 blank | 0 comment | 5 complexity | 0b1851dd0cc3bdfe99c4c14cf2062b5a MD5 | raw file
    

  1. (ns clojurewerkz.welle.test.indices-test
    2. 

  2.   (:use clojure.test
    3. 

  3.         [clojurewerkz.welle.testkit :only [drain]]
    4. 

  4.         [clojure.set :only [subset?]])
    5. 

  5.   (:require [clojurewerkz.welle.core    :as wc]
    6. 

  6.             [clojurewerkz.welle.buckets :as wb]
    7. 

  7.             [clojurewerkz.welle.kv      :as kv])
    8. 

  8.   (:import java.util.UUID
    9. 

  9.            com.basho.riak.client.http.util.Constants))
    10. 

  10. 

  11. (wc/connect!)
    12. 

  12. 

  13. (deftest ^{:2i true} test-indexes-on-converted-riak-objects
    14. 

  14.   (let [bucket-name "clojurewerkz.welle.test.indices-test"
    15. 

  15.         bucket      (wb/update bucket-name)
    16. 

  16.         k           (str (UUID/randomUUID))
    17. 

  17.         v           "value"
    18. 

  18.         indexes     {:email    #{"john@example.com"}
    19. 

  19.                      :username #{"johndoe"}}
    20. 

  20.         stored      (kv/store bucket-name k v :indexes indexes)
    21. 

  21.         [fetched]   (kv/fetch bucket-name k)]
    22. 

  22.     (is (:indexes fetched))
    23. 

  23.     (is (= indexes (:indexes fetched)))
    24. 

  24.     (kv/delete bucket-name k)))
    25. 

  25. 

  26. 

  27. (deftest ^{:2i true} test-basic-index-query-with-a-single-string-value
    28. 

  28.   (let [bucket-name "clojurewerkz.welle.test.indices-test"
    29. 

  29.         bucket      (wb/update bucket-name)
    30. 

  30.         k           (str (UUID/randomUUID))
    31. 

  31.         v           (.getBytes "value")
    32. 

  32.         indexes     {:email #{"johndoe@example.com" "timsmith@example.com"}}
    33. 

  33.         stored      (kv/store bucket-name k v :indexes indexes :content-type Constants/CTYPE_OCTET_STREAM)
    34. 

  34.         [idx-key]   (kv/index-query bucket-name :email "johndoe@example.com")
    35. 

  35.         [fetched]   (kv/fetch bucket-name idx-key)]
    36. 

  36.     (is (:indexes fetched))
    37. 

  37.     (is (= (String. ^bytes (:value fetched))
    38. 

  38.            (String. ^bytes v)))
    39. 

  39.     (is (= (:indexes fetched) indexes))
    40. 

  40.     (kv/delete bucket-name k)))
    41. 

  41. 

  42. 

  43. (deftest ^{:2i true} test-basic-index-query-with-a-range-of-string-values
    44. 

  44.   (let [bucket-name "clojurewerkz.welle.test.indices-test"
    45. 

  45.         bucket      (wb/update bucket-name)
    46. 

  46.         k1          (str (UUID/randomUUID))
    47. 

  47.         k2          (str (UUID/randomUUID))
    48. 

  48.         k3          (str (UUID/randomUUID))
    49. 

  49.         k4          (str (UUID/randomUUID))
    50. 

  50.         v           (.getBytes "value1")
    51. 

  51.         _           (kv/store bucket-name k1 v :indexes {:username #{"abc"}} :content-type Constants/CTYPE_OCTET_STREAM)
    52. 

  52.         _           (kv/store bucket-name k2 v :indexes {:username #{"bcd"}} :content-type Constants/CTYPE_OCTET_STREAM)
    53. 

  53.         _           (kv/store bucket-name k3 v :indexes {:username #{"cde"}} :content-type Constants/CTYPE_OCTET_STREAM)
    54. 

  54.         _           (kv/store bucket-name k4 v :indexes {:username #{"def"}} :content-type Constants/CTYPE_OCTET_STREAM)
    55. 

  55.         keys        (set (kv/index-query bucket-name :username ["b" "d"]))]
    56. 

  56.     (is (subset? #{k2 k3} keys))
    57. 

  57.     (kv/delete-all bucket-name [k1 k2 k3 k4])
    58. 

  58.     (kv/delete-all bucket-name keys)))
    59. 

  59. 

  60. 

  61. (deftest ^{:2i true} test-basic-index-query-with-a-single-integer-value
    62. 

  62.   (let [bucket-name "clojurewerkz.welle.test.alt-indices-test"
    63. 

  63.         bucket      (wb/update bucket-name)
    64. 

  64.         k           (str (UUID/randomUUID))
    65. 

  65.         v           "value"
    66. 

  66.         indexes     {:age 27}
    67. 

  67.         stored      (kv/store bucket-name k v :indexes indexes :content-type Constants/CTYPE_TEXT_UTF8)
    68. 

  68.         [idx-key]   (kv/index-query bucket-name :age 27)
    69. 

  69.         [fetched]   (kv/fetch bucket-name idx-key)]
    70. 

  70.     (is (:indexes fetched))
    71. 

  71.     (is (= (:value fetched) v))
    72. 

  72.     (is (= (:indexes fetched) {:age #{27}}))
    73. 

  73.     (kv/delete bucket-name k)))
    74. 

  74. 

  75. 

  76. (deftest ^{:2i true} test-basic-index-query-with-a-range-of-integer-values
    77. 

  77.   (let [bucket-name "clojurewerkz.welle.test.indices-test"
    78. 

  78.         bucket      (wb/update bucket-name)
    79. 

  79.         k1          (str (UUID/randomUUID))
    80. 

  80.         k2          (str (UUID/randomUUID))
    81. 

  81.         k3          (str (UUID/randomUUID))
    82. 

  82.         k4          (str (UUID/randomUUID))
    83. 

  83.         v           (.getBytes "value1")
    84. 

  84.         _           (kv/store bucket-name k1 v :indexes {:hops #{1 2 3 4}} :content-type Constants/CTYPE_OCTET_STREAM)
    85. 

  85.         _           (kv/store bucket-name k2 v :indexes {:hops #{5 6 7 8}} :content-type Constants/CTYPE_OCTET_STREAM)
    86. 

  86.         _           (kv/store bucket-name k3 v :indexes {:hops #{9 10 11 12}} :content-type Constants/CTYPE_OCTET_STREAM)
    87. 

  87.         _           (kv/store bucket-name k4 v :indexes {:hops #{13 14 18 77}} :content-type Constants/CTYPE_OCTET_STREAM)
    88. 

  88.         keys        (set (kv/index-query bucket-name :hops [2 11]))]
    89. 

  89.     (is (subset? #{k1 k2 k3} keys))
    90. 

  90.     (kv/delete-all bucket-name [k1 k2 k3 k4])
    91. 

  91.     (kv/delete-all bucket-name keys)))
    92. 

  92.