src/chrome/installer/util/wmi_unittest.cc

   1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <windows.h>
   6
   7 #include "chrome/installer/util/wmi.h"
   8 #include "testing/gtest/include/gtest/gtest.h"
   9
  10 namespace installer {
  11
  12 TEST(WMITest, TestLocalConnectionSecurityBlanket) {
  13   IWbemServices* services = NULL;
  14   EXPECT_TRUE(WMI::CreateLocalConnection(true, &services));
  15   ASSERT_TRUE(NULL != services);
  16   ULONG refs = services->Release();
  17   EXPECT_EQ(refs, 0);
  18 }
  19
  20 TEST(WMITest, TestLocalConnectionNoSecurityBlanket) {
  21   IWbemServices* services = NULL;
  22   EXPECT_TRUE(WMI::CreateLocalConnection(false, &services));
  23   ASSERT_TRUE(NULL != services);
  24   ULONG refs = services->Release();
  25   EXPECT_EQ(refs, 0);
  26 }
  27
  28 TEST(WMITest, TestCreateClassMethod) {
  29   IWbemServices* wmi_services = NULL;
  30   EXPECT_TRUE(WMI::CreateLocalConnection(true, &wmi_services));
  31   ASSERT_TRUE(NULL != wmi_services);
  32   IWbemClassObject* class_method = NULL;
  33   EXPECT_TRUE(WMI::CreateClassMethodObject(wmi_services,
  34                                                L"Win32_ShortcutFile",
  35                                                L"Rename", &class_method));
  36   ASSERT_TRUE(NULL != class_method);
  37   ULONG refs = class_method->Release();
  38   EXPECT_EQ(refs, 0);
  39   refs = wmi_services->Release();
  40   EXPECT_EQ(refs, 0);
  41 }
  42
  43 // Creates an instance of cmd which executes 'echo' and exits immediately.
  44 TEST(WMITest, TestLaunchProcess) {
  45   int pid = 0;
  46   bool result = WMIProcess::Launch(L"cmd.exe /c echo excelent!", &pid);
  47   EXPECT_TRUE(result);
  48   EXPECT_GT(pid, 0);
  49 }
  50
  51 }  // namespace installer