ruby-tokyotyrant 0.3.0

data/spec/start_tyrants.sh

@@ -0,0 +1,36 @@
+#!/bin/bash
+
+#
+# starting the tt servers (standard + table)
+#
+
+TMP=`pwd`/tmp
+SPEC=`pwd`/spec
+  # so that tt doesn't complain about relative paths...
+
+[ -d $TMP ] || mkdir $TMP
+
+ttserver \
+  -dmn \
+  -port 45000 \
+  -pid $TMP/t_spec.pid -rts $TMP/t_spec.rts \
+  -log $TMP/t.log \
+  -ext $SPEC/ext.lua \
+  $TMP/tyrant.tch
+
+ttserver \
+  -dmn \
+  -port 45001 \
+  -pid $TMP/tt1_spec.pid -rts $TMP/tt1_spec.rts \
+  -log $TMP/tt1.log \
+  -ext $SPEC/ext.lua \
+  $TMP/tyrant_table1.tct
+
+ttserver \
+  -dmn \
+  -port 45002 \
+  -pid $TMP/tt2_spec.pid -rts $TMP/tt2_spec.rts \
+  -log $TMP/tt2.log \
+  -ext $SPEC/ext.lua \
+  $TMP/tyrant_table2.tct
+

data/spec/stop_tyrants.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+# stopping the spec ttservers
+
+ruby -e "%w{ tmp/t_spec.pid tmp/tt1_spec.pid tmp/tt2_spec.pid }.each { |pf| File.exist?(pf) && Process.kill(9, File.read(pf).strip.to_i) }"
+
+rm tmp/t_spec.pid
+rm tmp/tt1_spec.pid
+rm tmp/tt2_spec.pid

data/spec/tokyo_tyrant_query_spec.rb

@@ -0,0 +1,159 @@
+require 'pathname'
+require Pathname(__FILE__).dirname.join('spec_base') unless $root
+
+describe TokyoTyrant::Query, "with an open database" do
+  before do
+    @db = TokyoTyrant::Table.new('127.0.0.1', 45001)
+    @db.clear
+    load('spec/plu_db.rb') unless $codes
+    @db.mput($codes)
+  end
+
+  it "should get a query object" do
+    @db.query.class.should == TokyoTyrant::Query
+  end
+
+  it "should get keys for search conditions" do
+    q = @db.query
+    q.addcond('type', :streq, 'Spinach')
+    q.search.sort.should == %w[3332 34173]
+  end
+
+  it "should get keys for negated search conditions" do
+    q = @db.query
+    q.addcond('type', '!streq', 'Spinach')
+    res = q.search
+    %w[3332 34173].each do |k|
+      res.should.not.include?(k)
+    end
+  end
+
+  it "should get ordered keys for search conditions with default order" do
+    q = @db.query
+    q.addcond('type', :streq, 'Spinach')
+    q.setorder('variety')
+    q.search.should == ["3332", "34173"]
+  end
+
+  it "should get ordered keys for search conditions with ascending order" do
+    q = @db.query
+    q.addcond('type', :streq, 'Spinach')
+    q.setorder('variety', :strasc)
+    q.search.should == ["3332", "34173"]
+  end
+
+  it "should get ordered keys for search conditions with decending order" do
+    q = @db.query
+    q.addcond('type', :streq, 'Spinach')
+    q.order_by('variety', :StrDesc)
+    q.search.should == ["34173", "3332"]
+  end
+
+  it "should get limited keys for search conditions with limit" do
+    q = @db.query{ |q| 
+      q.addcond('type', :streq, 'Apple')
+      q.order_by('variety', :strdesc)
+      q.setlimit(10)
+    }.should == ["4860", "3011", "3271", "3382", "4182", "3353", "4176", "3272", "3297", "3009"]
+  end
+
+  it "should get limited keys for search conditions with limit and offset" do
+    q = @db.query{ |q| 
+      q.addcond('type', :streq, 'Apple')
+      q.order_by('variety', :strdesc)
+      q.setlimit(10, 10)
+    }.should == ["3008", "3352", "3077", "3349", "3076", "3073", "3348", "3007", "3078", "3347"]
+  end
+
+  it "should get records for search conditions" do
+    q = @db.query
+    q.addcond('type', :streq, 'Garlic')
+    res = q.get.sort{ |x,y| x['variety'] <=> y['variety'] }
+    res.should == [{ '__id' => "4609", 'variety' => "Elephant", 'code' => "4609", 'type' => "Garlic" },
+                   { '__id' => "3401", 'variety' => "One-clove types", 'code' => "3401", 'type' => "Garlic" },
+                   { '__id' => "3052", 'variety' => "String", 'code' => "3052", 'type' => "Garlic" }]
+  end
+
+  it "should remove records for conditions" do
+    q = @db.query
+    q.addcond('type', :streq, 'Orange')
+    q.search.size.should == 11
+    q.searchout.should.be.true
+    q.search.size.should == 0
+  end
+
+  it "should chain search options" do
+    res = @db.query.condition('type', :streq, 'Cucumber').order_by('variety', :strdesc).limit(3).search
+    res.should == ["4596", "4595", "4594"]
+  end
+
+  it "should query with a block" do
+    res = @db.query do |q|
+      q.condition('type', :streq, 'Cucumber')
+      q.order_by('variety', :strdesc)
+      q.limit(3)
+    end
+    res.should == ["4596", "4595", "4594"]
+  end
+
+  it "should find with a block" do
+    res = @db.find do |q|
+      q.condition('type', :streq, 'Cucumber')
+      q.order_by('variety', :strdesc)
+      q.limit(3)
+    end
+    res.should == [{'type'=>"Cucumber", 'variety'=>"Pickling / Gherkin", '__id'=>"4596", 'code'=>"4596"},
+                   {'type'=>"Cucumber", 'variety'=>"Lemon", '__id'=>"4595", 'code'=>"4595"},
+                   {'type'=>"Cucumber", 'variety'=>"Japanese / White", '__id'=>"4594", 'code'=>"4594"}]
+  end
+
+  it "should show query count" do
+    res = @db.prepare_query{ |q| q.condition('type', :streq, 'Cucumber') }.count
+    res.should == 5
+  end
+
+  it "should allow a custom pkey for a result set" do
+    q = @db.query
+    q.condition('type', :streq, 'Cucumber')
+    q.set_pkey(:pk)
+    q.get.should == [{'type'=>"Cucumber", 'code'=>"4592", :pk=>"4592", 'variety'=>"Armenian"},
+                     {'type'=>"Cucumber", 'code'=>"4593", :pk=>"4593", 'variety'=>"English / Hot House / Long Seedless / Telegraph / Continental"},
+                     {'type'=>"Cucumber", 'code'=>"4594", :pk=>"4594", 'variety'=>"Japanese / White"},
+                     {'type'=>"Cucumber", 'code'=>"4595", :pk=>"4595", 'variety'=>"Lemon"},
+                     {'type'=>"Cucumber", 'code'=>"4596", :pk=>"4596", 'variety'=>"Pickling / Gherkin"}]
+  end
+
+  it "should return a query hint" do
+    @db.set_index(:type, :lexical)
+    q = @db.query
+    q.condition(:type, :streq, 'Cucumber')
+    q.order_by(:code)
+    q.limit(3)
+    q.run
+    q.hint.should == "\nusing an index: \"type\" asc (STREQ)\nresult set size: 5\nsorting the result set: \"code\"\n"
+  end
+
+  it "should query multiple servers" do
+    servers = { 1 => TokyoTyrant::Table.new('127.0.0.1', 45001),
+                2 => TokyoTyrant::Table.new('127.0.0.1', 45002) }
+
+    # dummy data
+    servers.each do |account_id, server|
+      server.clear
+      server["#{account_id}/1"] = {
+        :name => "Test 1 (#{account_id})",
+        :id => 1,
+      }
+      server["#{account_id}/2"] = {
+        :name => "Test 2 (#{account_id})",
+        :id => 2,
+      }
+    end
+    ones = proc{ |q| q.add_condition :id, :numeq, 1 }
+    queries = servers.collect{ |account_id,server| server.prepare_query(&ones) }
+    results = TokyoTyrant::Query.parallel_search(*queries)
+    expected = [{ "" => "1/1", "name" => "Test 1 (1)", "id" => "1" },
+                { "" => "2/1", "name" => "Test 1 (2)", "id" => "1" }]
+    results.should == expected
+  end
+end

data/spec/tokyo_tyrant_spec.rb

@@ -0,0 +1,254 @@
+require 'pathname'
+require Pathname(__FILE__).dirname.join('spec_base') unless $root
+
+describe TokyoTyrant::DB, "with an open database" do
+
+  before do
+    @db = TokyoTyrant::DB.new('127.0.0.1', 45000)
+    @db.clear
+  end
+
+  it "should not be nil" do
+    @db.should.not.be.nil
+  end
+
+  it "should close" do
+    @db.close.should.be.true
+    begin
+      @db.close
+    rescue => e
+      e.message.should == 'close error: invalid operation'
+    end
+  end
+
+  it "should return a server expression" do
+    @db.server.should == '127.0.0.1:45000'
+  end
+
+  it "should optimize" do
+    @db.optimize.should.be.true
+  end
+
+  it "should save a value" do
+    @db[:salad] = 'bacon bits'
+    @db[:salad].should == 'bacon bits'
+  end
+  
+  it "should return a value" do
+    @db[:salad] = 'bacon bits'
+    @db[:salad].should == 'bacon bits'
+  end
+
+  it "should accept binary data" do
+    s = "mango#{0.chr}salsa"
+    @db.put(s, s).should.be.true
+    @db[s].should.equal(s)
+  end
+
+  it "should save multiple values" do
+    h = { 'pizza' => 'old forge style',
+          'sandwich' => 'peanut butter jelly',
+          'yogurt' => 'greek',
+          'coffee' => 'black' }
+    @db.mput(h).should.be.empty
+    h.each_pair do |k,v|
+      @db[k].should == v
+    end
+  end
+
+  it "should delete multiple values" do
+    h = { 'pizza' => 'old forge style',
+          'sandwich' => 'peanut butter jelly',
+          'yogurt' => 'greek',
+          'coffee' => 'black' }
+    @db.mput(h)
+    @db.outlist('coffee', 'yogurt').should.be.empty
+    @db.keys.sort.should == ['pizza','sandwich']
+  end
+
+  it "should get multiple values" do
+    h = { 'pizza' => 'old forge style',
+          'sandwich' => 'peanut butter jelly',
+          'yogurt' => 'greek',
+          'coffee' => 'black' }
+    @db.mput(h)
+    @db.mget(h.keys).should == h
+  end
+
+  it "should out a value" do
+    k = :tomato
+    @db[k] = 'green'
+    @db.out(k).should.be.true
+    @db[k].should.be.nil
+    @db.out(k).should.be.false
+  end
+
+  it "should get a value size" do
+    k = :cereal
+    v = 'granola'
+    @db[k] = v
+    @db.vsiz(k).should == v.size
+  end
+
+  it "should check a key" do
+    k = :fruit
+    @db[k] = 'banana'
+    @db.check(k).should.be.true
+    @db.out(k)
+    @db.check(k).should.be.false
+  end
+
+  it "should iterate" do
+    @db[:cheese] = 'gouda'
+    @db[:grapes] = 'green'
+    @db[:oranges] = 'florida'
+    @db[:crackers] = 'triscuit'
+
+    @db.iterinit.should.be.true
+    @db.iternext.should == 'cheese'
+    @db.iternext.should == 'grapes'
+    @db.iternext.should == 'oranges'
+    @db.iternext.should == 'crackers'
+    @db.iternext.should == nil
+  end
+
+  it "should get forward matching keys" do
+    @db['apples/royalgala'] = '4173'
+    @db['apples/grannysmith'] = '4139'
+    @db['bananas/yellow'] = '4011'
+    @db['oranges/shamouti'] = '3027'
+    @db['grapefruit/deepred'] = '4288'
+    @db.fwmkeys('apples').sort.should == ["apples/grannysmith", "apples/royalgala"]
+  end
+
+  it "should delete forward matching keys" do
+    @db['apples/royalgala'] = '4173'
+    @db['apples/grannysmith'] = '4139'
+    @db['bananas/yellow'] = '4011'
+    @db['oranges/shamouti'] = '3027'
+    @db['grapefruit/deepred'] = '4288'
+    @db.delete_keys_with_prefix('apples').should == nil
+    @db.fwmkeys('apples').should.be.empty
+    @db.keys.sort.should == ['bananas/yellow', 'grapefruit/deepred', 'oranges/shamouti']
+  end
+
+  it "should get all keys" do
+    keys = %w[appetizers entree dessert]
+    values = %w[chips chicken\ caeser\ salad ice\ cream]
+    keys.each_with_index do |k,i|
+      @db[k] = values[i]
+    end
+    @db.keys.should == keys
+  end
+
+  it "should get all values" do
+    keys = %w[appetizers entree dessert]
+    values = %w[chips chicken\ caeser\ salad ice\ cream]
+    keys.each_with_index do |k,i|
+      @db[k] = values[i]
+    end
+    @db.values.should == values
+  end
+
+  it "should vanish all records" do
+    @db['chocolate'] = 'dark'
+    @db['tea'] = 'earl grey'
+    @db.empty?.should.be.false
+    @db.vanish.should.be.true
+    @db.empty?.should.be.true
+  end
+
+  it "should count records" do
+    @db['hummus'] = 'chickpeas'
+    @db['falafel'] = 'chickpeas'
+    @db.rnum.should == 2
+  end
+
+  it "should report db size" do
+    @db['rootbeer'] = 'virgils'
+    @db.db_size.should.not == 0
+  end
+
+  it "should fetch a record" do
+    @db.out('beer')
+    @db.fetch('beer'){ 'heineken' }.should == 'heineken'
+    @db['beer'].should == 'heineken'
+    @db.fetch('beer'){ 'becks' }.should == 'heineken'
+  end
+
+  it "should iterate through records" do
+    keys = %w[grapejuice tacoshells rice]
+    values = %w[Kedem Ortega Uncle\ Ben]
+    keys.each_with_index{ |k,i| @db[k] = values[i] }
+
+    i = 0
+    @db.each do |k,v|
+      k.should == keys[i]
+      v.should == values[i]
+      i = i += 1
+    end
+  end
+
+  it "should iterate through keys" do
+    keys = %w[burritos fajitas tacos tostas enchiladas]
+    values = Array.new(keys.size, 'yum')
+    keys.each_with_index{ |k,i| @db[k] = values[i] }
+
+    i = 0
+    @db.each_key do |k|
+      k.should == keys[i]
+      i = i += 1
+    end
+  end
+
+  it "should iterate through values" do
+    keys = %w[falafel humus couscous tabbouleh tzatziki]
+    values = %w[chickpeas chickpeas semolina bulgar yogurt]
+    keys.each_with_index{ |k,i| @db[k] = values[i] }
+
+    i = 0
+    @db.each_value do |v|
+      v.should == values[i]
+      i = i += 1
+    end
+  end
+
+  it "should add serialized integer values" do
+    key = 'counter'
+    @db.out(key)
+    @db.add_int(key, 1).should == 1
+    @db.add_int(key, 1).should == 2
+    @db.get_int(key).should == 2
+  end
+
+  it "should increment integer values" do
+    key = 'counter'
+    @db.out(key)
+    @db.increment(key).should == 1
+    @db.increment(key, 2).should == 3
+    @db.get_int(key).should == 3
+  end
+
+  it "should add serialized double values" do
+    key = 'counter'
+    @db.out(key)
+    @db.add_double(key, 1.0).should.be.close?(1.0, 0.005)
+    @db.add_double(key, 1.0).should.be.close?(2.0, 0.005)
+    @db.get_double(key).should.be.close?(2.0, 0.005)
+  end
+
+  it "should run lua extensions" do
+    @db.ext('echo', 'hello', 'world').should == "hello\tworld"
+  end
+
+  it "should serialize objects that respond to to_tokyo_tyrant" do
+    class Thing
+      def to_tokyo_tyrant
+        "success"
+      end
+    end
+
+    @db["to_tokyo_tyrant"] = Thing.new
+    @db["to_tokyo_tyrant"].should == "success"
+  end
+end