squidex/tests/Squidex.Domain.Apps.Core.Tests/Operations/ExtractReferenceIds/ReferenceExtractionTests.cs

// ==========================================================================
//  ReferenceExtractionTests.cs
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex Group
//  All rights reserved.
// ==========================================================================

using System;
using System.Collections.Generic;
using System.Linq;
using Newtonsoft.Json.Linq;
using Squidex.Domain.Apps.Core.Apps;
using Squidex.Domain.Apps.Core.Contents;
using Squidex.Domain.Apps.Core.ExtractReferenceIds;
using Squidex.Domain.Apps.Core.Schemas;
using Squidex.Infrastructure;
using Xunit;

#pragma warning disable xUnit2013 // Do not use equality check to check for collection size.

namespace Squidex.Domain.Apps.Core.Operations.ExtractReferenceIds
{
    public class ReferenceExtractionTests
    {
        private readonly Guid schemaId = Guid.NewGuid();
        private readonly Schema schema;
        private readonly LanguagesConfig languagesConfig = LanguagesConfig.Build(Language.EN, Language.DE);

        public ReferenceExtractionTests()
        {
            schema =
                new Schema("my-schema")
                    .AddField(new NumberField(1, "field1", Partitioning.Language))
                    .AddField(new NumberField(2, "field2", Partitioning.Invariant))
                    .AddField(new NumberField(3, "field3", Partitioning.Invariant))
                    .AddField(new AssetsField(5, "assets1", Partitioning.Invariant))
                    .AddField(new AssetsField(6, "assets2", Partitioning.Invariant))
                    .AddField(new JsonField(4, "json", Partitioning.Language))
                    .HideField(3);
        }

        [Fact]
        public void Should_remove_ids()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var input =
                new IdContentData()
                    .AddField(5,
                        new ContentFieldData()
                            .AddValue("iv", new JArray(id1.ToString(), id2.ToString())));

            var ids = input.GetReferencedIds(schema).ToArray();

            Assert.Equal(new[] { id1, id2 }, ids);
        }

        [Fact]
        public void Should_cleanup_deleted_ids()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var input =
                new IdContentData()
                    .AddField(5,
                        new ContentFieldData()
                            .AddValue("iv", new JArray(id1.ToString(), id2.ToString())));

            var actual = input.ToCleanedReferences(schema, new HashSet<Guid>(new[] { id2 }));

            var cleanedValue = (JArray)actual[5]["iv"];

            Assert.Equal(1, cleanedValue.Count);
            Assert.Equal(id1.ToString(), cleanedValue[0]);
        }

        [Fact]
        public void Should_return_ids_from_assets_field()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new AssetsField(1, "my-asset", Partitioning.Invariant);

            var result = sut.ExtractReferences(CreateValue(id1, id2)).ToArray();

            Assert.Equal(new[] { id1, id2 }, result);
        }

        [Fact]
        public void Should_empty_list_from_assets_field_for_referenced_ids_when_null()
        {
            var sut = new AssetsField(1, "my-asset", Partitioning.Invariant);

            var result = sut.ExtractReferences(null).ToArray();

            Assert.Empty(result);
        }

        [Fact]
        public void Should_empty_list_from_assets_field_for_referenced_ids_when_other_type()
        {
            var sut = new AssetsField(1, "my-asset", Partitioning.Invariant);

            var result = sut.ExtractReferences("invalid").ToArray();

            Assert.Empty(result);
        }

        [Fact]
        public void Should_return_null_from_assets_field_when_removing_references_from_null_array()
        {
            var sut = new AssetsField(1, "my-asset", Partitioning.Invariant);

            var result = sut.CleanReferences(null, null);

            Assert.Null(result);
        }

        [Fact]
        public void Should_remove_deleted_references_from_assets_field()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new AssetsField(1, "my-asset", Partitioning.Invariant);

            var result = sut.CleanReferences(CreateValue(id1, id2), new HashSet<Guid>(new[] { id2 }));

            Assert.Equal(CreateValue(id1), result);
        }

        [Fact]
        public void Should_return_same_token_from_assets_field_when_removing_references_and_nothing_to_remove()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new AssetsField(1, "my-asset", Partitioning.Invariant);

            var token = CreateValue(id1, id2);
            var result = sut.CleanReferences(token, new HashSet<Guid>(new[] { Guid.NewGuid() }));

            Assert.Same(token, result);
        }

        [Fact]
        public void Should_return_ids_from_references_field()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant,
                new ReferencesFieldProperties { SchemaId = schemaId });

            var result = sut.ExtractReferences(CreateValue(id1, id2)).ToArray();

            Assert.Equal(new[] { id1, id2, schemaId }, result);
        }

        [Fact]
        public void Should_return_list_from_references_field_with_schema_id_list_for_referenced_ids_when_null()
        {
            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant,
                new ReferencesFieldProperties { SchemaId = schemaId });

            var result = sut.ExtractReferences(null).ToArray();

            Assert.Equal(new[] { schemaId }, result);
        }

        [Fact]
        public void Should_return_list_from_references_field_with_schema_id_for_referenced_ids_when_other_type()
        {
            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant,
                new ReferencesFieldProperties { SchemaId = schemaId });

            var result = sut.ExtractReferences("invalid").ToArray();

            Assert.Equal(new[] { schemaId }, result);
        }

        [Fact]
        public void Should_return_null_from_references_field_when_removing_references_from_null_array()
        {
            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant);

            var result = sut.CleanReferences(null, null);

            Assert.Null(result);
        }

        [Fact]
        public void Should_remove_deleted_references_from_references_field()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant,
                new ReferencesFieldProperties { SchemaId = schemaId });

            var result = sut.CleanReferences(CreateValue(id1, id2), new HashSet<Guid>(new[] { id2 }));

            Assert.Equal(CreateValue(id1, schemaId), result);
        }

        [Fact]
        public void Should_remove_all_references_from_references_field_when_schema_is_removed()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant,
                new ReferencesFieldProperties { SchemaId = schemaId });

            var result = sut.CleanReferences(CreateValue(id1, id2), new HashSet<Guid>(new[] { schemaId }));

            Assert.Equal(CreateValue(), result);
        }

        [Fact]
        public void Should_return_same_token_from_references_field_when_removing_references_and_nothing_to_remove()
        {
            var id1 = Guid.NewGuid();
            var id2 = Guid.NewGuid();

            var sut = new ReferencesField(1, "my-refs", Partitioning.Invariant);

            var token = CreateValue(id1, id2);
            var result = sut.CleanReferences(token, new HashSet<Guid>(new[] { Guid.NewGuid() }));

            Assert.Same(token, result);
        }

        private static JToken CreateValue(params Guid[] ids)
        {
            return ids == null ? JValue.CreateNull() : (JToken)new JArray(ids.OfType<object>().ToArray());
        }
    }
}