heroku-tokyotyrant 0.4

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

data/spec/tokyo_tyrant_table_spec.rb

@@ -0,0 +1,301 @@
+require 'pathname'
+require Pathname(__FILE__).dirname.join('spec_base') unless $root
+
+describe TokyoTyrant::Table, "with an open database" do
+
+  before do
+    @db = TokyoTyrant::Table.new('127.0.0.1', 45001)
+    @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:45001'
+  end
+
+  it "should optimize" do
+    @db.optimize.should.be.true
+  end
+
+  it "should save a value" do
+    value = { 'lettuce' => 'Red Leaf', 'dressing' => 'ranch', 'extra' => 'bacon bits' }
+    @db[:salad] = value
+    @db[:salad].should == value
+  end
+  
+  it "should return a value" do
+    value = { 'lettuce' => 'Red Leaf', 'dressing' => 'ranch', 'extra' => 'bacon bits' }
+    @db[:salad] = value
+    @db[:salad].should == value
+  end
+
+  it "should accept binary data" do
+    s = "mango#{0.chr}salsa"
+    h = { s => s }
+    @db.put(s, h).should.be.true
+    @db[s].should.equal(h)
+  end
+
+  it "should save multiple values" do
+    h = { 'pizza' => { 'best' => 'old forge style', 'ok' => 'new york style', 'worst' => 'chicago style' },
+          'sandwich' => { 'best' => 'peanut butter jelly', 'ok' => 'turkey', 'worst' => 'olive loaf' },
+          'yogurt' => { 'best' => 'greek', 'ok' => 'organic', 'worst' => '+hfcs' },
+          'coffee' => { 'best' => 'black', 'ok' => 'espresso', 'worst' => 'latte' } }
+
+    @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' => { 'best' => 'old forge style', 'ok' => 'new york style', 'worst' => 'chicago style' },
+          'sandwich' => { 'best' => 'peanut butter jelly', 'ok' => 'turkey', 'worst' => 'olive loaf' },
+          'yogurt' => { 'best' => 'greek', 'ok' => 'organic', 'worst' => '+hfcs' },
+          'coffee' => { 'best' => 'black', 'ok' => 'espresso', 'worst' => 'latte' } }
+
+    @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' => { 'best' => 'old forge style', 'ok' => 'new york style', 'worst' => 'chicago style' },
+          'sandwich' => { 'best' => 'peanut butter jelly', 'ok' => 'turkey', 'worst' => 'olive loaf' },
+          'yogurt' => { 'best' => 'greek', 'ok' => 'organic', 'worst' => '+hfcs' },
+          'coffee' => { 'best' => 'black', 'ok' => 'espresso', 'worst' => 'latte' } }
+
+    @db.mput(h)
+    @db.mget(h.keys).should == h
+  end
+
+  it "should out a value" do
+    k = :tomato
+    @db[k] = { 'color' => 'red', 'variety' => 'beefy' }
+    @db.out(k).should.be.true
+    @db[k].should.be.nil
+    @db.out(k).should.be.false
+  end
+
+  it "should check a key" do
+    k = :fruit
+    @db[k] = { 'name' => 'banana', 'code' => '4011' }
+    @db.check(k).should.be.true
+    @db.out(k)
+    @db.check(k).should.be.false
+  end
+
+  it "should iterate" do
+    @db[:cheese] = { 'melty' => 'gouda', 'sharp' => 'cheddar', 'stringy' => 'mozerella' }
+    @db[:grapes] = { 'sour' => 'green', 'big' => 'red', 'wine' => 'purple' }
+    @db[:oranges] = { 'juicy' => 'florida', 'yellow' => 'california' }
+    @db[:crackers] = { 'wheat' => 'triscuit', 'snack' => 'ritz', 'soup' => 'saltine' }
+
+    @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'] = { 'code' => '4173', 'color' => 'red-yellow' }
+    @db['apples/grannysmith'] = { 'code' => '4139', 'color' => 'green' }
+    @db['bananas/yellow'] = { 'code' => '4011', 'color' => 'yellow' }
+    @db['oranges/shamouti'] = { 'code' => '3027', 'color' => 'orange' }
+    @db['grapefruit/deepred'] = { 'code' => '4288', 'color' => 'yellow/pink' }
+    @db.fwmkeys('apples').sort.should == ["apples/grannysmith", "apples/royalgala"]
+  end
+
+  it "should delete forward matching keys" do
+    @db['apples/royalgala'] = { 'code' => '4173', 'color' => 'red-yellow' }
+    @db['apples/grannysmith'] = { 'code' => '4139', 'color' => 'green' }
+    @db['bananas/yellow'] = { 'code' => '4011', 'color' => 'yellow' }
+    @db['oranges/shamouti'] = { 'code' => '3027', 'color' => 'orange' }
+    @db['grapefruit/deepred'] = { 'code' => '4288', 'color' => 'yellow/pink' }
+    @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 = [{ 'cheap' => 'chips', 'expensive' => 'sample everything platter' },
+              { 'cheap' => 'hoagie', 'expensive' => 'steak' },
+              { 'cheap' => 'water ice', 'expensive' => 'cheesecake' }]
+
+    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 = [{ 'cheap' => 'chips', 'expensive' => 'sample everything platter' },
+              { 'cheap' => 'hoagie', 'expensive' => 'steak' },
+              { 'cheap' => 'water ice', 'expensive' => 'cheesecake' }]
+
+    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'] = { 'type' => 'dark' }
+    @db['tea'] = { 'type' => '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'] = { 'ingredients' => 'chickpeas,garlic' }
+    @db['falafel'] = { 'ingredients' => 'chickpeas,herbs' }
+    @db.rnum.should == 2
+  end
+
+  it "should report db size" do
+    @db['rootbeer'] = { 'gourmet' => 'Virgils', 'natural' => 'Hansens' }
+    @db.db_size.should.not == 0
+  end
+
+  it "should fetch a record" do
+    @db.out('beer')
+    @db.fetch('beer'){{ 'import' => 'heineken' }}.should == { 'import' => 'heineken' }
+    @db['beer'].should == { 'import' => 'heineken' }
+    @db.fetch('beer'){{ 'import' => 'becks' }}.should == { 'import' => 'heineken' }
+  end
+
+  it "should iterate through records" do
+    keys = %w[grapejuice tacoshells rice]
+    values = [{ 'purple' => 'Kedem', 'green' => 'Juicy Juice' },
+              { 'crunchy' => 'Ortega', 'soft' => 'Taco Bell' },
+              { 'brown' => 'Instant', 'white' => '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, { 'tasty' => 'yes' })
+    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 = [{ 'ingredient' => 'chickpeas' },
+              { 'ingredient' => 'chickpeas' },
+              { 'ingredient' => 'semolina' },
+              { 'ingredient' => 'bulgar' },
+              { 'ingredient' => '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[key] = { 'title' => 'Bean Counter' }
+    @db.add_int(key, 1).should == 1
+    @db.add_int(key, 1).should == 2
+    @db.get_int(key).should == 2
+    @db[key].should == { 'title' => 'Bean Counter', '_num' => "2" }
+  end
+
+  it "should increment integer values" do
+    key = 'counter'
+    @db.out(key)
+    @db[key] = { 'title' => 'Bean Counter' }
+    @db.increment(key).should == 1
+    @db.increment(key, 2).should == 3
+    @db.get_int(key).should == 3
+    @db[key].should == { 'title' => 'Bean Counter', '_num' => "3" }
+  end
+
+  it "should add serialized double values" do
+    key = 'counter'
+    @db.out(key)
+    @db[key] = { 'title' => 'Bean Counter' }
+    @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)
+    @db[key].should == { 'title' => 'Bean Counter', '_num' => "2" }
+  end
+
+  it "should set an index" do
+    key = 'counter'
+    50.times do |i|
+      @db["key#{i}"] = { 'title' => %w{rice beans corn}.sort_by{rand}.first,
+                         'description' => 'a whole protein' }
+    end
+    @db.set_index('title', :lexical).should.be.true
+  end
+
+  it "should run lua extensions" do
+    @db.ext('echo', 'hello', 'world').should == "hello\tworld"
+  end
+
+  it "should perform meta search with multiple query objects" do
+    keys = %w[falafel humus couscous tabbouleh tzatziki peanutbutter]
+    values = [{ 'ingredient' => 'chickpeas' },
+              { 'ingredient' => 'chickpeas' },
+              { 'ingredient' => 'semolina' },
+              { 'ingredient' => 'bulgar' },
+              { 'ingredient' => 'yogurt' },
+              { 'ingredient' => 'peanuts'}]
+
+    keys.each_with_index{ |k,i| @db[k] = values[i] }
+
+    query1 = @db.prepare_query{ |q| q.condition('ingredient', :streq, 'yogurt').order_by('ingredient') }
+    query2 = @db.prepare_query{ |q| q.condition('ingredient', :streq, 'chickpeas').order_by('ingredient') }
+    query3 = @db.prepare_query{ |q| q.condition('ingredient', :ftsex, 'pea').order_by('ingredient') }
+
+    @db.search(:union, query1, query2).should == ["falafel", "humus", "tzatziki"]
+    @db.search(:diff, query1, query2).should == ["tzatziki"]
+    @db.search(:intersection, query2, query3).should == ["falafel", "humus"]
+  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" => Thing.new }
+    @db["to_tokyo_tyrant"].should == { "thing" => "success" }
+  end
+end