From 2e0042c70b8cb3cdd303f4e396be3b55cb3103e7 Mon Sep 17 00:00:00 2001
From: Sebastian <sebastian@squidex.io>
Date: Tue, 10 Aug 2021 15:49:52 +0200
Subject: [PATCH] Code fixes.

---
 .../Apps/Templates/Builders/AssetFieldBuilder.cs    | 13 ++++++++++++-
 .../DomainObject/ContentDomainObjectTests.cs        |  2 --
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/backend/src/Squidex.Domain.Apps.Entities/Apps/Templates/Builders/AssetFieldBuilder.cs b/backend/src/Squidex.Domain.Apps.Entities/Apps/Templates/Builders/AssetFieldBuilder.cs
index 894c4d9cc..346e248c8 100644
--- a/backend/src/Squidex.Domain.Apps.Entities/Apps/Templates/Builders/AssetFieldBuilder.cs
+++ b/backend/src/Squidex.Domain.Apps.Entities/Apps/Templates/Builders/AssetFieldBuilder.cs
@@ -5,6 +5,7 @@
 //  All rights reserved. Licensed under the MIT license.
 // ==========================================================================
 
+using Squidex.Domain.Apps.Core.Assets;
 using Squidex.Domain.Apps.Core.Schemas;
 using Squidex.Domain.Apps.Entities.Schemas.Commands;
 
@@ -21,7 +22,17 @@ namespace Squidex.Domain.Apps.Entities.Apps.Templates.Builders
         {
             Properties<AssetsFieldProperties>(p => p with
             {
-                MustBeImage = true
+                ExpectedType = AssetType.Image
+            });
+
+            return this;
+        }
+
+        public AssetFieldBuilder MustBe(AssetType type)
+        {
+            Properties<AssetsFieldProperties>(p => p with
+            {
+                ExpectedType = type
             });
 
             return this;
diff --git a/backend/tests/Squidex.Domain.Apps.Entities.Tests/Contents/DomainObject/ContentDomainObjectTests.cs b/backend/tests/Squidex.Domain.Apps.Entities.Tests/Contents/DomainObject/ContentDomainObjectTests.cs
index 8c135b191..be7be6af9 100644
--- a/backend/tests/Squidex.Domain.Apps.Entities.Tests/Contents/DomainObject/ContentDomainObjectTests.cs
+++ b/backend/tests/Squidex.Domain.Apps.Entities.Tests/Contents/DomainObject/ContentDomainObjectTests.cs
@@ -797,8 +797,6 @@ namespace Squidex.Domain.Apps.Entities.Contents.DomainObject
         [Fact]
         public async Task CancelContentSchedule_create_events_and_unset_schedule()
         {
-            var dueTime = Instant.MaxValue;
-
             var command = new CancelContentSchedule();
 
             await ExecuteCreateAsync();