bitwarden/test/Core.Test/Models/Api/Request/PushSendRequestModelTests.cs

#nullable enable
using System.ComponentModel.DataAnnotations;
using System.Text.Json;
using Bit.Core.Enums;
using Bit.Core.Models.Api;
using Bit.Core.Utilities;
using Bit.Test.Common.AutoFixture.Attributes;
using Xunit;

namespace Bit.Core.Test.Models.Api.Request;

public class PushSendRequestModelTests
{
    [Fact]
    public void Validate_UserIdOrganizationIdInstallationIdNull_Invalid()
    {
        var model = new PushSendRequestModel<string>
        {
            UserId = null,
            OrganizationId = null,
            InstallationId = null,
            Type = PushType.SyncCiphers,
            Payload = "test"
        };

        var results = Validate(model);

        Assert.Single(results);
        Assert.Contains(results,
            result => result.ErrorMessage == "UserId or OrganizationId or InstallationId is required.");
    }

    [Fact]
    public void Validate_UserIdProvidedOrganizationIdInstallationIdNull_Valid()
    {
        var model = new PushSendRequestModel<string>
        {
            UserId = Guid.NewGuid(),
            OrganizationId = null,
            InstallationId = null,
            Type = PushType.SyncCiphers,
            Payload = "test"
        };

        var results = Validate(model);

        Assert.Empty(results);
    }

    [Fact]
    public void Validate_OrganizationIdProvidedUserIdInstallationIdNull_Valid()
    {
        var model = new PushSendRequestModel<string>
        {
            UserId = null,
            OrganizationId = Guid.NewGuid(),
            InstallationId = null,
            Type = PushType.SyncCiphers,
            Payload = "test"
        };

        var results = Validate(model);

        Assert.Empty(results);
    }

    [Fact]
    public void Validate_InstallationIdProvidedUserIdOrganizationIdNull_Valid()
    {
        var model = new PushSendRequestModel<string>
        {
            UserId = null,
            OrganizationId = null,
            InstallationId = Guid.NewGuid(),
            Type = PushType.SyncCiphers,
            Payload = "test"
        };

        var results = Validate(model);

        Assert.Empty(results);
    }

    [Theory]
    [BitAutoData("Payload")]
    [BitAutoData("Type")]
    public void Validate_RequiredFieldNotProvided_Invalid(string requiredField)
    {
        var model = new PushSendRequestModel<string>
        {
            UserId = Guid.NewGuid(),
            OrganizationId = Guid.NewGuid(),
            Type = PushType.SyncCiphers,
            Payload = "test"
        };

        var dictionary = new Dictionary<string, object?>();
        foreach (var property in model.GetType().GetProperties())
        {
            if (property.Name == requiredField)
            {
                continue;
            }

            dictionary[property.Name] = property.GetValue(model);
        }

        var serialized = JsonSerializer.Serialize(dictionary, JsonHelpers.IgnoreWritingNull);
        var jsonException =
            Assert.Throws<JsonException>(() => JsonSerializer.Deserialize<PushSendRequestModel<string>>(serialized));
        Assert.Contains($"missing required properties, including the following: {requiredField}",
            jsonException.Message);
    }

    [Fact]
    public void Validate_AllFieldsPresent_Valid()
    {
        var model = new PushSendRequestModel<string>
        {
            UserId = Guid.NewGuid(),
            OrganizationId = Guid.NewGuid(),
            Type = PushType.SyncCiphers,
            Payload = "test payload",
            Identifier = Guid.NewGuid().ToString(),
            ClientType = ClientType.All,
            DeviceId = Guid.NewGuid()
        };

        var results = Validate(model);

        Assert.Empty(results);
    }

    private static List<ValidationResult> Validate<T>(PushSendRequestModel<T> model)
    {
        var results = new List<ValidationResult>();
        Validator.TryValidateObject(model, new ValidationContext(model), results, true);
        return results;
    }
}