Use pluginDef

Fixes #9314
elastic · Apr 4, 2018 · cf16429 · cf16429
1 parent ec3e8be
commit cf16429
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/logstash-core/src/test/java/org/logstash/config/ir/imperative/IfStatementTest.java b/logstash-core/src/test/java/org/logstash/config/ir/imperative/IfStatementTest.java
@@ -37,7 +37,7 @@ public void testEmptyIf() throws InvalidIRException {
     @Test
     public void testIfWithOneTrueStatement() throws InvalidIRException {
         PluginDefinition pluginDef = testPluginDefinition();
-        Statement trueStatement = new PluginStatement(randMeta(), testPluginDefinition());
+        Statement trueStatement = new PluginStatement(randMeta(), pluginDef);
         Statement falseStatement = new NoopStatement(randMeta());
         BooleanExpression ifExpression = createTestExpression();
         IfStatement ifStatement = new IfStatement(
@@ -53,7 +53,7 @@ public void testIfWithOneTrueStatement() throws InvalidIRException {
         Graph expected = new Graph();
         IfVertex expectedIf = DSL.gIf(randMeta(), ifExpression);
         expected.addVertex(expectedIf);
-        PluginVertex expectedT = DSL.gPlugin(randMeta(), testPluginDefinition());
+        PluginVertex expectedT = DSL.gPlugin(randMeta(), pluginDef);
         expected.chainVertices(true, expectedIf, expectedT);
 
         assertSyntaxEquals(expected, ifStatementGraph);