dystopia_core_spec.rb

#
# Specifying rufus-tokyo
#
require File.dirname(__FILE__) + '/spec_base'

require 'rufus/tokyo/dystopia'

FileUtils.mkdir('tmp') rescue nil

describe Rufus::Tokyo::Dystopia::Core do
  after do
    FileUtils.rm_rf( 'tmp/dystopia' )
  end
  
  it 'should use open with a block will auto close the db correctly' do
    res = Rufus::Tokyo::Dystopia::Core.open('tmp/dystopia') do |index|
      index.store( 1, "John Adams" )
      index.store( 3, "George Washington" )
      index.count.should.equal( 2 )
      :result
    end

    res.should.equal(:result)

    index = Rufus::Tokyo::Dystopia::Core.new('tmp/dystopia')
    r = index.search( "John" )
    r.size.should.equal( 1 )
    r.should.equal( [ 1 ] )
    index.close
  end

  it 'should use open without a block just like calling new correctly' do
    index = Rufus::Tokyo::Dystopia::Core.open('tmp/dystopia')
    index.store( 1, "John Adams" )
    index.store( 3, "George Washington" )
    index.count.should.equal( 2 )
    index.close

    index = Rufus::Tokyo::Dystopia::Core.new('tmp/dystopia')
    r = index.search( "John" )
    r.size.should.equal( 1 )
    r.should.equal( [ 1 ] )
    index.close
  end
end

describe 'Rufus::Tokyo::Dystopia::Core' do
  before do
    @db = Rufus::Tokyo::Dystopia::Core.new( 'tmp/dystopia' )
    @db.clear
  end

  after do
    @db.close
    FileUtils.rm_rf( 'tmp/dystopia' )
  end

  it 'creates a new directory when started' do
    File.directory?( 'tmp/dystopia' ).should.equal( true )
  end

  it 'knows its own full path' do
    p = @db.path
    File.directory?( p ).should.equal( true )
    p.should.equal( File.expand_path( "tmp/dystopia" ) )
  end

  it "knows its record count" do
    @db.count.should.equal( 0 )
    @db.store( 1, "John Adams" )
    @db.count.should.equal( 1 )
  end

  it "knows how much drive space it takes up in bytes" do
    b = @db.fsize
    @db.store( 1, "John Adams" )
    @db.store( 3, "George Washington" )
    @db.fsize.should > b
  end

  it "can add records" do
    @db.count.should.equal(0)
    @db.store( 1, "John Adams" )
    @db.store( 3, "George Washington" )
    @db.count.should.equal( 2 )
  end

  it "can remove records" do
    @db.count.should.equal(0)
    @db.store( 1, "John Adams" )
    @db.store( 3, "George Washington" )
    @db.count.should.equal( 2 )
    @db.delete( 3 )
    @db.count.should.equal( 1 )
  end

  it "retrieves the whole record of items it has stored" do
    @db.store( 1, "John Adams" )
    @db.store( 3, "George Washington" )

    @db.fetch( 3 ).should.equal( "George Washington" )
    @db.fetch( 1 ).should.equal( "John Adams" )
  end

  it "returns nil when the fetched document does not exist" do
    @db.store( 1, "John Adams" )
    @db.count.should.equal( 1 )
    @db.fetch( 3 ).should.be.nil
  end

  it "searches for text in the database" do
    [ 'John Adams', 'George Washington', 'Thomas Jefferson' ].each_with_index do |e, idx|
      @db.store( idx + 1, e )
    end
    r = @db.search( "on" )
    r.size.should.equal( 2 )
    r.sort.should.equal( [ 2, 3 ] )

    r = @db.search( "John" )
    r.size.should.equal( 1 )
    r.should.equal( [ 1 ] )
  end
end