diff --git a/CloudStub.Tests/BaseInMemoryCloudTableTests.cs b/CloudStub.Tests/BaseStubCloudTableTests.cs
similarity index 86%
rename from CloudStub.Tests/BaseInMemoryCloudTableTests.cs
rename to CloudStub.Tests/BaseStubCloudTableTests.cs
index 3e7fe4b..7386b98 100644
--- a/CloudStub.Tests/BaseInMemoryCloudTableTests.cs
+++ b/CloudStub.Tests/BaseStubCloudTableTests.cs
@@ -1,19 +1,21 @@
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
+using CloudStub.Core;
+using CloudStub.Core.StorageHandlers;
using Microsoft.Azure.Cosmos.Table;
namespace CloudStub.Tests
{
- public abstract class BaseInMemoryCloudTableTests : AzureStorageUnitTest
+ public abstract class BaseStubCloudTableTests : AzureStorageUnitTest
{
/// A temporary flag to easily switch between in-memory cloud table and actual Azure Storage Table.
protected static bool UseInMemory { get; } = true;
- protected BaseInMemoryCloudTableTests()
+ protected BaseStubCloudTableTests()
=> CloudTable = GetCloudTable(TestTableName);
- protected new string TestTableName { get; } = (nameof(BaseInMemoryCloudTableTests) + "TestTable" + Guid.NewGuid().ToString().Replace("-", "")).Substring(0, 63);
+ protected new string TestTableName { get; } = (nameof(BaseStubCloudTableTests) + "TestTable" + Guid.NewGuid().ToString().Replace("-", "")).Substring(0, 63);
protected CloudTable CloudTable { get; }
@@ -57,7 +59,7 @@ protected static async Task> GetAllEntitiesAsy
protected static CloudTable GetCloudTable(string tableName)
=> UseInMemory ?
- new InMemoryCloudTable(tableName) :
+ new StubCloudTable(new StubTable(tableName, new InMemoryTableStorageHandler())) :
CloudStorageAccount
.Parse(AzureStorageConnectionString)
.CreateCloudTableClient()
diff --git a/CloudStub.Tests/InMemoryCloudTableQueryComparisonTestData.cs b/CloudStub.Tests/StubCloudTableQueryComparisonTestData.cs
similarity index 99%
rename from CloudStub.Tests/InMemoryCloudTableQueryComparisonTestData.cs
rename to CloudStub.Tests/StubCloudTableQueryComparisonTestData.cs
index 69c4166..776253b 100644
--- a/CloudStub.Tests/InMemoryCloudTableQueryComparisonTestData.cs
+++ b/CloudStub.Tests/StubCloudTableQueryComparisonTestData.cs
@@ -6,7 +6,7 @@
namespace CloudStub.Tests
{
- public class InMemoryCloudTableQueryComparisonTestData : IEnumerable