|
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;
|
@@ -967,16 +966,19 @@ public async Task Does_Not_Mount_Invalid_BYOS_Accounts()
|
967 | 966 |
|
968 | 967 | var meshInitServiceClient = new Mock<IMeshServiceClient>(MockBehavior.Strict);
|
969 | 968 |
|
| 969 | + bool called = false; |
970 | 970 | meshInitServiceClient.Setup(client =>
|
971 |
| - client.MountCifs(Utility.BuildStorageConnectionString(account1, accessKey1, CloudConstants.AzureStorageSuffix), share1, targetPath1)).Returns(Task.FromResult(true)); |
| 971 | + client.MountCifs(Utility.BuildStorageConnectionString(account1, accessKey1, CloudConstants.AzureStorageSuffix), share1, targetPath1)) |
| 972 | + .Returns(Task.FromResult(true)) |
| 973 | + .Callback(() => called = true); |
972 | 974 |
|
973 | 975 | var instanceManager = new AtlasInstanceManager(_optionsFactory, _httpClientFactory, _scriptWebEnvironment, _environment,
|
974 | 976 | _loggerFactory.CreateLogger<AtlasInstanceManager>(), new TestMetricsLogger(), meshInitServiceClient.Object,
|
975 | 977 | _runFromPackageHandler, _packageDownloadHandler.Object);
|
976 | 978 |
|
977 | 979 | instanceManager.StartAssignment(hostAssignmentContext);
|
978 | 980 |
|
979 |
| - await Task.Delay(TimeSpan.FromSeconds(0.5)); |
| 981 | + await TestHelpers.Await(() => called); |
980 | 982 |
|
981 | 983 | meshInitServiceClient.Verify(
|
982 | 984 | client => client.MountCifs(Utility.BuildStorageConnectionString(account1, accessKey1, CloudConstants.AzureStorageSuffix), share1,
|
|
0 commit comments