small fixes
This commit is contained in:
@@ -444,7 +444,20 @@ class Ec2ContainerManagerTest {
|
||||
|
||||
@Test
|
||||
void userDataExecutionCommandRunsScriptDirectlySoShebangIsHonored() {
|
||||
assertArrayEquals(new String[]{"/var/lib/user-data.sh"}, Ec2ContainerManager.userDataExecutionCommand());
|
||||
String[] command = Ec2ContainerManager.userDataExecutionCommand();
|
||||
|
||||
assertArrayEquals(new String[]{"sh", "-c"}, Arrays.copyOf(command, 2));
|
||||
assertTrue(command[2].contains("/var/lib/user-data.sh"));
|
||||
assertFalse(command[2].contains("sh /var/lib/user-data.sh"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void userDataExecutionCommandRestoresPolicyAndStartsPackageServices() {
|
||||
String command = Ec2ContainerManager.userDataExecutionCommand()[2];
|
||||
|
||||
assertTrue(command.contains("mv \"$backup\" \"$policy\""));
|
||||
assertTrue(command.contains("service \"$service\" \"$action\""));
|
||||
assertTrue(command.contains("exit \"$status\""));
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -1320,7 +1333,7 @@ class Ec2ContainerManagerTest {
|
||||
verify(copy).withRemotePath("/var/lib");
|
||||
verify(copy, never()).withRemotePath("/tmp");
|
||||
assertTrue(harness.executedCommands.stream()
|
||||
.anyMatch(command -> Arrays.equals(command, new String[]{"/var/lib/user-data.sh"})));
|
||||
.anyMatch(command -> Arrays.equals(command, Ec2ContainerManager.userDataExecutionCommand())));
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -1780,7 +1793,7 @@ class Ec2ContainerManagerTest {
|
||||
});
|
||||
when(execCreate.exec()).thenAnswer(invocation -> {
|
||||
String[] command = currentCommand.get();
|
||||
return command != null && command.length == 1 && "/var/lib/user-data.sh".equals(command[0])
|
||||
return Arrays.equals(command, Ec2ContainerManager.userDataExecutionCommand())
|
||||
? userDataExec : metadataExec;
|
||||
});
|
||||
|
||||
@@ -1969,7 +1982,7 @@ class Ec2ContainerManagerTest {
|
||||
});
|
||||
when(execCreate.exec()).thenAnswer(invocation -> {
|
||||
String[] command = currentCommand.get();
|
||||
if (command != null && command.length == 1 && "/var/lib/user-data.sh".equals(command[0])) {
|
||||
if (Arrays.equals(command, Ec2ContainerManager.userDataExecutionCommand())) {
|
||||
return userDataExec;
|
||||
}
|
||||
return metadataExec;
|
||||
|
||||
Reference in New Issue
Block a user