Adding support for using a test connection string.

TrackableEntities · Feb 18, 2024 · 73174a3 · 73174a3
1 parent a3289cd
commit 73174a3
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 6 deletions.
diff --git a/TrackableEntities.EF.Core.Tests/Helpers/DbContextOptionsHelper.cs b/TrackableEntities.EF.Core.Tests/Helpers/DbContextOptionsHelper.cs
@@ -0,0 +1,30 @@
+using System;
+
+using Microsoft.Data.SqlClient;
+using Microsoft.EntityFrameworkCore;
+
+namespace TrackableEntities.EF.Core.Tests.Helpers
+{
+    public static class DbContextOptionsHelper
+    {
+        public static DbContextOptions<TContext> GetContextOptions<TContext>(string initialCatalog)
+            where TContext : DbContext
+        {
+            // default connection string for tests 'LocalDb'
+            var connectionString =
+                @"Data Source=(localdb)\MSSQLLocalDB; Initial Catalog=FamilyTest; Integrated Security=True; MultipleActiveResultSets=True";
+
+            if (Environment.GetEnvironmentVariable("ConnectionString") != null)
+            {
+                // LocalDb isn't supported in Linux. To support running tests, allow env variable for conn string
+                connectionString = Environment.GetEnvironmentVariable("ConnectionString");
+            }
+
+            var builder = new SqlConnectionStringBuilder(connectionString) { InitialCatalog = initialCatalog };
+
+            return new DbContextOptionsBuilder<TContext>()
+                .UseSqlServer(builder.ConnectionString)
+                .Options;
+        }
+    }
+}
diff --git a/TrackableEntities.EF.Core.Tests/Helpers/FamilyDbContextFixture.cs b/TrackableEntities.EF.Core.Tests/Helpers/FamilyDbContextFixture.cs
@@ -25,9 +25,7 @@ public void Initialize(bool useInMemory = true, Action seedData = null)
             }
             else
             {
-                _options = new DbContextOptionsBuilder<FamilyDbContext>()
-                    .UseSqlServer(@"Data Source=(localdb)\MSSQLLocalDB; Initial Catelog=FamilyTest; Integrated Security=True; MultipleActiveResultSets=True")
-                    .Options;
+                _options = DbContextOptionsHelper.GetContextOptions<FamilyDbContext>("FamilyTest");
             }
             _context = new FamilyDbContext(_options);
             _context.Database.EnsureCreated();

diff --git a/TrackableEntities.EF.Core.Tests/Helpers/NorthwindDbContextFixture.cs b/TrackableEntities.EF.Core.Tests/Helpers/NorthwindDbContextFixture.cs
@@ -26,9 +26,7 @@ public void Initialize(bool useInMemory = true, Action seedData = null)
             }
             else
             {
-                _options = new DbContextOptionsBuilder<NorthwindDbContext>()
-                    .UseSqlServer(@"Data Source=(localdb)\MSSQLLocalDB; Initial Catalog=NorthwindTest; Integrated Security=True; MultipleActiveResultSets=True")
-                    .Options;
+                _options = DbContextOptionsHelper.GetContextOptions<NorthwindDbContext>("NorthwindTest");
             }
             _context = new NorthwindDbContext(_options);
             _context.Database.EnsureCreated(); // If login error, manually create NorthwindTest database