Merge branch 'master' of https://twinetfs.visualstudio.com/Tango/_git/Tango

1 files changed, 10 insertions, 1 deletions

diff --git a/Software/Visual_Studio/Tango.BL/Builders/RmlsCollectionBuilder.cs b/Software/Visual_Studio/Tango.BL/Builders/RmlsCollectionBuilder.cs
index b23eb9507..8a5b4b00c 100644
--- a/Software/Visual_Studio/Tango.BL/Builders/RmlsCollectionBuilder.cs
+++ b/Software/Visual_Studio/Tango.BL/Builders/RmlsCollectionBuilder.cs
@@ -5,6 +5,7 @@ using System.Text;
 using System.Threading.Tasks;
 using Tango.BL.Entities;
 using System.Data.Entity;
+using Tango.BL.Enumerations;
 
 namespace Tango.BL.Builders
 {
@@ -38,10 +39,18 @@ namespace Tango.BL.Builders
             });
         }
 
-        public virtual RmlsCollectionBuilder WithCCT()
+        public virtual RmlsCollectionBuilder ForHeadType(HeadTypes headType)
         {
             return AddQueryStep(2, (query) =>
             {
+                return query.Where(x => x.HeadType == (int)headType);
+            });
+        }
+
+        public virtual RmlsCollectionBuilder WithCCT()
+        {
+            return AddQueryStep(3, (query) =>
+            {
                 return query.Include(x => x.Cct);
             });
         }