chore: refactor dotnet lib

Author: olavloite

spannerlib/dotnet-spannerlib/Google.Cloud.SpannerLib.Tests/BasicTests.cs

@@ -1,16 +1,83 @@
-namespace Google.Cloud.SpannerLib.Tests;
+using Google.Cloud.Spanner.V1;
+using Google.Cloud.SpannerLib.Tests.MockServer;
+
+namespace Google.Cloud.SpannerLib.Tests;
 
 public class BasicTests
 {
+    private SpannerMockServerFixture _fixture;
+        
+    private string ConnectionString =>  $"{_fixture.Host}:{_fixture.Port}/projects/p1/instances/i1/databases/d1;UsePlainText=true";
+        
     [SetUp]
     public void Setup()
     {
+        _fixture = new SpannerMockServerFixture();
+        _fixture.SpannerMock.AddOrUpdateStatementResult("SELECT 1", StatementResult.CreateSelect1ResultSet());
+    }
+        
+    [TearDown]
+    public void Teardown()
+    {
+        _fixture.Dispose();
     }
 
     [Test]
     public void TestCreatePool()
     {
-        var pool = Spanner.CreatePool("projects/my-project/instances/my-instance/databases/my-database");
-        Spanner.ClosePool(pool);
+        var pool = LibPool.Create(ConnectionString);
+        pool.Close();
+    }
+
+    [Test]
+    public void TestCreateConnection()
+    {
+        using var pool = LibPool.Create(ConnectionString);
+        var connection = pool.CreateConnection();
+        connection.Close();
+    }
+
+    [Test]
+    public void TestExecuteQuery()
+    {
+        using var pool = LibPool.Create(ConnectionString);
+        using var connection = pool.CreateConnection();
+        using var rows = connection.Execute(new ExecuteSqlRequest { Sql = "SELECT 1" });
+        for (var row = rows.Next(); row != null; row = rows.Next())
+        {
+            Assert.That(row.Values.Count, Is.EqualTo(1));
+            Assert.That(row.Values[0].HasStringValue);
+            Assert.That(row.Values[0].StringValue, Is.EqualTo("1"));
+        }
+    }
+
+    [Test]
+    public void TestReadOnlyTransaction()
+    {
+        using var pool = LibPool.Create(ConnectionString);
+        using var connection = pool.CreateConnection();
+        using var transaction = connection.BeginTransaction(new TransactionOptions
+            { ReadOnly = new TransactionOptions.Types.ReadOnly() });
+        using var rows = transaction.Execute(new ExecuteSqlRequest { Sql = "SELECT 1" });
+        for (var row = rows.Next(); row != null; row = rows.Next())
+        {
+            Assert.That(row.Values.Count, Is.EqualTo(1));
+            Assert.That(row.Values[0].HasStringValue);
+            Assert.That(row.Values[0].StringValue, Is.EqualTo("1"));
+        }
+    }
+
+    [Test]
+    public void TestReadWriteTransaction()
+    {
+        var sql = "update table1 set value='one' where id=1";
+        _fixture.SpannerMock.AddOrUpdateStatementResult(sql, StatementResult.CreateUpdateCount(1));
+        using var pool = LibPool.Create(ConnectionString);
+        using var connection = pool.CreateConnection();
+        using var transaction = connection.BeginTransaction(new TransactionOptions
+            { ReadWrite = new TransactionOptions.Types.ReadWrite() });
+        using var rows = transaction.Execute(new ExecuteSqlRequest { Sql = sql });
+        Assert.That(rows.UpdateCount, Is.EqualTo(1));
+        transaction.Commit();
     }
 }

spannerlib/dotnet-spannerlib/Google.Cloud.SpannerLib.Tests/Google.Cloud.SpannerLib.Tests.csproj

@@ -10,6 +10,7 @@
 
   <ItemGroup>
     <PackageReference Include="coverlet.collector" Version="6.0.2"/>
+    <PackageReference Include="Grpc.AspNetCore.Server" Version="2.71.0" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.12.0"/>
     <PackageReference Include="NUnit" Version="4.2.2"/>
     <PackageReference Include="NUnit.Analyzers" Version="4.4.0"/>

spannerlib/dotnet-spannerlib/Google.Cloud.SpannerLib.Tests/MockServer/MockServerStartup.cs

@@ -0,0 +1,74 @@
+// 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.
+
+using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Hosting;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+
+namespace Google.Cloud.SpannerLib.Tests.MockServer;
+
+/// <summary>
+/// Helper class for starting an in-memory mock Spanner server that is used for testing.
+/// </summary>
+public class MockServerStartup
+{
+    /// <summary>
+    /// The in-mem Spanner service.
+    /// </summary>
+    private MockSpannerService MockSpannerService { get; }
+
+    /// <summary>
+    /// The in-mem Spanner database admin service for executing DDL operations.
+    /// </summary>
+    private MockDatabaseAdminService MockDatabaseAdminService { get; }
+        
+    public MockServerStartup(MockSpannerService mockSpannerService, MockDatabaseAdminService mockDatabaseAdminService)
+    {
+        MockSpannerService = mockSpannerService;
+        MockDatabaseAdminService = mockDatabaseAdminService;
+    }
+
+    /// <summary>
+    /// Configures the services that will be available on this gRPC server.
+    /// This method is called by reflection when the tests are started.
+    /// </summary>
+    /// <param name="services">The services collection where the services should be added</param>
+    public void ConfigureServices(IServiceCollection services)
+    {
+        services.AddGrpc();
+        services.AddSingleton(MockSpannerService);
+        services.AddSingleton(MockDatabaseAdminService);
+    }
+
+    /// <summary>
+    /// Configures the gRPC server. This method is called by reflection when the tests are started.
+    /// </summary>
+    /// <param name="app">The builder for the application that will be hosting the service</param>
+    /// <param name="env">The webhost environment that is hosting the service</param>
+    public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
+    {
+        if (env.IsDevelopment())
+        {
+            app.UseDeveloperExceptionPage();
+        }
+        app.UseRouting();
+        app.UseEndpoints(endpoints =>
+        {
+            endpoints.MapGrpcService<MockSpannerService>();
+            endpoints.MapGrpcService<MockDatabaseAdminService>();
+        });
+    }
+}
+