Removed exposure of properties variable.

codefeedr · Feb 20, 2019 · 9dcd0a0 · 9dcd0a0
1 parent 42cac2c
commit 9dcd0a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/codefeedr-core/src/main/scala/org/codefeedr/pipeline/Stage.scala b/codefeedr-core/src/main/scala/org/codefeedr/pipeline/Stage.scala
@@ -73,7 +73,7 @@ protected[codefeedr] abstract class Stage[
     *
     * @return The properties of this stage.
     */
-  def properties: Properties =
+  private def properties: Properties =
     if (pipeline != null) pipeline.propertiesOf(this) else null
 
   /** Setups the pipeline object with a pipeline.

diff --git a/codefeedr-core/src/test/scala/org/codefeedr/pipeline/StageTest.scala b/codefeedr-core/src/test/scala/org/codefeedr/pipeline/StageTest.scala
@@ -102,7 +102,7 @@ class StageTest extends FunSuite {
 
     stage.setUp(pipeline)
 
-    assert(stage.properties.get("a").get == "b")
-    assert(stage.properties.get("c").isEmpty)
+    assert(stage.getContext.stageProperties.get("a").get == "b")
+    assert(stage.getContext.stageProperties.get("c").isEmpty)
   }
 }