1 files changed, 69 insertions, 0 deletions
diff --git a/base/xdg_util_unittest.cc b/base/xdg_util_unittest.cc
new file mode 100644
index 0000000..d5ddc86
--- /dev/null
+++ b/base/xdg_util_unittest.cc
@@ -0,0 +1,69 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "base/xdg_util.h"
+
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+#include "base/env_var.h"
+
+using ::testing::_;
+using ::testing::Return;
+using ::testing::SetArgumentPointee;
+using ::testing::StrEq;
+
+namespace {
+
+class MockEnvVarGetter : public base::EnvVarGetter {
+ public:
+  MOCK_METHOD2(GetEnv, bool(const char*, std::string* result));
+};
+
+const char* kGnome = "gnome";
+const char* kKDE4 = "kde4";
+const char* kKDE = "kde";
+const char* kXFCE = "xfce";
+
+}  // namespace
+
+TEST(XDGUtilTest, GetDesktopEnvironmentGnome) {
+  MockEnvVarGetter getter;
+  EXPECT_CALL(getter, GetEnv(_, _)).WillRepeatedly(Return(false));
+  EXPECT_CALL(getter, GetEnv(StrEq("DESKTOP_SESSION"), _))
+      .WillOnce(DoAll(SetArgumentPointee<1>(kGnome), Return(true)));
+
+  EXPECT_EQ(base::DESKTOP_ENVIRONMENT_GNOME,
+            base::GetDesktopEnvironment(&getter));
+}
+
+TEST(XDGUtilTest, GetDesktopEnvironmentKDE4) {
+  MockEnvVarGetter getter;
+  EXPECT_CALL(getter, GetEnv(_, _)).WillRepeatedly(Return(false));
+  EXPECT_CALL(getter, GetEnv(StrEq("DESKTOP_SESSION"), _))
+      .WillOnce(DoAll(SetArgumentPointee<1>(kKDE4), Return(true)));
+
+  EXPECT_EQ(base::DESKTOP_ENVIRONMENT_KDE4,
+            base::GetDesktopEnvironment(&getter));
+}
+
+TEST(XDGUtilTest, GetDesktopEnvironmentKDE3) {
+  MockEnvVarGetter getter;
+  EXPECT_CALL(getter, GetEnv(_, _)).WillRepeatedly(Return(false));
+  EXPECT_CALL(getter, GetEnv(StrEq("DESKTOP_SESSION"), _))
+      .WillOnce(DoAll(SetArgumentPointee<1>(kKDE), Return(true)));
+
+  EXPECT_EQ(base::DESKTOP_ENVIRONMENT_KDE3,
+            base::GetDesktopEnvironment(&getter));
+}
+
+TEST(XDGUtilTest, GetDesktopEnvironmentXFCE) {
+  MockEnvVarGetter getter;
+  EXPECT_CALL(getter, GetEnv(_, _)).WillRepeatedly(Return(false));
+  EXPECT_CALL(getter, GetEnv(StrEq("DESKTOP_SESSION"), _))
+      .WillOnce(DoAll(SetArgumentPointee<1>(kXFCE), Return(true)));
+
+  EXPECT_EQ(base::DESKTOP_ENVIRONMENT_XFCE,
+            base::GetDesktopEnvironment(&getter));
+}