Complete adding coverage

CMakeLists.txt

@@ -29,6 +29,10 @@ option(CODE_COVERAGE "Collect coverage from test library"         ON)
 
 set(CMAKE_BUILD_TYPE "Debug")
 
+# CMakes modules
+# --------------
+set(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake/modules ${CMAKE_MODULE_PATH})
+
 
 # Subdirectories
 # --------------
@@ -38,13 +42,16 @@ add_subdirectory(sample)
 
 # Testing
 # -------
-
 if (ENABLE_TESTING)
+  MESSAGE("Testing enabled")
   enable_testing()
-  #if (CODE_COVERAGE)
-    SET(GCC_COVERAGE_COMPILE_FLAGS "-fprofile-arcs -ftest-coverage")
-    SET(GCC_COVERAGE_LINK_FLAGS    "--coverage")
-  #endif (CODE_COVERAGE)
+  if (CODE_COVERAGE)
+    include(CodeCoverage)
+    MESSAGE("Code coverage enabled")
+    set(CMAKE_C_FLAGS " ${CMAKE_C_FLAGS} -fprofile-arcs -ftest-coverage")
+    set(CMAKE_CXX_FLAGS " ${CMAKE_CXX_FLAGS} -fprofile-arcs -ftest-coverage")
+    SET(GCC_COVERAGE_LINK_FLAGS " ${GCC_COVERAGE_LINK_FLAGS} -lgcov --coverage")
+  endif (CODE_COVERAGE)
   find_package(Catch2 3 REQUIRED)
   include(CTest)
   include(Catch)