public async Task GetSurveyExtensionByTenantAndSlugName() { var mock = new Mock <IUDFAzureTable>(); var customEntity = new CustomEntity { Id = 1 }; mock.Setup(t => t.GetExtensionByPartitionRowKeyAsync(typeof(CustomEntity), "tenant", "slugname")).ReturnsAsync(customEntity); var store = new SurveyStore(default(IAzureTable <SurveyRow>), default(IAzureTable <QuestionRow>), mock.Object); var retrievedExtension = await store.GetSurveyExtensionByTenantAndSlugNameAsync("tenant", "slugname", typeof(CustomEntity)); Assert.AreEqual(customEntity.ToString(), retrievedExtension.ToString()); }