spannerlib-ruby 0.1.0.alpha1

data/lib/spannerlib/rows.rb

@@ -0,0 +1,67 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+module SpannerLib
+  class Rows
+    include Enumerable
+
+    attr_reader :id, :connection
+
+    def initialize(connection, rows_id)
+      @connection = connection
+      @id = rows_id
+      @closed = false
+    end
+
+    def each
+      return enum_for(:each) unless block_given?
+
+      while (row = self.next)
+        yield row
+      end
+    ensure
+      close
+    end
+
+    def next
+      return nil if @closed
+
+      row_data = SpannerLib.next(connection.pool_id, connection.conn_id, id, 1, 0)
+
+      if row_data.nil? || row_data.empty? || (row_data.respond_to?(:values) && row_data.values.empty?)
+        close
+        return nil
+      end
+
+      row_data
+    end
+
+    def metadata
+      SpannerLib.metadata(connection.pool_id, connection.conn_id, id)
+    end
+
+    def result_set_stats
+      SpannerLib.result_set_stats(connection.pool_id, connection.conn_id, id)
+    end
+
+    def close
+      return if @closed
+
+      SpannerLib.close_rows(connection.pool_id, connection.conn_id, id)
+      @closed = true
+    end
+  end
+end

data/lib/spannerlib/ruby/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Spannerlib
+  module Ruby
+    VERSION = "0.1.0.alpha1"
+  end
+end

data/lib/spannerlib/ruby.rb

@@ -0,0 +1,24 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+require_relative "ruby/version"
+
+module Spannerlib
+  module Ruby
+    class Error < StandardError; end
+    # Your code goes here...
+  end
+end

data/sig/spannerlib/ruby.rbs

@@ -0,0 +1,6 @@
+module Spannerlib
+  module Ruby
+    VERSION: String
+    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+  end
+end

data/spannerlib-ruby.gemspec

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require_relative "lib/spannerlib/ruby/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "spannerlib-ruby"
+  spec.version = Spannerlib::Ruby::VERSION
+  spec.authors = ["Google LLC"]
+  spec.email = ["cloud-spanner-developers@googlegroups.com"]
+
+  spec.summary = "Ruby wrapper for the Spanner native library"
+  spec.description = "Lightweight Ruby FFI bindings for the Spanner native library produced from the Go implementation."
+  spec.homepage = "https://github.com/googleapis/go-sql-spanner/tree/main/spannerlib/wrappers/spannerlib-ruby"
+  spec.license = "Apache 2.0 License"
+  spec.required_ruby_version = ">= 3.1.0"
+
+  spec.metadata["rubygems_mfa_required"] = "true"
+
+  # Include both git-tracked files (for local development) and any built native libraries
+  # that exist on disk (for CI). We do this so:
+  #  - During local development, spec.files is driven by `git ls-files` (keeps the gem manifest clean).
+  #  - In CI we build native shared libraries into lib/spannerlib/<platform>/ and those files are
+  #    not checked into git; we therefore also glob lib/spannerlib/** to pick up the CI-built binaries
+  #    so the gem produced in CI actually contains the native libraries.
+  #  - We explicitly filter out common build artifacts and non-distributable files to avoid accidentally
+  #    packaging object files, headers, or temporary files.
+  # This allows us to publish a single multi-platform gem that contains prebuilt shared libraries
+
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    files = []
+    # prefer git-tracked files when available (local dev), but also pick up built files present on disk (CI)
+    files += `git ls-files -z`.split("\x0") if system("git rev-parse --is-inside-work-tree > /dev/null 2>&1")
+
+    # include any built native libs (CI places them under lib/spannerlib/)
+    files += Dir.glob("lib/spannerlib/**/*").select { |f| File.file?(f) }
+
+    # dedupe and reject unwanted entries
+    files.map! { |f| f.sub(%r{\A\./}, "") }.uniq!
+    files.reject do |f|
+      f.match(%r{^(pkg|Gemfile\.lock|.*\.gem|Rakefile|spec/|.*\.o|.*\.h)$})
+    end
+  end
+
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "ffi"
+  spec.add_dependency "google-cloud-spanner-v1", "~> 1.7"
+  spec.add_dependency "google-protobuf", "~> 3.19"
+  spec.add_dependency "grpc", "~> 1.60"
+end

data/spec/integration/batch_emulator_spec.rb

@@ -0,0 +1,165 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+require "spec_helper"
+require "google/cloud/spanner/v1"
+
+RSpec.describe "Batch API test", :integration do
+  before(:all) do
+    @emulator_host = ENV.fetch("SPANNER_EMULATOR_HOST", nil)
+    skip "SPANNER_EMULATOR_HOST not set" unless @emulator_host && !@emulator_host.empty?
+
+    require "spannerlib/pool"
+    @dsn = "projects/your-project-id/instances/test-instance/databases/test-database?autoConfigEmulator=true"
+
+    @pool = Pool.create_pool(@dsn)
+    conn = @pool.create_connection
+    ddl_batch_req = Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest.new(
+      statements: [
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(sql: "DROP TABLE IF EXISTS test_table"),
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "CREATE TABLE test_table (id INT64 NOT NULL, name STRING(100)) PRIMARY KEY(id)"
+        )
+      ]
+    )
+    conn.execute_batch(ddl_batch_req)
+    conn.close
+  end
+
+  after(:all) do
+    @pool&.close
+  end
+
+  before do
+    @conn = @pool.create_connection
+    delete_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          delete: Google::Cloud::Spanner::V1::Mutation::Delete.new(
+            table: "test_table",
+            key_set: Google::Cloud::Spanner::V1::KeySet.new(all: true)
+          )
+        )
+      ]
+    )
+    @conn.write_mutations(delete_req)
+  end
+
+  after do
+    @conn&.close
+  end
+
+  it "tests a batch DML request" do
+    dml_batch_req = Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest.new(
+      statements: [
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "INSERT INTO test_table (id, name) VALUES (1, 'name1')"
+        ),
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "INSERT INTO test_table (id, name) VALUES (2, 'name2')"
+        ),
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "UPDATE test_table SET name='name3' WHERE id=1"
+        )
+      ]
+    )
+    resp = @conn.execute_batch(dml_batch_req)
+    expect(resp.result_sets.length).to eq 3
+
+    select_req = Google::Cloud::Spanner::V1::ExecuteSqlRequest.new(
+      sql: "SELECT id, name FROM test_table ORDER BY id"
+    )
+    rows = @conn.execute(select_req)
+    all_rows = rows.map { |row_bytes| Google::Protobuf::ListValue.decode(row_bytes) }
+
+    expect(all_rows.length).to eq 2
+    expect(all_rows[0].values[0].string_value).to eq "1"
+    expect(all_rows[0].values[1].string_value).to eq "name3"
+    expect(all_rows[1].values[0].string_value).to eq "2"
+    expect(all_rows[1].values[1].string_value).to eq "name2"
+  end
+
+  it "tests a batch DDL request" do
+    ddl_batch_req = Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest.new(
+      statements: [
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "DROP TABLE IF EXISTS test_table"
+        ),
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "CREATE TABLE test_table (key INT64 NOT NULL, data STRING(MAX)) PRIMARY KEY(key)"
+        )
+      ]
+    )
+
+    expect { @conn.execute_batch(ddl_batch_req) }.not_to raise_error
+
+    insert_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          insert: Google::Cloud::Spanner::V1::Mutation::Write.new(
+            table: "test_table",
+            columns: %w[key data],
+            values: [
+              Google::Protobuf::ListValue.new(values: [
+                                                Google::Protobuf::Value.new(string_value: "101"),
+                                                Google::Protobuf::Value.new(string_value: "VerificationData")
+                                              ])
+            ]
+          )
+        )
+      ]
+    )
+    expect { @conn.write_mutations(insert_req) }.not_to raise_error
+  end
+
+  it "queries data using parameters" do
+    insert_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          insert: Google::Cloud::Spanner::V1::Mutation::Write.new(
+            table: "test_table",
+            columns: %w[key data],
+            values: [
+              Google::Protobuf::ListValue.new(values: [Google::Protobuf::Value.new(string_value: "1"),
+                                                       Google::Protobuf::Value.new(string_value: "Alice")]),
+              Google::Protobuf::ListValue.new(values: [Google::Protobuf::Value.new(string_value: "2"),
+                                                       Google::Protobuf::Value.new(string_value: "Bob")])
+            ]
+          )
+        )
+      ]
+    )
+    @conn.write_mutations(insert_req)
+
+    # Execute the parameterized query.
+    select_req = Google::Cloud::Spanner::V1::ExecuteSqlRequest.new(
+      sql: "SELECT key, data FROM test_table WHERE data = @dataParam",
+      params: Google::Protobuf::Struct.new(
+        fields: {
+          "dataParam" => Google::Protobuf::Value.new(string_value: "Alice")
+        }
+      ),
+      param_types: {
+        "dataParam" => Google::Cloud::Spanner::V1::Type.new(code: Google::Cloud::Spanner::V1::TypeCode::STRING)
+      }
+    )
+    rows = @conn.execute(select_req)
+    all_rows = rows.map { |row_bytes| Google::Protobuf::ListValue.decode(row_bytes) }
+
+    expect(all_rows.length).to eq(1)
+    expect(all_rows[0].values[1].string_value).to eq("Alice")
+  end
+end

data/spec/integration/connection_emulator_spec.rb

@@ -0,0 +1,151 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+require "spec_helper"
+require "google/cloud/spanner/v1"
+
+RSpec.describe "Connection APIs against Spanner emulator", :integration do
+  before(:all) do
+    @emulator_host = ENV.fetch("SPANNER_EMULATOR_HOST", nil)
+    skip "SPANNER_EMULATOR_HOST not set" unless @emulator_host && !@emulator_host.empty?
+
+    require "spannerlib/pool"
+    @dsn = "projects/your-project-id/instances/test-instance/databases/test-database?autoConfigEmulator=true"
+
+    pool = Pool.create_pool(@dsn)
+    conn = pool.create_connection
+    ddl_batch_req = Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest.new(
+      statements: [
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(sql: "DROP TABLE IF EXISTS test_table"),
+        Google::Cloud::Spanner::V1::ExecuteBatchDmlRequest::Statement.new(
+          sql: "CREATE TABLE test_table (id INT64 NOT NULL, name STRING(100)) PRIMARY KEY(id)"
+        )
+      ]
+    )
+    conn.execute_batch(ddl_batch_req)
+    conn.close
+    pool.close
+  end
+
+  before do
+    @pool = Pool.create_pool(@dsn)
+    @conn = @pool.create_connection
+    delete_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          delete: Google::Cloud::Spanner::V1::Mutation::Delete.new(
+            table: "test_table",
+            key_set: Google::Cloud::Spanner::V1::KeySet.new(all: true)
+          )
+        )
+      ]
+    )
+    @conn.write_mutations(delete_req)
+  end
+
+  after do
+    @conn.close
+    @pool.close
+  end
+
+  it "creates a connection pool" do
+    expect(@pool.id).to be > 0
+    expect(@conn.conn_id).to be > 0
+  end
+
+  it "creates two connections from the same pool" do
+    conn2 = @pool.create_connection
+    expect(@conn.conn_id).not_to eq(conn2.conn_id)
+    expect(conn2.conn_id).to be > 0
+    conn2.close
+  end
+
+  it "writes and reads data in a read-write transaction" do
+    @conn.begin_transaction
+    insert_data_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          insert: Google::Cloud::Spanner::V1::Mutation::Write.new(
+            table: "test_table",
+            columns: %w[id name],
+            values: [
+              Google::Protobuf::ListValue.new(values: [Google::Protobuf::Value.new(string_value: "1"),
+                                                       Google::Protobuf::Value.new(string_value: "Alice")]),
+              Google::Protobuf::ListValue.new(values: [Google::Protobuf::Value.new(string_value: "2"),
+                                                       Google::Protobuf::Value.new(string_value: "Bob")])
+            ]
+          )
+        )
+      ]
+    )
+    @conn.write_mutations(insert_data_req)
+    @conn.commit
+
+    select_req = Google::Cloud::Spanner::V1::ExecuteSqlRequest.new(sql: "SELECT id, name FROM test_table ORDER BY id")
+    rows = @conn.execute(select_req)
+
+    all_rows = rows.map { |row_bytes| Google::Protobuf::ListValue.decode(row_bytes) }
+
+    expect(all_rows.length).to eq(2)
+    expect(all_rows[0].values[1].string_value).to eq("Alice")
+  end
+
+  it "writes and reads data without an explicit transaction (autocommit)" do
+    insert_data_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          insert: Google::Cloud::Spanner::V1::Mutation::Write.new(
+            table: "test_table",
+            columns: %w[id name],
+            values: [
+              Google::Protobuf::ListValue.new(values: [Google::Protobuf::Value.new(string_value: "3"),
+                                                       Google::Protobuf::Value.new(string_value: "Charlie")]),
+              Google::Protobuf::ListValue.new(values: [Google::Protobuf::Value.new(string_value: "4"),
+                                                       Google::Protobuf::Value.new(string_value: "David")])
+            ]
+          )
+        )
+      ]
+    )
+    @conn.write_mutations(insert_data_req)
+
+    select_req = Google::Cloud::Spanner::V1::ExecuteSqlRequest.new(sql: "SELECT id, name FROM test_table ORDER BY id")
+    rows = @conn.execute(select_req)
+
+    all_rows = rows.map { |row_bytes| Google::Protobuf::ListValue.decode(row_bytes) }
+
+    expect(all_rows.length).to eq(2)
+    expect(all_rows[0].values[1].string_value).to eq("Charlie")
+  end
+
+  it "raises an error when writing in a read-only transaction" do
+    transaction_options = Google::Cloud::Spanner::V1::TransactionOptions.new(
+      read_only: Google::Cloud::Spanner::V1::TransactionOptions::ReadOnly.new(strong: true)
+    )
+    @conn.begin_transaction(transaction_options)
+
+    insert_data_req = Google::Cloud::Spanner::V1::BatchWriteRequest::MutationGroup.new(
+      mutations: [
+        Google::Cloud::Spanner::V1::Mutation.new(
+          insert: Google::Cloud::Spanner::V1::Mutation::Write.new(table: "test_table")
+        )
+      ]
+    )
+    expect { @conn.write_mutations(insert_data_req) }.to raise_error(SpannerLibException, /read-only transactions cannot write/)
+
+    @conn.rollback
+  end
+end

data/spec/integration/pool_emulator_spec.rb

@@ -0,0 +1,42 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+require "spec_helper"
+
+RSpec.describe "Connection APIs against Spanner emulator", :integration do
+  before(:all) do
+    @emulator_host = ENV.fetch("SPANNER_EMULATOR_HOST", nil)
+    skip "SPANNER_EMULATOR_HOST not set; skipping emulator integration tests" unless @emulator_host && !@emulator_host.empty?
+
+    begin
+      require "spannerlib/pool"
+    rescue LoadError, StandardError => e
+      skip "Could not load native spanner library; skipping emulator integration tests: #{e.class}: #{e.message}"
+    end
+    @dsn = "projects/your-project-id/instances/test-instance/databases/test-database?autoConfigEmulator=true"
+  end
+
+  it "creates a pool and a connection against the emulator" do
+    pool = Pool.create_pool(@dsn)
+    expect(pool.id).to be > 0
+
+    conn = pool.create_connection
+    expect(conn).to respond_to(:pool_id)
+    expect(conn).to respond_to(:conn_id)
+
+    expect { pool.close }.not_to raise_error
+  end
+end

data/spec/spannerlib/connection_spec.rb

@@ -0,0 +1,53 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language gove
+
+# frozen_string_literal: true
+
+require "spec_helper"
+require "spannerlib/pool"
+require "spannerlib/ffi"
+require "spannerlib/exceptions"
+
+RSpec.describe Connection do
+  let(:dsn) { "localhost:1234/projects/p/instances/i/databases/d?usePlainText=true" }
+  let(:pool) { Pool.create_pool(dsn) }
+
+  before do
+    allow(SpannerLib).to receive(:create_pool).and_return(1)
+  end
+
+  describe "creation" do
+    it "is created by a Pool" do
+      allow(SpannerLib).to receive(:create_connection).with(1).and_return(2)
+
+      # The object under test is the one returned by `pool.create_connection`
+      conn = pool.create_connection
+
+      expect(conn).to be_a(described_class)
+      expect(conn.conn_id).to eq(2)
+      expect(conn.pool_id).to eq(1)
+    end
+
+    it "raises a SpannerLibException when the FFI call fails" do
+      allow(SpannerLib).to receive(:create_connection).with(1).and_raise(StandardError.new("boom"))
+
+      expect { pool.create_connection }.to raise_error(SpannerLibException)
+    end
+
+    it "raises when the FFI call returns a non-positive id" do
+      allow(SpannerLib).to receive(:create_connection).with(1).and_return(0)
+
+      expect { pool.create_connection }.to raise_error(SpannerLibException)
+    end
+  end
+end

data/spec/spannerlib/pool_spec.rb

@@ -0,0 +1,53 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+require "spec_helper"
+require "spannerlib/pool"
+require "spannerlib/ffi"
+require "spannerlib/exceptions"
+
+RSpec.describe Pool do
+  let(:dsn) { "localhost:1234/projects/p/instances/i/databases/d?usePlainText=true" }
+
+  describe ".create_pool" do
+    it "creates a pool and returns an object with id > 0" do
+      allow(SpannerLib).to receive(:create_pool).with(dsn).and_return(42)
+
+      pool = described_class.create_pool(dsn)
+
+      expect(pool).to be_a(described_class)
+      expect(pool.id).to be > 0
+    end
+
+    it "raises a SpannerLibException when create_session/create_pool fails" do
+      allow(SpannerLib).to receive(:create_pool).with(dsn).and_raise(StandardError.new("Not allowed"))
+
+      expect { described_class.create_pool(dsn) }.to raise_error(SpannerLibException)
+    end
+
+    it "raises when create_pool returns nil" do
+      allow(SpannerLib).to receive(:create_pool).with(dsn).and_return(nil)
+
+      expect { described_class.create_pool(dsn) }.to raise_error(SpannerLibException)
+    end
+
+    it "raises when create_pool returns a non-positive id" do
+      allow(SpannerLib).to receive(:create_pool).with(dsn).and_return(0)
+
+      expect { described_class.create_pool(dsn) }.to raise_error(SpannerLibException)
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,32 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# frozen_string_literal: true
+
+require "rubygems"
+require "bundler/setup"
+
+require "spannerlib/ruby"
+
+RSpec.configure do |config|
+  # Enable flags like --only-failures and --next-failure
+  config.example_status_persistence_file_path = ".rspec_status"
+
+  # Disable RSpec exposing methods globally on `Module` and `main`
+  config.disable_monkey_patching!
+
+  config.expect_with :rspec do |c|
+    c.syntax = :expect
+  end
+end