speckle-sharp-sdk/tests/Speckle.Sdk.Serialization.Tests/ServerObjectManagerTests.cs

using System.Diagnostics.CodeAnalysis;
using Moq;
using RichardSzalay.MockHttp;
using Speckle.Newtonsoft.Json;
using Speckle.Newtonsoft.Json.Linq;
using Speckle.Sdk.Helpers;
using Speckle.Sdk.Logging;
using Speckle.Sdk.Serialisation.V2;

namespace Speckle.Sdk.Serialization.Tests;

[ExcludeFromCodeCoverage]
public abstract class MoqTest : IDisposable
{
  protected MoqTest() => Repository = new(MockBehavior.Strict);

  public void Dispose() => Repository.VerifyAll();

  protected MockRepository Repository { get; private set; } = new(MockBehavior.Strict);

  protected Mock<T> Create<T>(MockBehavior behavior = MockBehavior.Strict)
    where T : class => Repository.Create<T>(behavior);
}

public class ServerObjectManagerTests : MoqTest
{
  [Fact]
  public async Task DownloadObjects()
  {
    var id = Guid.Parse("6f422a35-6183-48b9-8021-d22ec97e8674").ToString();
    var id2 = Guid.Parse("ef2f7ea0-495a-46af-a9ad-f18a8a298597").ToString();
    var ct = new CancellationToken();
    var token = "token";
    var timeout = 2;
    var uri = new Uri("http://localhost");
    var streamId = "streamId";
    var jObject = new JObject();
    jObject.Add("id", id);
    jObject.Add("value", true);
    var jObject2 = new JObject();
    jObject2.Add("id", id2);
    jObject2.Add("value", true);
    var mockHttp = new MockHttpMessageHandler();
    Dictionary<string, string> postParameters = new()
    {
      { "objects", JsonConvert.SerializeObject(new List<string> { id, id2 }) },
    };

    string serializedPayload = JsonConvert.SerializeObject(postParameters);
    mockHttp
      .When(HttpMethod.Post, $"http://localhost/api/getobjects/{streamId}")
      .WithContent(serializedPayload)
      .Respond(
        "application/json",
        $"{id}\t{jObject.ToString(Formatting.None)}\n{id2}\t{jObject2.ToString(Formatting.None)}\n"
      );
    var httpClient = mockHttp.ToHttpClient();
    var http = Create<ISpeckleHttp>();
    http.Setup(x => x.CreateHttpClient(It.IsAny<HttpClientHandler>(), timeout, token)).Returns(httpClient);

    var activityFactory = Create<ISdkActivityFactory>();
    activityFactory.Setup(x => x.Start(null, "DownloadObjects")).Returns((ISdkActivity?)null);

    var serverObjectManager = new ServerObjectManager(
      http.Object,
      activityFactory.Object,
      uri,
      streamId,
      token,
      timeout
    );
    var results = serverObjectManager.DownloadObjects(new List<string> { id, id2 }, null, ct);
    var objects = new JObject();
    await foreach (var (x, json) in results)
    {
      objects.Add(x, JToken.Parse(json));
    }

    await VerifyJson(objects.ToString(Formatting.Indented));
  }

  [Fact]
  public async Task DownloadSingleObject()
  {
    var id = Guid.Parse("6f422a35-6183-48b9-8021-d22ec97e8674").ToString();
    var ct = new CancellationToken();
    var token = "token";
    var timeout = 2;
    var uri = new Uri("http://localhost");
    var streamId = "streamId";
    var jObject = new JObject();
    jObject.Add("id", id);
    jObject.Add("value", true);
    var mockHttp = new MockHttpMessageHandler();
    mockHttp
      .When(HttpMethod.Get, $"http://localhost/objects/{streamId}/{id}/single")
      .Respond("application/json", $"{jObject.ToString(Formatting.None)}\n");
    var httpClient = mockHttp.ToHttpClient();
    var http = Create<ISpeckleHttp>();
    http.Setup(x => x.CreateHttpClient(It.IsAny<HttpClientHandler>(), timeout, token)).Returns(httpClient);

    var activityFactory = Create<ISdkActivityFactory>();
    activityFactory.Setup(x => x.Start(null, "DownloadSingleObject")).Returns((ISdkActivity?)null);

    var serverObjectManager = new ServerObjectManager(
      http.Object,
      activityFactory.Object,
      uri,
      streamId,
      token,
      timeout
    );
    var json = await serverObjectManager.DownloadSingleObject(id, null, ct);
    await VerifyJson(json);
  }
}