public async Task GetSurveyExtensionsByTenant() { var mock = new Mock <IUDFAzureTable>(); var customEntities = new[] { new CustomEntity { Id = 1 }, new CustomEntity { Id = 2 } }; mock.Setup(t => t.GetExtensionsByPartitionKeyAsync(typeof(CustomEntity), "tenant")).ReturnsAsync(customEntities); var store = new SurveyStore(default(IAzureTable <SurveyRow>), default(IAzureTable <QuestionRow>), mock.Object); var retrievedExtensions = await store.GetSurveyExtensionsByTenantAsync("tenant", typeof(CustomEntity)); Assert.AreEqual(2, retrievedExtensions.Count()); Assert.AreEqual(customEntities[0].ToString(), retrievedExtensions.ToList()[0].ToString()); Assert.AreEqual(customEntities[1].ToString(), retrievedExtensions.ToList()[1].ToString()); }