From 38596907bbb60b61219616ed5d5b9aed60edf4f4 Mon Sep 17 00:00:00 2001
From: Sounak Gupta <sounak.gupta@intel.com>
Date: Wed, 4 Jan 2023 17:33:03 -0800
Subject: [PATCH] added code coverage support in CacheLib

---
 cachelib/CMakeLists.txt |  5 +++++
 run_code_coverage.sh    | 20 ++++++++++++++++++++
 2 files changed, 25 insertions(+)
 create mode 100755 run_code_coverage.sh

diff --git a/cachelib/CMakeLists.txt b/cachelib/CMakeLists.txt
index 36df0dc19f..e77c25085c 100644
--- a/cachelib/CMakeLists.txt
+++ b/cachelib/CMakeLists.txt
@@ -85,6 +85,11 @@ set(CMAKE_MODULE_PATH
 set(CMAKE_CXX_STANDARD 17)
 set(CMAKE_CXX_STANDARD_REQUIRED True)
 
+if(COVERAGE_ENABLED)
+  # Add code coverage
+  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} --coverage -fprofile-arcs -ftest-coverage")
+endif()
+
 # include(fb_cxx_flags)
 message(STATUS "Update CXXFLAGS: ${CMAKE_CXX_FLAGS}")
 
diff --git a/run_code_coverage.sh b/run_code_coverage.sh
new file mode 100755
index 0000000000..7722e262bf
--- /dev/null
+++ b/run_code_coverage.sh
@@ -0,0 +1,20 @@
+#!/bin/bash
+
+#Build CacheLib with flag -DCOVERAGE_ENABLED=ON
+
+# Track coverage
+lcov -c -i -b . -d . -o Coverage.baseline
+./run_tests.sh
+lcov -c -d . -b . -o Coverage.out
+lcov -a Coverage.baseline -a Coverage.out -o Coverage.combined
+
+# Generate report
+COVERAGE_DIR='coverage_report'
+genhtml Coverage.combined -o ${COVERAGE_DIR}
+COVERAGE_REPORT="${COVERAGE_DIR}.tgz"
+tar -zcvf ${COVERAGE_REPORT} ${COVERAGE_DIR}
+echo "Created coverage report ${COVERAGE_REPORT}"
+
+# Cleanup
+rm Coverage.baseline Coverage.out Coverage.combined
+rm -rf ${COVERAGE_DIR}