improving test: check the return value from the key storage

ehe-pd · Apr 21, 2021 · fcf4a9f · fcf4a9f
1 parent d0eac7b
commit fcf4a9f
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/...oovy/com/dtolabs/rundeck/core/resources/ScriptPluginResourceModelSourceFactorySpec.groovy b/...oovy/com/dtolabs/rundeck/core/resources/ScriptPluginResourceModelSourceFactorySpec.groovy
@@ -31,18 +31,7 @@ class ScriptPluginResourceModelSourceFactorySpec  extends Specification{
         given:
 
         def storageTree = Mock(KeyStorageTree)
-        storageTree.getResource(_) >> Mock(Resource) {
-            getContents() >> Mock(ResourceMeta) {
-                writeContent(_) >> { args ->
-                    args[0].write('password'.bytes)
-                    return 6L
-                }
-            }
-        }
-
-        Services services = Mock(Services){
-            getService(KeyStorageTree.class) >> storageTree
-        }
+        Services services = Mock(Services)
 
         Properties configuration = new Properties()
         configuration.put("project",PROJECT_NAME)
@@ -93,8 +82,19 @@ class ScriptPluginResourceModelSourceFactorySpec  extends Specification{
         def resource = pluginFactory.createResourceModelSource(services, configuration)
 
         then:
-        resource !=null
+        1 * services.getService(KeyStorageTree.class) >> storageTree
+        storageTree.getResource(_) >> Mock(Resource) {
+            1* getContents() >> Mock(ResourceMeta) {
+                writeContent(_) >> { args ->
+                    args[0].write('password'.bytes)
+                    return 6L
+                }
+            }
+        }
 
+        resource !=null
+        resource.configuration.getProperty("token") != null
+        resource.configuration.getProperty("token") == "password"
     }