diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/GraphStageLogicSpec.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/GraphStageLogicSpec.scala index 80bb3c6cd7..15efa87308 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/GraphStageLogicSpec.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/GraphStageLogicSpec.scala @@ -14,7 +14,7 @@ import akka.stream.impl.fusing.GraphInterpreter._ import org.scalactic.ConversionCheckedTripleEquals import scala.concurrent.duration.Duration -class GraphStageLogicSpec extends GraphInterpreterSpecKit with ConversionCheckedTripleEquals { +class GraphStageLogicSpec extends AkkaSpec with GraphInterpreterSpecKit with ConversionCheckedTripleEquals { implicit val mat = ActorMaterializer() @@ -174,4 +174,4 @@ class GraphStageLogicSpec extends GraphInterpreterSpecKit with ConversionChecked } -} \ No newline at end of file +} diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterFailureModesSpec.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterFailureModesSpec.scala index a1b0d01f59..7f46c03e98 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterFailureModesSpec.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterFailureModesSpec.scala @@ -2,8 +2,9 @@ package akka.stream.impl.fusing import akka.stream.testkit.Utils.TE import akka.testkit.EventFilter +import akka.stream.testkit.AkkaSpec -class GraphInterpreterFailureModesSpec extends GraphInterpreterSpecKit { +class GraphInterpreterFailureModesSpec extends AkkaSpec with GraphInterpreterSpecKit { "GraphInterpreter" must { diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterPortsSpec.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterPortsSpec.scala index 88a9a1a527..7a48944826 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterPortsSpec.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterPortsSpec.scala @@ -6,7 +6,7 @@ package akka.stream.impl.fusing import akka.stream.testkit.AkkaSpec import akka.stream.testkit.Utils._ -class GraphInterpreterPortsSpec extends GraphInterpreterSpecKit { +class GraphInterpreterPortsSpec extends AkkaSpec with GraphInterpreterSpecKit { "Port states" must { diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpec.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpec.scala index 56ea126be9..990c64120f 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpec.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpec.scala @@ -8,7 +8,7 @@ import akka.stream.testkit.AkkaSpec import akka.stream.scaladsl.{ Merge, Broadcast, Balance, Zip } import GraphInterpreter._ -class GraphInterpreterSpec extends GraphInterpreterSpecKit { +class GraphInterpreterSpec extends AkkaSpec with GraphInterpreterSpecKit { import GraphStages._ "GraphInterpreter" must { diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpecKit.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpecKit.scala index 6428195ebb..5825ba3c55 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpecKit.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/GraphInterpreterSpecKit.scala @@ -11,8 +11,9 @@ import akka.stream.stage.{ GraphStage, GraphStageLogic, InHandler, OutHandler, _ import akka.stream.testkit.AkkaSpec import akka.stream.testkit.Utils.TE import akka.stream.impl.fusing.GraphInterpreter.GraphAssembly +import akka.event.NoLogging -trait GraphInterpreterSpecKit extends AkkaSpec { +trait GraphInterpreterSpecKit { abstract class Builder { private var _interpreter: GraphInterpreter = _ @@ -70,7 +71,7 @@ trait GraphInterpreterSpecKit extends AkkaSpec { val assembly = buildAssembly() val (inHandlers, outHandlers, logics, _) = assembly.materialize(Attributes.none) - _interpreter = new GraphInterpreter(assembly, NoMaterializer, Logging(system, classOf[TestSetup]), inHandlers, outHandlers, logics, (_, _, _) ⇒ ()) + _interpreter = new GraphInterpreter(assembly, NoMaterializer, NoLogging, inHandlers, outHandlers, logics, (_, _, _) ⇒ ()) for ((upstream, i) ← upstreams.zipWithIndex) { _interpreter.attachUpstreamBoundary(i, upstream._1) @@ -86,7 +87,7 @@ trait GraphInterpreterSpecKit extends AkkaSpec { def manualInit(assembly: GraphAssembly): Unit = { val (inHandlers, outHandlers, logics, _) = assembly.materialize(Attributes.none) - _interpreter = new GraphInterpreter(assembly, NoMaterializer, Logging(system, classOf[TestSetup]), inHandlers, outHandlers, logics, (_, _, _) ⇒ ()) + _interpreter = new GraphInterpreter(assembly, NoMaterializer, NoLogging, inHandlers, outHandlers, logics, (_, _, _) ⇒ ()) } def builder(stages: GraphStage[_ <: Shape]*): AssemblyBuilder = new AssemblyBuilder(stages) diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/InterpreterSupervisionSpec.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/InterpreterSupervisionSpec.scala index 238c246678..9a5ffeb5c7 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/InterpreterSupervisionSpec.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/InterpreterSupervisionSpec.scala @@ -11,6 +11,7 @@ import akka.stream.stage.PushPullStage import akka.stream.stage.Stage import akka.stream.stage.TerminationDirective import akka.stream.stage.SyncDirective +import akka.stream.testkit.AkkaSpec object InterpreterSupervisionSpec { val TE = new Exception("TEST") with NoStackTrace { @@ -70,7 +71,7 @@ object InterpreterSupervisionSpec { } -class InterpreterSupervisionSpec extends GraphInterpreterSpecKit { +class InterpreterSupervisionSpec extends AkkaSpec with GraphInterpreterSpecKit { import InterpreterSupervisionSpec._ import Supervision.stoppingDecider import Supervision.resumingDecider diff --git a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/LifecycleInterpreterSpec.scala b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/LifecycleInterpreterSpec.scala index 97e7b0de5f..5b8c260d39 100644 --- a/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/LifecycleInterpreterSpec.scala +++ b/akka-stream-tests/src/test/scala/akka/stream/impl/fusing/LifecycleInterpreterSpec.scala @@ -9,7 +9,7 @@ import akka.stream.testkit.Utils.TE import scala.concurrent.duration._ -class LifecycleInterpreterSpec extends GraphInterpreterSpecKit { +class LifecycleInterpreterSpec extends AkkaSpec with GraphInterpreterSpecKit { import akka.stream.Supervision._ "Interpreter" must {