From 9ee25fd358bfc051c54eda7bec8802dd184c4afc Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?=
 =?utf8?q?=EC=96=B4Lab=28SR=29/Senior=20Engineer/=EC=82=BC=EC=84=B1?=
 =?utf8?q?=EC=A0=84=EC=9E=90?= <jh1302.park@samsung.com>
Date: Fri, 6 Apr 2018 09:52:36 +0900
Subject: [PATCH] [NNAPI Unittests] Add 'env' unittest (#442)

This commit adds a 'env' unittest in NNAPI unittests module.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
---
 tools/nnapi_unittests/CMakeLists.txt   |  5 +++++
 tools/nnapi_unittests/lib/env.test.cpp | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 37 insertions(+)
 create mode 100644 tools/nnapi_unittests/lib/env.test.cpp

diff --git a/tools/nnapi_unittests/CMakeLists.txt b/tools/nnapi_unittests/CMakeLists.txt
index e15ea91..3ccc547 100644
--- a/tools/nnapi_unittests/CMakeLists.txt
+++ b/tools/nnapi_unittests/CMakeLists.txt
@@ -1,10 +1,15 @@
 file(GLOB_RECURSE NNAPI_UNITTEST_LIB_SOURCES "lib/*.cpp")
+file(GLOB_RECURSE NNAPI_UNITTEST_LIB_TESTS "lib/*.test.cpp")
+list(REMOVE_ITEM NNAPI_UNITTEST_LIB_SOURCES ${NNAPI_UNITTEST_LIB_TESTS})
 
 add_library(nnapi_unittest_common ${NNAPI_UNITTEST_LIB_SOURCES})
 target_include_directories(nnapi_unittest_common PUBLIC "inc")
 target_link_libraries(nnapi_unittest_common nnfw_util)
 target_link_libraries(nnapi_unittest_common nnfw_support_tflite)
 
+add_executable(nnapi_unittest_lib_env_test "lib/env.test.cpp")
+target_link_libraries(nnapi_unittest_lib_env_test nnapi_unittest_common)
+
 function(add_nnapi_unittest NAME)
   add_executable(nnapi_unittest_${NAME} "tests/${NAME}.cpp")
   target_link_libraries(nnapi_unittest_${NAME} nnapi_unittest_common)
diff --git a/tools/nnapi_unittests/lib/env.test.cpp b/tools/nnapi_unittests/lib/env.test.cpp
new file mode 100644
index 0000000..0505a48
--- /dev/null
+++ b/tools/nnapi_unittests/lib/env.test.cpp
@@ -0,0 +1,32 @@
+#include "env.h"
+
+#include <string>
+
+#include <cstdlib>
+#include <cassert>
+
+inline void ensure(int err)
+{
+  assert(err == 0);
+}
+
+int main(int argc, char **argv)
+{
+  const std::string key{"TEST"};
+  const int num{3};
+
+  const auto str = std::to_string(num);
+
+  ensure(unsetenv(key.c_str()));
+  ensure(setenv(key.c_str(), str.c_str(), 0));
+
+  int value = 0;
+
+  assert(value != num);
+
+  IntVar buffer(key, value);
+
+  assert(buffer() == num);
+
+  return 0;
+}
-- 
2.7.4