diff options
Diffstat (limited to 'tests/profiling/gatordmock/tests/GatordMockTests.cpp')
-rw-r--r-- | tests/profiling/gatordmock/tests/GatordMockTests.cpp | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/tests/profiling/gatordmock/tests/GatordMockTests.cpp b/tests/profiling/gatordmock/tests/GatordMockTests.cpp index 53f580deaf..26c0361200 100644 --- a/tests/profiling/gatordmock/tests/GatordMockTests.cpp +++ b/tests/profiling/gatordmock/tests/GatordMockTests.cpp @@ -3,12 +3,13 @@ // SPDX-License-Identifier: MIT // -#include "../GatordMockService.hpp" -#include "../PeriodicCounterCaptureCommandHandler.hpp" - #include <CommandHandlerRegistry.hpp> #include <DirectoryCaptureCommandHandler.hpp> #include <ProfilingService.hpp> +#include <GatordMockService.hpp> +#include <PeriodicCounterCaptureCommandHandler.hpp> +#include <StreamMetadataCommandHandler.hpp> +#include <TimelineDirectoryCaptureCommandHandler.hpp> #include <test/SendCounterPacketTests.hpp> @@ -117,15 +118,23 @@ BOOST_AUTO_TEST_CASE(GatorDMockEndToEnd) profiling::CommandHandlerRegistry registry; // Update with derived functors + gatordmock::StreamMetadataCommandHandler streamMetadataCommandHandler( + 0, 0, packetVersionResolver.ResolvePacketVersion(0, 0).GetEncodedValue(), true); + gatordmock::PeriodicCounterCaptureCommandHandler counterCaptureCommandHandler( 0, 4, packetVersionResolver.ResolvePacketVersion(0, 4).GetEncodedValue(), true); profiling::DirectoryCaptureCommandHandler directoryCaptureCommandHandler( 0, 2, packetVersionResolver.ResolvePacketVersion(0, 2).GetEncodedValue(), true); + gatordmock::TimelineDirectoryCaptureCommandHandler timelineDirectoryCaptureCommandHandler( + 1, 0, packetVersionResolver.ResolvePacketVersion(1, 0).GetEncodedValue(), true); + // Register different derived functors + registry.RegisterFunctor(&streamMetadataCommandHandler); registry.RegisterFunctor(&counterCaptureCommandHandler); registry.RegisterFunctor(&directoryCaptureCommandHandler); + registry.RegisterFunctor(&timelineDirectoryCaptureCommandHandler); // Setup the mock service to bind to the UDS. std::string udsNamespace = "gatord_namespace"; gatordmock::GatordMockService mockService(registry, false); |