// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

#include "TraceModule.h"

#include <gtest/gtest.h>
#include <thread>

using namespace Aws::IoTFleetWise::Platform::Linux;

TEST( TraceModuleTest, TraceModulePrint )
{
    TraceModule::get().setVariable( TraceVariable::READ_SOCKET_FRAMES_0, 10 );
    TraceModule::get().setVariable( TraceVariable::READ_SOCKET_FRAMES_0, 20 );
    TraceModule::get().setVariable( TraceVariable::READ_SOCKET_FRAMES_0, 15 );

    TraceModule::get().sectionBegin( TraceSection::BUILD_MQTT );
    std::this_thread::sleep_for( std::chrono::milliseconds( 4 ) );
    TraceModule::get().sectionEnd( TraceSection::BUILD_MQTT );

    std::this_thread::sleep_for( std::chrono::milliseconds( 9 ) );

    TraceModule::get().sectionBegin( TraceSection::BUILD_MQTT );
    std::this_thread::sleep_for( std::chrono::milliseconds( 5 ) );
    TraceModule::get().sectionEnd( TraceSection::BUILD_MQTT );

    std::this_thread::sleep_for( std::chrono::milliseconds( 11 ) );

    TraceModule::get().sectionBegin( TraceSection::BUILD_MQTT );
    std::this_thread::sleep_for( std::chrono::milliseconds( 6 ) );
    TraceModule::get().sectionEnd( TraceSection::BUILD_MQTT );

    ASSERT_EQ( TraceModule::get().getVariableMax( TraceVariable::READ_SOCKET_FRAMES_0 ), 20 );
    TraceModule::get().print();
    TraceModule::get().startNewObservationWindow();
    TraceModule::get().print();
    TraceModule::get().startNewObservationWindow();
}