@@ -12,6 +12,7 @@ import ru.d10xa.jsonlogviewer.decline.Config
1212import ru .d10xa .jsonlogviewer .decline .FieldNamesConfig
1313import ru .d10xa .jsonlogviewer .decline .TimestampConfig
1414import ru .d10xa .jsonlogviewer .query .QueryCompiler
15+ import ru .d10xa .jsonlogviewer .shell .ShellImpl
1516
1617import scala .concurrent .duration .*
1718
@@ -90,13 +91,9 @@ class LogViewerStreamIntegrationTest extends CatsEffectSuite {
9091 logInputChannel.stream
9192 }
9293
93- // Save original implementation and use test implementation
94- originalImpl = LogViewerStream .getStdInLinesStreamImpl
95- _ <- IO (LogViewerStream .setStdInLinesStreamImpl(testStreamImpl))
96-
9794 // Start stream processing in background
9895 streamFiber <- LogViewerStream
99- .stream(baseConfig, configRef)
96+ .stream(baseConfig, configRef, testStreamImpl, new ShellImpl )
10097 .evalTap(result => IO (results.append(result)))
10198 .compile
10299 .drain
@@ -130,7 +127,6 @@ class LogViewerStreamIntegrationTest extends CatsEffectSuite {
130127
131128 // Cleanup
132129 _ <- streamFiber.cancel
133- _ <- IO (LogViewerStream .setStdInLinesStreamImpl(originalImpl))
134130
135131 } yield {
136132 // Verify initial INFO filter
@@ -215,13 +211,9 @@ class LogViewerStreamIntegrationTest extends CatsEffectSuite {
215211 logInputChannel.stream
216212 }
217213
218- // Save original implementation and use test implementation
219- originalImpl = LogViewerStream .getStdInLinesStreamImpl
220- _ <- IO (LogViewerStream .setStdInLinesStreamImpl(testStreamImpl))
221-
222214 // Start stream processing in background
223215 streamFiber <- LogViewerStream
224- .stream(errorFilterConfig, configRef)
216+ .stream(errorFilterConfig, configRef, testStreamImpl, new ShellImpl )
225217 .evalTap(result => IO (results.append(result)))
226218 .compile
227219 .drain
@@ -255,7 +247,6 @@ class LogViewerStreamIntegrationTest extends CatsEffectSuite {
255247
256248 // Cleanup
257249 _ <- streamFiber.cancel
258- _ <- IO (LogViewerStream .setStdInLinesStreamImpl(originalImpl))
259250
260251 } yield {
261252 // Verify initial field name mapping
@@ -287,4 +278,4 @@ class LogViewerStreamIntegrationTest extends CatsEffectSuite {
287278 )
288279 }
289280 }
290- }
281+ }
0 commit comments