mash: Allow a subset of D-Bus clients to be created in DBusThreadManager

chromeos/dbus/dbus_client_bundle_unittest.cc

 // Copyright 2014 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 "chromeos/dbus/dbus_client_bundle.h"
 
-#include "base/compiler_specific.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chromeos {
 
+// Tests that real and fake clients can be created and that some clients can be
+// skipped.
+TEST(DBusClientBundleTest, CreateClientBundle) {
+  const DBusClientTypeMask real_clients =
+      DBusClientBundle::CRAS | DBusClientBundle::CROS_DISKS;
+  const DBusClientTypeMask fake_clients =
+      DBusClientBundle::PERMISSION_BROKER | DBusClientBundle::POWER_MANAGER;
+  DBusClientBundle bundle(real_clients, fake_clients);
+  EXPECT_TRUE(bundle.IsUsingAnyRealClient());
+
+  // Real clients are real.
+  EXPECT_TRUE(bundle.cras_audio_client());
+  EXPECT_TRUE(bundle.cros_disks_client());
+  EXPECT_TRUE(bundle.IsUsingReal(DBusClientBundle::CRAS));
+  EXPECT_TRUE(bundle.IsUsingReal(DBusClientBundle::CROS_DISKS));
+  EXPECT_FALSE(bundle.IsUsingFake(DBusClientBundle::CRAS));
+  EXPECT_FALSE(bundle.IsUsingFake(DBusClientBundle::CROS_DISKS));
+
+  // Fake clients are fake.
+  EXPECT_TRUE(bundle.permission_broker_client());
+  EXPECT_TRUE(bundle.power_manager_client());
+  EXPECT_TRUE(bundle.IsUsingFake(DBusClientBundle::PERMISSION_BROKER));
+  EXPECT_TRUE(bundle.IsUsingFake(DBusClientBundle::POWER_MANAGER));
+  EXPECT_FALSE(bundle.IsUsingReal(DBusClientBundle::PERMISSION_BROKER));
+  EXPECT_FALSE(bundle.IsUsingReal(DBusClientBundle::POWER_MANAGER));
+
+  // Uninitialized clients don't exist.
+  EXPECT_FALSE(bundle.update_engine_client());
+  EXPECT_FALSE(bundle.IsUsingReal(DBusClientBundle::UPDATE_ENGINE));
+  EXPECT_FALSE(bundle.IsUsingFake(DBusClientBundle::UPDATE_ENGINE));
+}
+
 TEST(DBusClientBundleTest, UnstubFlagParser) {
   EXPECT_EQ(0, DBusClientBundle::ParseUnstubList("foo"));
 
   EXPECT_EQ(DBusClientBundle::BLUETOOTH,
             DBusClientBundle::ParseUnstubList("BLUETOOTH"));
 
   EXPECT_EQ(DBusClientBundle::BLUETOOTH,
             DBusClientBundle::ParseUnstubList("bluetooth"));
 
   EXPECT_EQ(
       DBusClientBundle::CRAS |
           DBusClientBundle::CROS_DISKS |
           DBusClientBundle::DEBUG_DAEMON |
           DBusClientBundle::SHILL,
       DBusClientBundle::ParseUnstubList(
             "Cras,Cros_Disks,debug_daemon,Shill"));
 
   EXPECT_EQ(
       DBusClientBundle::CRAS |
           DBusClientBundle::CROS_DISKS |
           DBusClientBundle::DEBUG_DAEMON |
           DBusClientBundle::SHILL,
       DBusClientBundle::ParseUnstubList(
             "foo,Cras,Cros_Disks,debug_daemon,Shill"));
 }
 
 }  // namespace chromeos