Prototype Exec API (#271)
This commit is contained in:
Frederik Carlier
committed by
Kubernetes Prow Robot
Kubernetes Prow Robot
parent
644cf76009
commit
466e33995d
@@ -2,20 +2,23 @@
|
||||
* These tests are for the netcoreapp2.1 version of the client (there are separate tests for netstandard that don't actually connect to a server).
|
||||
*/
|
||||
|
||||
using k8s.Models;
|
||||
using Microsoft.Rest;
|
||||
using Microsoft.Rest.Serialization;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Diagnostics;
|
||||
using System.IO;
|
||||
using System.Net.Http;
|
||||
using System.Net.WebSockets;
|
||||
using System.Text;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.Rest;
|
||||
using Xunit;
|
||||
using Xunit.Abstractions;
|
||||
|
||||
namespace k8s.Tests
|
||||
{
|
||||
{
|
||||
/// <summary>
|
||||
/// Tests for <see cref="KubeApiClient"/>'s exec-in-pod functionality.
|
||||
/// </summary>
|
||||
@@ -92,5 +95,241 @@ namespace k8s.Tests
|
||||
WebSocketTestAdapter.CompleteTest();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
[Fact]
|
||||
public void GetExitCodeOrThrow_Success()
|
||||
{
|
||||
var status = new V1Status()
|
||||
{
|
||||
Metadata = null,
|
||||
Status = "Success",
|
||||
};
|
||||
|
||||
Assert.Equal(0, Kubernetes.GetExitCodeOrThrow(status));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void GetExitCodeOrThrow_NonZeroExitCode()
|
||||
{
|
||||
var status = new V1Status()
|
||||
{
|
||||
Metadata = null,
|
||||
Status = "Failure",
|
||||
Message = "command terminated with non-zero exit code: Error executing in Docker Container: 1",
|
||||
Reason = "NonZeroExitCode",
|
||||
Details = new V1StatusDetails()
|
||||
{
|
||||
Causes = new List<V1StatusCause>()
|
||||
{
|
||||
new V1StatusCause()
|
||||
{
|
||||
Reason = "ExitCode",
|
||||
Message = "1"
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
Assert.Equal(1, Kubernetes.GetExitCodeOrThrow(status));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void GetExitCodeOrThrow_InvalidExitCode()
|
||||
{
|
||||
var status = new V1Status()
|
||||
{
|
||||
Metadata = null,
|
||||
Status = "Failure",
|
||||
Message = "command terminated with non-zero exit code: Error executing in Docker Container: 1",
|
||||
Reason = "NonZeroExitCode",
|
||||
Details = new V1StatusDetails()
|
||||
{
|
||||
Causes = new List<V1StatusCause>()
|
||||
{
|
||||
new V1StatusCause()
|
||||
{
|
||||
Reason = "ExitCode",
|
||||
Message = "abc"
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
var ex = Assert.Throws<KubernetesException>(() => Kubernetes.GetExitCodeOrThrow(status));
|
||||
Assert.Equal(status, ex.Status);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void GetExitCodeOrThrow_NoExitCode()
|
||||
{
|
||||
var status = new V1Status()
|
||||
{
|
||||
Metadata = null,
|
||||
Status = "Failure",
|
||||
Message = "command terminated with non-zero exit code: Error executing in Docker Container: 1",
|
||||
Reason = "NonZeroExitCode",
|
||||
Details = new V1StatusDetails()
|
||||
{
|
||||
Causes = new List<V1StatusCause>()
|
||||
{
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
var ex = Assert.Throws<KubernetesException>(() => Kubernetes.GetExitCodeOrThrow(status));
|
||||
Assert.Equal(status, ex.Status);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void GetExitCodeOrThrow_OtherError()
|
||||
{
|
||||
var status = new V1Status()
|
||||
{
|
||||
Metadata = null,
|
||||
Status = "Failure",
|
||||
Reason = "SomethingElse"
|
||||
};
|
||||
|
||||
var ex = Assert.Throws<KubernetesException>(() => Kubernetes.GetExitCodeOrThrow(status));
|
||||
Assert.Equal(status, ex.Status);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task NamespacedPodExecAsync_ActionNull()
|
||||
{
|
||||
using (MemoryStream stdIn = new MemoryStream())
|
||||
using (MemoryStream stdOut = new MemoryStream())
|
||||
using (MemoryStream stdErr = new MemoryStream())
|
||||
using (MemoryStream errorStream = new MemoryStream())
|
||||
{
|
||||
var muxedStream = new Moq.Mock<IStreamDemuxer>();
|
||||
muxedStream.Setup(m => m.GetStream(null, ChannelIndex.StdIn)).Returns(stdIn);
|
||||
muxedStream.Setup(m => m.GetStream(ChannelIndex.StdOut, null)).Returns(stdOut);
|
||||
muxedStream.Setup(m => m.GetStream(ChannelIndex.StdErr, null)).Returns(stdErr);
|
||||
muxedStream.Setup(m => m.GetStream(ChannelIndex.Error, null)).Returns(errorStream);
|
||||
|
||||
var kubernetesMock = new Moq.Mock<Kubernetes>(
|
||||
new object[] { Moq.Mock.Of<ServiceClientCredentials>(), new DelegatingHandler[] { } });
|
||||
var command = new string[] { "/bin/bash", "-c", "echo Hello, World!" };
|
||||
|
||||
kubernetesMock.Setup(m => m.MuxedStreamNamespacedPodExecAsync("pod-name", "pod-namespace", command, "my-container", true, true, true, false, WebSocketProtocol.V4BinaryWebsocketProtocol, null, CancellationToken.None))
|
||||
.Returns(Task.FromResult(muxedStream.Object));
|
||||
|
||||
using (Kubernetes client = kubernetesMock.Object)
|
||||
{
|
||||
await Assert.ThrowsAsync<ArgumentNullException>(() => client.NamespacedPodExecAsync("pod-name", "pod-namespace", "my-container", command, false, null, CancellationToken.None)).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task NamespacedPodExecAsync_HttpException_WithStatus()
|
||||
{
|
||||
var kubernetesMock = new Moq.Mock<Kubernetes>(
|
||||
new object[] { Moq.Mock.Of<ServiceClientCredentials>(), new DelegatingHandler[] { } });
|
||||
var command = new string[] { "/bin/bash", "-c", "echo Hello, World!" };
|
||||
var handler = new ExecAsyncCallback((stdIn, stdOut, stdError) => Task.CompletedTask);
|
||||
|
||||
var status = new V1Status();
|
||||
kubernetesMock.Setup(m => m.MuxedStreamNamespacedPodExecAsync("pod-name", "pod-namespace", command, "my-container", true, true, true, false, WebSocketProtocol.V4BinaryWebsocketProtocol, null, CancellationToken.None))
|
||||
.Throws(new HttpOperationException() { Body = status });
|
||||
|
||||
using (Kubernetes client = kubernetesMock.Object)
|
||||
{
|
||||
var ex = await Assert.ThrowsAsync<KubernetesException>(() => client.NamespacedPodExecAsync("pod-name", "pod-namespace", "my-container", command, false, handler, CancellationToken.None)).ConfigureAwait(false);
|
||||
Assert.Same(status, ex.Status);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task NamespacedPodExecAsync_HttpException_NoStatus()
|
||||
{
|
||||
var kubernetesMock = new Moq.Mock<Kubernetes>(
|
||||
new object[] { Moq.Mock.Of<ServiceClientCredentials>(), new DelegatingHandler[] { } });
|
||||
var command = new string[] { "/bin/bash", "-c", "echo Hello, World!" };
|
||||
var handler = new ExecAsyncCallback((stdIn, stdOut, stdError) => Task.CompletedTask);
|
||||
|
||||
var exception = new HttpOperationException();
|
||||
kubernetesMock.Setup(m => m.MuxedStreamNamespacedPodExecAsync("pod-name", "pod-namespace", command, "my-container", true, true, true, false, WebSocketProtocol.V4BinaryWebsocketProtocol, null, CancellationToken.None))
|
||||
.Throws(exception);
|
||||
|
||||
using (Kubernetes client = kubernetesMock.Object)
|
||||
{
|
||||
var ex = await Assert.ThrowsAsync<HttpOperationException>(() => client.NamespacedPodExecAsync("pod-name", "pod-namespace", "my-container", command, false, handler, CancellationToken.None)).ConfigureAwait(false);
|
||||
Assert.Same(exception, ex);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task NamespacedPodExecAsync_GenericException()
|
||||
{
|
||||
var kubernetesMock = new Moq.Mock<Kubernetes>(
|
||||
new object[] { Moq.Mock.Of<ServiceClientCredentials>(), new DelegatingHandler[] { } });
|
||||
var command = new string[] { "/bin/bash", "-c", "echo Hello, World!" };
|
||||
var handler = new ExecAsyncCallback((stdIn, stdOut, stdError) => Task.CompletedTask);
|
||||
|
||||
var exception = new Exception();
|
||||
kubernetesMock.Setup(m => m.MuxedStreamNamespacedPodExecAsync("pod-name", "pod-namespace", command, "my-container", true, true, true, false, WebSocketProtocol.V4BinaryWebsocketProtocol, null, CancellationToken.None))
|
||||
.Throws(exception);
|
||||
|
||||
using (Kubernetes client = kubernetesMock.Object)
|
||||
{
|
||||
var ex = await Assert.ThrowsAsync<Exception>(() => client.NamespacedPodExecAsync("pod-name", "pod-namespace", "my-container", command, false, handler, CancellationToken.None)).ConfigureAwait(false);
|
||||
Assert.Same(exception, ex);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task NamespacedPodExecAsync_ExitCode_NonZero()
|
||||
{
|
||||
var processStatus = new V1Status()
|
||||
{
|
||||
Metadata = null,
|
||||
Status = "Failure",
|
||||
Message = "command terminated with non-zero exit code: Error executing in Docker Container: 1",
|
||||
Reason = "NonZeroExitCode",
|
||||
Details = new V1StatusDetails()
|
||||
{
|
||||
Causes = new List<V1StatusCause>()
|
||||
{
|
||||
new V1StatusCause()
|
||||
{
|
||||
Reason = "ExitCode",
|
||||
Message = "1"
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
var processStatusJson = Encoding.UTF8.GetBytes(SafeJsonConvert.SerializeObject(processStatus));
|
||||
var handler = new ExecAsyncCallback((stdIn, stdOut, stdError) => Task.CompletedTask);
|
||||
|
||||
using (MemoryStream stdIn = new MemoryStream())
|
||||
using (MemoryStream stdOut = new MemoryStream())
|
||||
using (MemoryStream stdErr = new MemoryStream())
|
||||
using (MemoryStream errorStream = new MemoryStream(processStatusJson))
|
||||
{
|
||||
var muxedStream = new Moq.Mock<IStreamDemuxer>();
|
||||
muxedStream.Setup(m => m.GetStream(null, ChannelIndex.StdIn)).Returns(stdIn);
|
||||
muxedStream.Setup(m => m.GetStream(ChannelIndex.StdOut, null)).Returns(stdOut);
|
||||
muxedStream.Setup(m => m.GetStream(ChannelIndex.StdErr, null)).Returns(stdErr);
|
||||
muxedStream.Setup(m => m.GetStream(ChannelIndex.Error, null)).Returns(errorStream);
|
||||
|
||||
var kubernetesMock = new Moq.Mock<Kubernetes>(
|
||||
new object[] { Moq.Mock.Of<ServiceClientCredentials>(), new DelegatingHandler[] { } });
|
||||
var command = new string[] { "/bin/bash", "-c", "echo Hello, World!" };
|
||||
|
||||
var exception = new Exception();
|
||||
kubernetesMock.Setup(m => m.MuxedStreamNamespacedPodExecAsync("pod-name", "pod-namespace", command, "my-container", true, true, true, false, WebSocketProtocol.V4BinaryWebsocketProtocol, null, CancellationToken.None))
|
||||
.Returns(Task.FromResult(muxedStream.Object));
|
||||
|
||||
using (Kubernetes client = kubernetesMock.Object)
|
||||
{
|
||||
var exitCode = await client.NamespacedPodExecAsync("pod-name", "pod-namespace", "my-container", command, false, handler, CancellationToken.None).ConfigureAwait(false);
|
||||
Assert.Equal(1, exitCode);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
<PropertyGroup>
|
||||
<IsPackable>false</IsPackable>
|
||||
<RootNamespace>k8s.tests</RootNamespace>
|
||||
@@ -29,6 +29,7 @@
|
||||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.0.1" />
|
||||
<PackageReference Include="xunit" Version="2.4.1" />
|
||||
<PackageReference Include="xunit.runner.visualstudio" Version="2.4.1" NoWarn="NU1701" />
|
||||
<PackageReference Include="Moq" Version="4.10.1" />
|
||||
|
||||
<DotNetCliToolReference Include="dotnet-xunit" Version="2.3.1" />
|
||||
</ItemGroup>
|
||||
|
||||
Reference in New Issue
Block a user