|
16 | 16 | using Microsoft.Azure.WebJobs.Script.WebHost.Management.LinuxSpecialization; |
17 | 17 | using Microsoft.Azure.WebJobs.Script.WebHost.Models; |
18 | 18 | using Microsoft.Azure.WebJobs.Script.Workers.Rpc; |
19 | | -using Microsoft.Extensions.Configuration; |
20 | 19 | using Microsoft.Extensions.Logging; |
21 | 20 | using Microsoft.Extensions.Logging.Abstractions; |
22 | 21 | using Microsoft.WebJobs.Script.Tests; |
@@ -974,16 +973,19 @@ public async Task Does_Not_Mount_Invalid_BYOS_Accounts() |
974 | 973 |
|
975 | 974 | var meshInitServiceClient = new Mock<IMeshServiceClient>(MockBehavior.Strict); |
976 | 975 |
|
| 976 | + bool called = false; |
977 | 977 | meshInitServiceClient.Setup(client => |
978 | | - client.MountCifs(Utility.BuildStorageConnectionString(account1, accessKey1, CloudConstants.AzureStorageSuffix), share1, targetPath1)).Returns(Task.FromResult(true)); |
| 978 | + client.MountCifs(Utility.BuildStorageConnectionString(account1, accessKey1, CloudConstants.AzureStorageSuffix), share1, targetPath1)) |
| 979 | + .Returns(Task.FromResult(true)) |
| 980 | + .Callback(() => called = true); |
979 | 981 |
|
980 | 982 | var instanceManager = new AtlasInstanceManager(_optionsFactory, _httpClientFactory, _scriptWebEnvironment, _environment, |
981 | 983 | _loggerFactory.CreateLogger<AtlasInstanceManager>(), new TestMetricsLogger(), meshInitServiceClient.Object, |
982 | 984 | _runFromPackageHandler, _packageDownloadHandler.Object); |
983 | 985 |
|
984 | 986 | instanceManager.StartAssignment(hostAssignmentContext); |
985 | 987 |
|
986 | | - await Task.Delay(TimeSpan.FromSeconds(0.5)); |
| 988 | + await TestHelpers.Await(() => called); |
987 | 989 |
|
988 | 990 | meshInitServiceClient.Verify( |
989 | 991 | client => client.MountCifs(Utility.BuildStorageConnectionString(account1, accessKey1, CloudConstants.AzureStorageSuffix), share1, |
|
0 commit comments