MS. Display the proper error message when trying to delete Organization. Item 4510.

1 files changed, 15 insertions, 0 deletions

diff --git a/Software/Visual_Studio/Tango.BL/Builders/OrganizationBuilder.cs b/Software/Visual_Studio/Tango.BL/Builders/OrganizationBuilder.cs
index 3b4dbb28a..a4a207185 100644
--- a/Software/Visual_Studio/Tango.BL/Builders/OrganizationBuilder.cs
+++ b/Software/Visual_Studio/Tango.BL/Builders/OrganizationBuilder.cs
@@ -34,5 +34,20 @@ namespace Tango.BL.Builders
                .Include(x => x.UsersRoles).ToSynchronizedObservableCollection();
             });
         }
+
+        public virtual OrganizationBuilder WithMachines()
+        {
+            return AddQueryStep(2, (query) =>
+            {
+                return query.Include(x => x.Machines);
+            });
+        }
+        public virtual OrganizationBuilder WithSites()
+        {
+            return AddQueryStep(3, (query) =>
+            {
+                return query.Include(x => x.Sites);
+            });
+        }
     }
 }