From a1c042ea61a3aae23f48ee016be6ff8857c00bc7 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Sun, 20 Dec 2020 13:42:21 +0100
Subject: [PATCH] Handle WorkflowInstanceIdSpecification explicitly

---
 .../Stores/YesSqlWorkflowInstanceStore.cs                | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs b/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs
index 1af2809d7..6a15733c5 100644
--- a/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs
+++ b/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs
@@ -22,8 +22,9 @@ namespace Elsa.Persistence.YesSql.Stores
         protected override IQuery<WorkflowInstanceDocument> MapSpecification(ISpecification<WorkflowInstance> specification) =>
             specification switch
             {
-                EntityIdSpecification<WorkflowInstance> entityIdSpecification => Query<WorkflowInstanceIndex>(x => x.InstanceId == entityIdSpecification.Id),
-                BlockingActivityTypeSpecification blockingActivityTypeSpecification => Query<WorkflowInstanceBlockingActivitiesIndex>(x => x.ActivityType == blockingActivityTypeSpecification.ActivityType),
+                EntityIdSpecification<WorkflowInstance> spec => Query<WorkflowInstanceIndex>(x => x.InstanceId == spec.Id),
+                WorkflowInstanceIdSpecification spec => Query<WorkflowInstanceIndex>(x => x.InstanceId == spec.Id),
+                BlockingActivityTypeSpecification spec => Query<WorkflowInstanceBlockingActivitiesIndex>(x => x.ActivityType == spec.ActivityType),
                 _ => AutoMapSpecification<WorkflowInstanceIndex>(specification)
             };
 
@@ -34,13 +35,13 @@ namespace Elsa.Persistence.YesSql.Stores
                 case BlockingActivityTypeSpecification:
                 {
                     var indexedQuery = query.With<WorkflowInstanceBlockingActivitiesIndex>();
-                    var expression = orderBy.OrderByExpression.ConvertType<WorkflowInstance, WorkflowInstanceBlockingActivitiesIndex>();
+                    var expression = orderBy.OrderByExpression.ConvertType<WorkflowInstance, WorkflowInstanceDocument>().ConvertType<WorkflowInstanceDocument, WorkflowInstanceBlockingActivitiesIndex>();
                     return orderBy.SortDirection == SortDirection.Ascending ? indexedQuery.OrderBy(expression) : indexedQuery.OrderByDescending(expression);
                 }
                 default:
                 {
                     var indexedQuery = query.With<WorkflowInstanceIndex>();
-                    var expression = orderBy.OrderByExpression.ConvertType<WorkflowInstance, WorkflowInstanceIndex>();
+                    var expression = orderBy.OrderByExpression.ConvertType<WorkflowInstance, WorkflowInstanceDocument>().ConvertType<WorkflowInstanceDocument, WorkflowInstanceIndex>();
                     return orderBy.SortDirection == SortDirection.Ascending ? indexedQuery.OrderBy(expression) : indexedQuery.OrderByDescending(expression);
                 }
             }