squidex/backend/tools/TestSuite/TestSuite.ApiTests/ContentQueryTests.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using Squidex.ClientLibrary;
using TestSuite.Model;
using Xunit;

#pragma warning disable SA1300 // Element should begin with upper-case letter

namespace TestSuite.ApiTests
{
    public class ContentQueryTests : IClassFixture<ContentQueryFixture>
    {
        public ContentQueryFixture _ { get; }

        public ContentQueryTests(ContentQueryFixture fixture)
        {
            _ = fixture;
        }

        [Fact]
        public async Task Should_query_newly_created_schema()
        {
            for (var i = 0; i < 20; i++)
            {
                var schemaName = $"schema-{Guid.NewGuid()}";

                await TestEntity.CreateSchemaAsync(_.Schemas, _.AppName, schemaName);

                var contentClient = _.ClientManager.CreateContentsClient<TestEntity, TestEntityData>(schemaName);
                var contentItems = await contentClient.GetAsync();

                Assert.Equal(0, contentItems.Total);
            }
        }

        [Fact]
        public async Task Should_query_by_ids()
        {
            var q = new ContentQuery { OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            var itemsById = await _.Contents.GetAsync(new HashSet<string>(items.Items.Take(3).Select(x => x.Id)));

            Assert.Equal(3, itemsById.Items.Count);
            Assert.Equal(3, itemsById.Total);

            foreach (var item in itemsById.Items)
            {
                Assert.Equal(_.AppName, item.AppName);
                Assert.Equal(_.SchemaName, item.SchemaName);
            }
        }

        [Fact]
        public async Task Should_return_all_with_odata()
        {
            var q = new ContentQuery { OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_all_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv", order = "ascending"
                        }
                    }
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_with_odata()
        {
            var q = new ContentQuery { OrderBy = "data/number/iv asc", Skip = 5 };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv", order = "ascending"
                        }
                    },
                    skip = 5
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_and_top_with_odata()
        {
            var q = new ContentQuery { Skip = 2, Top = 5, OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 3, 4, 5, 6, 7 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_and_top_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    skip = 2,
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv", order = "ascending"
                        }
                    },
                    top = 5
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 3, 4, 5, 6, 7 });
        }

        [Fact]
        public async Task Should_return_items_by_filter_with_odata()
        {
            var q = new ContentQuery { Filter = "data/number/iv gt 3 and data/number/iv lt 7", OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_return_items_by_filter_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv", order = "ascending"
                        }
                    },
                    filter = new
                    {
                        and = new[]
                        {
                            new
                            {
                                path = "data.number.iv",
                                op = "gt",
                                value = 3
                            },
                            new
                            {
                                path = "data.number.iv",
                                op = "lt",
                                value = 7
                            }
                        }
                    }
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_return_items_by_json_filter_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.json.iv.nested1.nested2", order = "ascending"
                        }
                    },
                    filter = new
                    {
                        and = new[]
                        {
                            new
                            {
                                path = "data.json.iv.nested1.nested2",
                                op = "gt",
                                value = 3
                            },
                            new
                            {
                                path = "data.json.iv.nested1.nested2",
                                op = "lt",
                                value = 7
                            }
                        }
                    }
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_return_items_by_full_text_with_odata()
        {
            var q = new ContentQuery { Search = "1" };

            // Query multiple times to wait for async text indexer.
            for (var i = 0; i < 10; i++)
            {
                await Task.Delay(500);

                var items = await _.Contents.GetAsync(q);

                if (items.Items.Any())
                {
                    AssertItems(items, 1, new[] { 1 });
                    return;
                }
            }

            Assert.False(true);
        }

        [Fact]
        public async Task Should_return_items_by_full_text_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    fullText = "2"
                }
            };

            // Query multiple times to wait for async text indexer.
            for (var i = 0; i < 10; i++)
            {
                await Task.Delay(500);

                var items = await _.Contents.GetAsync(q);

                if (items.Items.Any())
                {
                    AssertItems(items, 1, new[] { 2 });
                    return;
                }
            }

            Assert.False(true);
        }

        [Fact]
        public async Task Should_return_items_by_near_location_with_odata()
        {
            var q = new ContentQuery { Filter = "geo.distance(data/geo/iv, geography'POINT(3 3)') lt 1000" };

            // Query multiple times to wait for async text indexer.
            for (var i = 0; i < 20; i++)
            {
                await Task.Delay(500);

                var items = await _.Contents.GetAsync(q);

                if (items.Items.Any())
                {
                    AssertItems(items, 1, new[] { 3 });
                    return;
                }
            }

            Assert.False(true);
        }

        [Fact]
        public async Task Should_return_items_by_near_location_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    filter = new
                    {
                        path = "data.geo.iv",
                        op = "lt",
                        value = new
                        {
                            longitude = 3,
                            latitude = 3,
                            distance = 1000
                        }
                    }
                }
            };

            // Query multiple times to wait for async text indexer.
            for (var i = 0; i < 20; i++)
            {
                await Task.Delay(500);

                var items = await _.Contents.GetAsync(q);

                if (items.Items.Any())
                {
                    AssertItems(items, 1, new[] { 3 });
                    return;
                }
            }

            Assert.False(true);
        }

        [Fact]
        public async Task Should_return_items_by_near_geoson_location_with_odata()
        {
            var q = new ContentQuery { Filter = "geo.distance(data/geo/iv, geography'POINT(4 4)') lt 1000" };

            // Query multiple times to wait for async text indexer.
            for (var i = 0; i < 10; i++)
            {
                await Task.Delay(500);

                var items = await _.Contents.GetAsync(q);

                if (items.Items.Any())
                {
                    AssertItems(items, 1, new[] { 4 });
                    return;
                }
            }

            Assert.False(true);
        }

        [Fact]
        public async Task Should_return_items_by_near_geoson_location_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    filter = new
                    {
                        path = "data.geo.iv",
                        op = "lt",
                        value = new
                        {
                            longitude = 4,
                            latitude = 4,
                            distance = 1000
                        }
                    }
                }
            };

            // Query multiple times to wait for async text indexer.
            for (var i = 0; i < 10; i++)
            {
                await Task.Delay(500);

                var items = await _.Contents.GetAsync(q);

                if (items.Items.Any())
                {
                    AssertItems(items, 1, new[] { 4 });
                    return;
                }
            }

            Assert.False(true);
        }

        [Fact]
        public async Task Should_create_and_query_with_inline_graphql()
        {
            var query = new
            {
                query = @"
                    mutation {
                        createMyReadsContent(data: {
                            number: {
                                iv: 555
                            }
                        }) {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }"
            };

            var result = await _.Contents.GraphQlAsync<JObject>(query);

            var value = result["createMyReadsContent"]["data"]["number"]["iv"].Value<int>();

            Assert.Equal(555, value);
        }

        [Fact]
        public async Task Should_create_and_query_with_variable_graphql()
        {
            var query = new
            {
                query = @"
                    mutation Mutation($data: MyReadsDataInputDto!) {
                        createMyReadsContent(data: $data) {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    data = new
                    {
                        number = new
                        {
                            iv = 998
                        }
                    }
                }
            };

            var result = await _.Contents.GraphQlAsync<JObject>(query);

            var value = result["createMyReadsContent"]["data"]["number"]["iv"].Value<int>();

            Assert.Equal(998, value);
        }

        [Fact]
        public async Task Should_batch_query_items_with_graphql()
        {
            var query1 = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            var query2 = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 4 and data/number/iv lt 7"
                }
            };

            var results = await _.Contents.GraphQlAsync<QueryResult>(new[] { query1, query2 });

            var items1 = results.ElementAt(0).Data.Items;
            var items2 = results.ElementAt(1).Data.Items;

            Assert.Equal(items1.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
            Assert.Equal(items2.Select(x => x.Data.Number).ToArray(), new[] { 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_with_graphql()
        {
            var query = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            var result = await _.Contents.GraphQlAsync<QueryResult>(query);

            var items = result.Items;

            Assert.Equal(items.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_with_graphql_get()
        {
            var query = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            var result = await _.Contents.GraphQlGetAsync<QueryResult>(query);

            var items = result.Items;

            Assert.Equal(items.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_with_graphql_with_dynamic()
        {
            var query = new
            {
                query = @"
                {
                    queryMyReadsContents(filter: ""data/number/iv gt 3 and data/number/iv lt 7"", orderby: ""data/number/iv asc"") {
                      id,
                      data {
                        number {
                          iv
                        }
                      }
                    }
                }"
            };

            var result = await _.Contents.GraphQlAsync<JObject>(query);

            var items = result["queryMyReadsContents"];

            Assert.Equal(items.Select(x => x["data"]["number"]["iv"].Value<int>()).ToArray(), new[] { 4, 5, 6 });
        }

        private sealed class QueryResult
        {
            [JsonProperty("queryMyReadsContents")]
            public QueryItem[] Items { get; set; }
        }

        private sealed class QueryItem
        {
            public Guid Id { get; set; }

            public QueryItemData Data { get; set; }
        }

        private sealed class QueryItemData
        {
            [JsonConverter(typeof(InvariantConverter))]
            public int Number { get; set; }
        }

        private static void AssertItems(ContentsResult<TestEntity, TestEntityData> entities, int total, int[] expected)
        {
            Assert.Equal(total, entities.Total);
            Assert.Equal(expected, entities.Items.Select(x => x.Data.Number).ToArray());
        }
    }
}