break; case TCP: probe.tcpSocket = new TCPSocketAction(); probe.tcpSocket.port = String.valueOf(healthConfig.port); break;
private Container createPodContainer() { Container podContainer = new Container(); podContainer.name = CONTAINER_NAME; podContainer.image = CONTAINER_NAME; podContainer.command = Arrays.asList("app.sh"); podContainer.args = Arrays.asList("start"); podContainer.workingDir = CONTAINER_WORKING_DIR; ContainerPort port = new ContainerPort(); port.containerPort = 80; port.protocol = "TCP"; port.hostPort = 8080; podContainer.ports = Arrays.asList(port); EnvVar env = new EnvVar(); env.name = "name"; env.value = "value"; podContainer.env = Arrays.asList(env); SecurityContext ctx = new SecurityContext(); ctx.privileged = Boolean.TRUE; podContainer.securityContext = ctx; Probe probe = new Probe(); probe.tcpSocket = new TCPSocketAction(); probe.timeoutSeconds = 60; probe.failureThreshold = 3; podContainer.livenessProbe = probe; return podContainer; }
@Test public void testConvertPodContainerProbeToHealthConfigTCP() { Container podContainer = new Container(); podContainer.livenessProbe = new Probe(); podContainer.livenessProbe.tcpSocket = new TCPSocketAction(); podContainer.livenessProbe.tcpSocket.port = "8080"; podContainer.livenessProbe.timeoutSeconds = 3; HealthConfig expectedHealthConfig = new HealthConfig(); expectedHealthConfig.protocol = RequestProtocol.TCP; expectedHealthConfig.port = 8080; expectedHealthConfig.timeoutMillis = 3000; HealthConfig actualHealthConfig = fromPodContainerProbeToContainerDescriptionHealthConfig( podContainer); assertNotNull(actualHealthConfig); assertEquals(actualHealthConfig.protocol, expectedHealthConfig.protocol); assertEquals(actualHealthConfig.port, expectedHealthConfig.port); assertEquals(actualHealthConfig.timeoutMillis, expectedHealthConfig.timeoutMillis); }
expectedProbe3.tcpSocket = new TCPSocketAction(); expectedProbe3.tcpSocket.port = "32000";