From 0c32ccfcd0b770cdd4eeb9d778b7f72a233e229a Mon Sep 17 00:00:00 2001
From: Finn Williams <Finn.Williams@arm.com>
Date: Tue, 12 May 2020 13:37:06 +0100
Subject: IVGCVSW-4737 Add check for CpuRef back-end before using it in tests

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: Id7fbb7a39d2e9c7085636df44d4c8603a60cd4bf
---
 src/profiling/test/FileOnlyProfilingDecoratorTests.cpp | 10 ++++++----
 tests/profiling/gatordmock/tests/GatordMockTests.cpp   |  5 +++++
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp b/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp
index 164a94a9ec..80236ae4eb 100644
--- a/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp
+++ b/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp
@@ -3,10 +3,7 @@
 // SPDX-License-Identifier: MIT
 //
 
-#include <armnn/utility/IgnoreUnused.hpp>
-#include <FileOnlyProfilingConnection.hpp>
 #include <Filesystem.hpp>
-#include <NullProfilingConnection.hpp>
 #include <ProfilingService.hpp>
 #include <Runtime.hpp>
 #include "PrintPacketHeaderHandler.hpp"
@@ -17,7 +14,6 @@
 #include <boost/test/unit_test.hpp>
 
 #include <cstdio>
-#include <fstream>
 #include <sstream>
 #include <sys/stat.h>
 
@@ -53,6 +49,12 @@ std::string UniqueFileName()
 
 BOOST_AUTO_TEST_CASE(TestFileOnlyProfiling)
 {
+    // This test requires the CpuRef backend to be enabled
+    if(!BackendRegistryInstance().IsBackendRegistered("CpuRef"))
+    {
+        return;
+    }
+
     // Create a temporary file name.
     boost::filesystem::path tempPath = boost::filesystem::temp_directory_path();
     boost::filesystem::path tempFile = UniqueFileName();
diff --git a/tests/profiling/gatordmock/tests/GatordMockTests.cpp b/tests/profiling/gatordmock/tests/GatordMockTests.cpp
index 98ca8ffe90..0170962261 100644
--- a/tests/profiling/gatordmock/tests/GatordMockTests.cpp
+++ b/tests/profiling/gatordmock/tests/GatordMockTests.cpp
@@ -377,6 +377,11 @@ BOOST_AUTO_TEST_CASE(GatorDMockEndToEnd)
 
 BOOST_AUTO_TEST_CASE(GatorDMockTimeLineActivation)
 {
+    // This test requires the CpuRef backend to be enabled
+    if(!BackendRegistryInstance().IsBackendRegistered("CpuRef"))
+    {
+        return;
+    }
     armnn::MockBackendInitialiser initialiser;
     // Setup the mock service to bind to the UDS.
     std::string udsNamespace = "gatord_namespace";
-- 
cgit v1.2.1