Update ambient-viewer unittest

Change-Id: I3760ffe2991363d9f9409649f34dc051c321342b
Signed-off-by: mk5004.lee <mk5004.lee@samsung.com>

diff --git a/unittest/src/test_ambient_viewer.cc b/unittest/src/test_ambient_viewer.cc
index 8b431c5..1b47edc 100644 (file)
--- a/unittest/src/test_ambient_viewer.cc
+++ b/unittest/src/test_ambient_viewer.cc
@@ -19,6 +19,8 @@
 #include <gmock/gmock.h>
 
 #include <ambient-viewer.hh>
+#include <top-app-surface.hh>
+#include <watch-surface.hh>
 
 using namespace std;
 using namespace ambient_viewer;
@@ -42,13 +44,24 @@ class AmbientViewerTest : public ::testing::Test {
   virtual void SetUp() {
     Evas_Object* win = elm_win_add(NULL, "Widget Viewer", ELM_WIN_BASIC);
     stub = new AmbientViewerStub(win);
+    AmbientViewerTest::stub->Monitor();
   }
   virtual void TearDown() {
+    AmbientViewerTest::stub->Unmonitor();
     delete stub;
   }
 };
 
-
 TEST_F(AmbientViewerTest, create) {
   EXPECT_NE(AmbientViewerTest::stub, nullptr);
 }
+
+TEST_F(AmbientViewerTest, NotifyAmbientEvent) {
+  Bundle b;
+  EXPECT_EQ(AmbientViewerTest::stub->NotifyAmbientEvent(true,
+    AmbientViewer::DIRECTION_ALL, b), 0);
+}
+
+TEST_F(AmbientViewerTest, BlockUpdate) {
+  AmbientViewerTest::stub->BlockUpdate(true);
+}