shard-crystal-db/spec/std/db/dummy_driver_spec.cr

require "spec"
require "db"
require "./dummy_driver"

describe DummyDriver do
  it "should return statements" do
    get_dummy.prepare("the query").should be_a(DB::Statement)
  end

  describe DummyDriver::DummyStatement do
    it "exec should return a result_set" do
      statement = get_dummy.prepare("a,b 1,2")
      result_set = statement.exec
      result_set.should be_a(DB::ResultSet)
      result_set.statement.should be(statement)
    end

    it "should enumerate records by spaces" do
      result_set = get_dummy.prepare("").exec
      result_set.has_next.should be_false

      result_set = get_dummy.prepare("a,b").exec
      result_set.has_next.should be_true
      result_set.has_next.should be_false

      result_set = get_dummy.prepare("a,b 1,2").exec
      result_set.has_next.should be_true
      result_set.has_next.should be_true
      result_set.has_next.should be_false

      result_set = get_dummy.prepare("a,b 1,2 c,d").exec
      result_set.has_next.should be_true
      result_set.has_next.should be_true
      result_set.has_next.should be_true
      result_set.has_next.should be_false
    end

    it "should enumerate string fields" do
      result_set = get_dummy.prepare("a,b 1,2").exec
      result_set.has_next
      result_set.read(String).should eq("a")
      result_set.read(String).should eq("b")
      result_set.has_next
      result_set.read(String).should eq("1")
      result_set.read(String).should eq("2")
    end

    it "should enumerate uint64 fields" do
      result_set = get_dummy.prepare("3,4 1,2").exec
      result_set.has_next
      result_set.read(UInt64).should eq(3)
      result_set.read(UInt64).should eq(4)
      result_set.has_next
      result_set.read(UInt64).should eq(1)
      result_set.read(UInt64).should eq(2)
    end

    it "should enumerate records using each" do
      nums = [] of UInt64
      result_set = get_dummy.prepare("3,4 1,2").exec
      result_set.each do
        nums << result_set.read(UInt64)
        nums << result_set.read(UInt64)
      end

      nums.should eq([3, 4, 1, 2])
    end
  end
end