@@ -444,7 +444,7 @@ class FunctionPullingContainerProxyTests
444
444
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, transid)
445
445
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
446
446
client.expectMsg(StartClient )
447
- client.send(machine, ClientCreationCompleted ())
447
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
448
448
449
449
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
450
450
expectInitialized(probe)
@@ -509,7 +509,7 @@ class FunctionPullingContainerProxyTests
509
509
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
510
510
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
511
511
client.expectMsg(StartClient )
512
- client.send(machine, ClientCreationCompleted ())
512
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
513
513
514
514
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
515
515
expectInitialized(probe)
@@ -573,7 +573,7 @@ class FunctionPullingContainerProxyTests
573
573
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
574
574
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
575
575
client.expectMsg(StartClient )
576
- client.send(machine, ClientCreationCompleted ())
576
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
577
577
578
578
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
579
579
expectInitialized(probe)
@@ -642,7 +642,7 @@ class FunctionPullingContainerProxyTests
642
642
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
643
643
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
644
644
client.expectMsg(StartClient )
645
- client.send(machine, ClientCreationCompleted ())
645
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
646
646
647
647
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
648
648
expectInitialized(probe)
@@ -883,7 +883,7 @@ class FunctionPullingContainerProxyTests
883
883
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
884
884
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
885
885
client.expectMsg(StartClient )
886
- client.send(machine, ClientCreationCompleted ())
886
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
887
887
888
888
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
889
889
expectInitialized(probe)
@@ -943,7 +943,7 @@ class FunctionPullingContainerProxyTests
943
943
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
944
944
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
945
945
client.expectMsg(StartClient )
946
- client.send(machine, ClientCreationCompleted ())
946
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
947
947
948
948
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
949
949
expectInitialized(probe)
@@ -1015,7 +1015,7 @@ class FunctionPullingContainerProxyTests
1015
1015
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1016
1016
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1017
1017
client.expectMsg(StartClient )
1018
- client.send(machine, ClientCreationCompleted ())
1018
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1019
1019
1020
1020
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1021
1021
expectInitialized(probe)
@@ -1094,7 +1094,7 @@ class FunctionPullingContainerProxyTests
1094
1094
1095
1095
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1096
1096
client.expectMsg(StartClient )
1097
- client.send(machine, ClientCreationCompleted ())
1097
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1098
1098
1099
1099
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1100
1100
expectInitialized(probe)
@@ -1180,7 +1180,7 @@ class FunctionPullingContainerProxyTests
1180
1180
1181
1181
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1182
1182
client.expectMsg(StartClient )
1183
- client.send(machine, ClientCreationCompleted ())
1183
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1184
1184
1185
1185
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1186
1186
expectInitialized(probe)
@@ -1265,7 +1265,7 @@ class FunctionPullingContainerProxyTests
1265
1265
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1266
1266
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1267
1267
client.expectMsg(StartClient )
1268
- client.send(machine, ClientCreationCompleted ())
1268
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1269
1269
1270
1270
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1271
1271
expectInitialized(probe)
@@ -1342,7 +1342,7 @@ class FunctionPullingContainerProxyTests
1342
1342
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1343
1343
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1344
1344
client.expectMsg(StartClient )
1345
- client.send(machine, ClientCreationCompleted ())
1345
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1346
1346
1347
1347
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1348
1348
expectInitialized(probe)
@@ -1423,7 +1423,7 @@ class FunctionPullingContainerProxyTests
1423
1423
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1424
1424
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1425
1425
client.expectMsg(StartClient )
1426
- client.send(machine, ClientCreationCompleted ())
1426
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1427
1427
1428
1428
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1429
1429
expectInitialized(probe)
@@ -1500,7 +1500,7 @@ class FunctionPullingContainerProxyTests
1500
1500
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1501
1501
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1502
1502
client.expectMsg(StartClient )
1503
- client.send(machine, ClientCreationCompleted ())
1503
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1504
1504
1505
1505
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1506
1506
expectInitialized(probe)
@@ -1576,7 +1576,11 @@ class FunctionPullingContainerProxyTests
1576
1576
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1577
1577
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1578
1578
client.expectMsg(StartClient )
1579
- client.send(machine, ClientCreationCompleted ())
1579
+ client.send(machine, ClientCreationCompleted (Some (client.ref)))
1580
+
1581
+ awaitAssert {
1582
+ machine.underlyingActor.stateData.getContainer should not be None
1583
+ }
1580
1584
1581
1585
val containerId = machine.underlyingActor.stateData.getContainer match {
1582
1586
case Some (container) => container.containerId
@@ -1666,7 +1670,7 @@ class FunctionPullingContainerProxyTests
1666
1670
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1667
1671
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1668
1672
client.expectMsg(StartClient )
1669
- client.send(machine, ClientCreationCompleted ())
1673
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1670
1674
dataManagementService.expectMsg(
1671
1675
RegisterData (
1672
1676
ContainerKeys .existingContainers(
@@ -1775,7 +1779,7 @@ class FunctionPullingContainerProxyTests
1775
1779
machine ! Initialize (invocationNamespace.asString, fqn, noLogsAction, schedulerHost, rpcPort, messageTransId)
1776
1780
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1777
1781
client.expectMsg(StartClient )
1778
- client.send(machine, ClientCreationCompleted ())
1782
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1779
1783
1780
1784
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1781
1785
expectInitialized(probe)
@@ -1906,7 +1910,7 @@ class FunctionPullingContainerProxyTests
1906
1910
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1907
1911
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
1908
1912
client.expectMsg(StartClient )
1909
- client.send(machine, ClientCreationCompleted ())
1913
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1910
1914
1911
1915
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1912
1916
expectInitialized(probe)
@@ -1980,7 +1984,7 @@ class FunctionPullingContainerProxyTests
1980
1984
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
1981
1985
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
1982
1986
client.expectMsg(StartClient )
1983
- client.send(machine, ClientCreationCompleted ())
1987
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
1984
1988
1985
1989
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
1986
1990
expectInitialized(probe)
@@ -2057,7 +2061,7 @@ class FunctionPullingContainerProxyTests
2057
2061
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2058
2062
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2059
2063
client.expectMsg(StartClient )
2060
- client.send(machine, ClientCreationCompleted ())
2064
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2061
2065
2062
2066
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2063
2067
expectInitialized(probe)
@@ -2136,7 +2140,7 @@ class FunctionPullingContainerProxyTests
2136
2140
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2137
2141
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2138
2142
client.expectMsg(StartClient )
2139
- client.send(machine, ClientCreationCompleted ())
2143
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2140
2144
2141
2145
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2142
2146
expectInitialized(probe)
@@ -2226,7 +2230,7 @@ class FunctionPullingContainerProxyTests
2226
2230
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2227
2231
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2228
2232
client.expectMsg(StartClient )
2229
- client.send(machine, ClientCreationCompleted ())
2233
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2230
2234
2231
2235
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2232
2236
expectInitialized(probe)
@@ -2289,7 +2293,7 @@ class FunctionPullingContainerProxyTests
2289
2293
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2290
2294
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2291
2295
client.expectMsg(StartClient )
2292
- client.send(machine, ClientCreationCompleted ())
2296
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2293
2297
2294
2298
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2295
2299
expectInitialized(probe)
@@ -2355,7 +2359,7 @@ class FunctionPullingContainerProxyTests
2355
2359
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2356
2360
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2357
2361
client.expectMsg(StartClient )
2358
- client.send(machine, ClientCreationCompleted ())
2362
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2359
2363
2360
2364
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2361
2365
expectInitialized(probe)
@@ -2425,7 +2429,7 @@ class FunctionPullingContainerProxyTests
2425
2429
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2426
2430
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2427
2431
client.expectMsg(StartClient )
2428
- client.send(machine, ClientCreationCompleted ())
2432
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2429
2433
2430
2434
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2431
2435
expectInitialized(probe)
@@ -2508,7 +2512,7 @@ class FunctionPullingContainerProxyTests
2508
2512
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2509
2513
probe.expectMsg(Transition (machine, Uninitialized , CreatingClient ))
2510
2514
client.expectMsg(StartClient )
2511
- client.send(machine, ClientCreationCompleted ())
2515
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2512
2516
2513
2517
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2514
2518
expectInitialized(probe)
@@ -2585,7 +2589,7 @@ class FunctionPullingContainerProxyTests
2585
2589
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2586
2590
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
2587
2591
client.expectMsg(StartClient )
2588
- client.send(machine, ClientCreationCompleted ())
2592
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2589
2593
2590
2594
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2591
2595
expectInitialized(probe)
@@ -2652,7 +2656,7 @@ class FunctionPullingContainerProxyTests
2652
2656
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2653
2657
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
2654
2658
client.expectMsg(StartClient )
2655
- client.send(machine, ClientCreationCompleted ())
2659
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2656
2660
2657
2661
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2658
2662
expectInitialized(probe)
@@ -2738,7 +2742,7 @@ class FunctionPullingContainerProxyTests
2738
2742
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2739
2743
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
2740
2744
client.expectMsg(StartClient )
2741
- client.send(machine, ClientCreationCompleted ())
2745
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2742
2746
2743
2747
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2744
2748
expectInitialized(probe)
@@ -2818,7 +2822,7 @@ class FunctionPullingContainerProxyTests
2818
2822
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, transid)
2819
2823
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
2820
2824
client.expectMsg(StartClient )
2821
- client.send(machine, ClientCreationCompleted ())
2825
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2822
2826
2823
2827
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2824
2828
expectInitialized(probe)
@@ -2889,7 +2893,7 @@ class FunctionPullingContainerProxyTests
2889
2893
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2890
2894
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
2891
2895
client.expectMsg(StartClient )
2892
- client.send(machine, ClientCreationCompleted ())
2896
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2893
2897
2894
2898
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2895
2899
expectInitialized(probe)
@@ -2951,7 +2955,7 @@ class FunctionPullingContainerProxyTests
2951
2955
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
2952
2956
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
2953
2957
client.expectMsg(StartClient )
2954
- client.send(machine, ClientCreationCompleted ())
2958
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
2955
2959
2956
2960
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
2957
2961
expectInitialized(probe)
@@ -3017,7 +3021,7 @@ class FunctionPullingContainerProxyTests
3017
3021
machine ! Initialize (invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
3018
3022
probe.expectMsg(Transition (machine, ContainerCreated , CreatingClient ))
3019
3023
client.expectMsg(StartClient )
3020
- client.send(machine, ClientCreationCompleted ())
3024
+ client.send(machine, ClientCreationCompleted (Some (client.ref) ))
3021
3025
3022
3026
probe.expectMsg(Transition (machine, CreatingClient , ClientCreated ))
3023
3027
expectInitialized(probe)
0 commit comments