From ede318c2dd2b1a9a92bed5eac69020b43c6eba97 Mon Sep 17 00:00:00 2001
From: Marko Budiselic <marko.budiselic@memgraph.io>
Date: Wed, 22 Feb 2017 16:34:02 +0100
Subject: [PATCH] Memgraph gmock support has been added

Summary: Memgraph gmock support has been added

Reviewers: mislav.bradac

Reviewed By: mislav.bradac

Subscribers: buda

Differential Revision: https://phabricator.memgraph.io/D59
---
 CMakeLists.txt            | 2 +-
 tests/unit/CMakeLists.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 07c43fbc3..a93ee08c0 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -236,11 +236,11 @@ message(STATUS "Add unit test binaries: ${UNIT_TESTS}")
 # -----------------------------------------------------------------------------
 
 # includes
-include_directories(${CMAKE_SOURCE_DIR}/include)
 include_directories(${src_dir})
 include_directories(${build_include_dir})
 include_directories(${fmt_source_dir})
 include_directories(${yaml_include_dir})
+include_directories(${GTEST_INCLUDE_DIRS} ${GMOCK_INCLUDE_DIRS})
 # -----------------------------------------------------------------------------
 
 # openCypher parser -----------------------------------------------------------
diff --git a/tests/unit/CMakeLists.txt b/tests/unit/CMakeLists.txt
index 191ebc0e3..6c1aec109 100644
--- a/tests/unit/CMakeLists.txt
+++ b/tests/unit/CMakeLists.txt
@@ -27,7 +27,7 @@ foreach(test_cpp ${test_type_cpps})
 
     # link libraries
     # gtest
-    target_link_libraries(${target_name} gtest gtest_main)
+    target_link_libraries(${target_name} gtest gtest_main gmock)
     # filesystem
     target_link_libraries(${target_name} stdc++fs)
     # threads (cross-platform)