36 lines
1.2 KiB
C#
36 lines
1.2 KiB
C#
using ReverseLlama.Client;
|
|
using Xunit;
|
|
|
|
namespace ReverseLlama.Client.Tests;
|
|
|
|
public sealed class UpstreamRequestTests
|
|
{
|
|
private static readonly Uri Upstream = new("http://localhost:11434");
|
|
|
|
[Theory]
|
|
[InlineData("/api/tags", "http://localhost:11434/api/tags")]
|
|
[InlineData("/api/tags?model=llama3.1", "http://localhost:11434/api/tags?model=llama3.1")]
|
|
[InlineData("/api//tags", "http://localhost:11434/api//tags")]
|
|
public void BuildUpstreamUri_AcceptsOriginFormPaths(string pathAndQuery, string expected)
|
|
{
|
|
var uri = UpstreamRequest.BuildUpstreamUri(Upstream, pathAndQuery);
|
|
|
|
Assert.Equal(expected, uri.AbsoluteUri);
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData("//169.254.169.254/latest")]
|
|
[InlineData("http://169.254.169.254/latest")]
|
|
[InlineData("https://localhost:11434/api/tags")]
|
|
[InlineData(@"\\169.254.169.254\latest")]
|
|
[InlineData(@"/\169.254.169.254/latest")]
|
|
[InlineData("api/tags")]
|
|
public void BuildUpstreamUri_RejectsPathsThatCanEscapeTheUpstreamOrigin(string pathAndQuery)
|
|
{
|
|
var exception = Assert.Throws<InvalidOperationException>(
|
|
() => UpstreamRequest.BuildUpstreamUri(Upstream, pathAndQuery));
|
|
|
|
Assert.Contains("origin-form path", exception.Message);
|
|
}
|
|
}
|