removing services

LiveUI · Mar 20, 2018 · cafcb00 · cafcb00
1 parent 4014186
commit cafcb00
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/Sources/VaporTestTools/Extensions/Services/Services+Testing.swift b/Sources/VaporTestTools/Extensions/Services/Services+Testing.swift
@@ -0,0 +1,24 @@
+//
+//  Services+Testing.swift
+//  VaporTestTools
+//
+//  Created by Ondrej Rafaj on 20/03/2018.
+//
+
+import Foundation
+@testable import Service
+
+
+extension TestableProperty where TestableType == Services {
+
+    public mutating func remove<S>(type: S.Type) {
+        if let existing = element.factories.index(where: {
+            $0.serviceType is S.Type
+        }) {
+            element.factories.remove(at: existing)
+        }
+    }
+
+}
+
+
diff --git a/Sources/VaporTestTools/Properties/TestableProperty.swift b/Sources/VaporTestTools/Properties/TestableProperty.swift
@@ -10,7 +10,7 @@ import Foundation
 
 public struct TestableProperty<TestableType> {
 
-    public let element: TestableType
+    public var element: TestableType
 
     init(_ obj: TestableType) {
         element = obj

diff --git a/Sources/VaporTestTools/TestTools.swift b/Sources/VaporTestTools/TestTools.swift
@@ -36,6 +36,7 @@ extension Request: Testable { }
 extension HTTPRequest: Testable { }
 extension Response: Testable { }
 extension Application: Testable { }
+extension Services: Testable { }
 
 
 extension Dictionary: Testable where Key == String, Value == String {