Add rake tasks for generating orderable strings test cases

These tasks generate JSON files with test cases that can be used for
tests in other clients (e.g. StashRS).

Author: CDThomas

Rakefile

@@ -28,6 +28,104 @@ task :release do
   sh "git release"
 end
 
+require "securerandom"
+require "./lib/cipherstash/index"
+require "json"
+
+task :generate_ordered_string_test_cases do
+  id = SecureRandom.uuid
+  settings = {
+    "meta" => {
+      "$indexId" => id,
+      "$indexName" => "titleSort",
+      "$prfKey" => SecureRandom.hex(16),
+      "$prpKey" => SecureRandom.hex(16),
+    },
+    "mapping" => {
+      "kind" => "range",
+      "field" => "title",
+      "fieldType"=>"string",
+    }
+  }
+  schema_versions = {:first=>0, :last=>0, :searchable=>true}
+
+  index = CipherStash::Index.generate(id, settings, schema_versions)
+
+  random_ascii_string = -> () do
+    unicode_char_max = 127
+    max_string_length = 200
+    (0..rand(max_string_length -1)).map { rand(unicode_char_max + 1).chr }.join
+  end
+
+  num_test_cases = 1_000
+
+  orderise_string_cases = (0..(num_test_cases - 1)).map do
+    str = random_ascii_string.call
+    output = index.__send__ :orderise_string, str
+    {input: str, output: output}
+  end
+
+  File.write("orderise_string_test_cases.json", orderise_string_cases.to_json)
+
+  num_test_cases = 100
+
+  string_comparison_cases = (0..(num_test_cases - 1)).map do
+    # TODO: this only gives single-char strings of alpha chars for debugging
+    # why this test case doesn't work. This looks like a bug on the ruby side.
+    # The comparison logic using encrypted terms is busted.
+    # random_ascii_string = -> () do
+    #   max_string_length = 10
+    #   (0..0).map { (rand(26) + 97).chr }.join
+    # end
+
+    # str_a = random_ascii_string.call
+    # terms_a = index.analyze(SecureRandom.uuid, {"title" => str_a})[:terms]
+
+    # if terms_a.length != 1
+    #   raise "Expected terms_a to have exactly one item. Had #{terms_a.length}"
+    # end
+
+    # term_a = terms_a.first[:term]
+
+    # str_b = random_ascii_string.call
+    # terms_b = index.analyze(SecureRandom.uuid, {"title" => str_b})[:terms]
+
+    # if terms_b.length != 1
+    #   raise "Expected terms_b to have exactly one item. Had #{terms_b.length}"
+    # end
+
+    # term_b = terms_b.first[:term]
+
+    # output = case term_a <=> term_b
+    #   when -1
+    #     "<"
+    #   when 0
+    #     "=="
+    #   when 1
+    #     ">"
+    #   end
+
+    str_a = random_ascii_string.call
+    terms_a = index.__send__ :orderise_string, str_a
+
+    str_b = random_ascii_string.call
+    terms_b = index.__send__ :orderise_string, str_b
+
+    output = case terms_a <=> terms_b
+      when -1
+        "<"
+      when 0
+        "=="
+      when 1
+        ">"
+      end
+
+    {input: [str_a, str_b], output: output}
+  end
+
+  File.write("string_comparison_test_cases.json", string_comparison_cases.to_json)
+end
+
 require 'yard'
 
 YARD::Rake::YardocTask.new :doc do |yardoc|