public async Task GetTotalCount_IfSearchCriteriaIsNull_ShouldBeEqual(int count)
        {
            // arrange
            var dataQuery     = Mock.Of <DataQuery>();
            var searchService = Mock.Of <IListEntrySearchService>();
            var dataSource    = new BaseDataSource(searchService, dataQuery);

            // act
            var result = await dataSource.GetTotalCountAsync();

            // assert
            result.Should().Be(count);
        }
        public async Task GetTotalCount_IfListEntriesNotNull_ShouldBeEqual(int count)
        {
            // arrange
            var dataQuery     = Mock.Of <DataQuery>(t => t.ListEntries == new[] { new ListEntryBase() });
            var searchService = new Mock <IListEntrySearchService>();

            var dataSource = new BaseDataSource(searchService.Object, dataQuery);

            // act
            var result = await dataSource.GetTotalCountAsync();

            // assert
            result.Should().Be(count);
        }
        public async Task GetTotalCount_IfSearchCriteriaIsNotNull_ShouldBeEqual(int count)
        {
            // arrange
            var dataQuery             = Mock.Of <DataQuery>(t => t.SearchCriteria == new CatalogListEntrySearchCriteria());
            var searchService         = new Mock <IListEntrySearchService>();
            var listEntrySearchResult = new ListEntrySearchResult {
                TotalCount = 1
            };

            searchService.Setup(t => t.SearchAsync(It.IsAny <CatalogListEntrySearchCriteria>())).ReturnsAsync(listEntrySearchResult);

            var dataSource = new BaseDataSource(searchService.Object, dataQuery);

            // act
            var result = await dataSource.GetTotalCountAsync();

            // assert
            result.Should().Be(count);
        }