# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-arrays=z:1,z:2 -tf-input-shapes=':' -tf-output-arrays=z:2,z:1,a:0 -o - | FileCheck %s
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-prune-unused-nodes -tf-input-arrays=z:1,z:2 -tf-input-shapes=':' -tf-output-arrays=z:2,z:1,a:0 -o - | FileCheck --check-prefix=PRUNE %s

John Howard <******@****.***> 1717524306 -0700

zirain <******@****.***> 1703405786 +0800

Jacek Ewertowski <******@****.***> 1701261434 +0100

        then:
        1 * listener.onOutput({ it.message == 'debug' && it.logLevel == LogLevel.DEBUG })
        1 * listener.onOutput({ it.message == 'info' && it.logLevel == LogLevel.INFO })
        1 * listener.onOutput({ it.message == 'lifecycle' && it.logLevel == LogLevel.LIFECYCLE })
        1 * listener.onOutput({ it.message == 'quiet' && it.logLevel == LogLevel.QUIET })

        when:
        renderer.onOutput(userInputResumeEvent)

        then:
        1 * listener.onOutput(userInputResumeEvent)
        1 * listener.onOutput(testOutputEvent1)
        1 * listener.onOutput(testOutputEvent2)
        0 * listener.onOutput(_)
        0 * userInput._
        renderer.eventQueue.empty
    }

    def "passes through other events than user input events"() {
        given:

        then:
        1 * outputEventBroadcaster.onOutput(_ as UserInputRequestEvent)
        1 * outputEventBroadcaster.onOutput(_) >> { YesNoQuestionPromptEvent event ->
            assert event.question == 'question?'
        }
        1 * userInputReader.readInput() >> new UserInputReader.TextResponse(enteredUserInput)
        1 * outputEventBroadcaster.onOutput(_ as UserInputResumeEvent)
        0 * outputEventBroadcaster._

    def "consume correctly LogLevelChangeEvent"() {
        when:
        renderer.onOutput(new LogLevelChangeEvent(LogLevel.WARN))
        renderer.onOutput(event("lifecycle", LogLevel.LIFECYCLE))
        renderer.onOutput(event("warn", LogLevel.WARN))

        then:
        1 * listener.onOutput({it instanceof LogLevelChangeEvent})
        1 * listener.onOutput({it instanceof LogEvent && it.message == "warn"})
        0 * _
    }

        RenderableOutputEvent event1 = new StyledTextOutputEvent(tenAm, 'category', LogLevel.INFO, null, 'message')
        RenderableOutputEvent event2 = new StyledTextOutputEvent(tenAm, 'category2', LogLevel.INFO, null, 'message2')

        when:
        renderer.onOutput(new LogLevelChangeEvent(LogLevel.DEBUG))
        renderer.onOutput(event1)
        renderer.onOutput(event2)

        then:

        def logEvent3 = Mock(StyledTextOutputEvent)
        def logEvent4 = Mock(LogEvent)

        when:
        listener.onOutput(logEvent1)
        listener.withWorkerLoggingProtocol(protocol2) {
            listener.onOutput(logEvent2)
            listener.onOutput(logEvent3)
        }
        listener.onOutput(logEvent4)

        then:
        1 * protocol1.sendOutputEvent(logEvent1)
        1 * protocol2.sendOutputEvent(logEvent2)