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());
}
