Add private system extension API to get system update status

chrome/browser/chromeos/accessibility/accessibility_private_extension_apitest.cc

+// Copyright (c) 2011 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 "chrome/browser/chromeos/dbus/mock_dbus_thread_manager.h"
+#include "chrome/browser/chromeos/dbus/mock_update_engine_client.h"
+#include "chrome/browser/extensions/extension_apitest.h"
+
+using chromeos::UpdateEngineClient;
+using ::testing::Return;
+
+class AccessibilityPrivateApiTest : public ExtensionApiTest {
+ public:
+  AccessibilityPrivateApiTest() : mock_update_engine_client_(NULL) {}
+
+  virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
+    ExtensionApiTest::SetUpInProcessBrowserTestFixture();
+    chromeos::MockDBusThreadManager* mock_dbus_thread_manager =
+        new chromeos::MockDBusThreadManager;
+    chromeos::DBusThreadManager::InitializeForTesting(mock_dbus_thread_manager);
+    mock_update_engine_client_ =
+        mock_dbus_thread_manager->mock_update_engine_client();
+  }
+
+  virtual void TearDownInProcessBrowserTestFixture() OVERRIDE {
+    chromeos::DBusThreadManager::Shutdown();
+    ExtensionApiTest::TearDownInProcessBrowserTestFixture();
+  }
+
+ protected:
+  chromeos::MockUpdateEngineClient* mock_update_engine_client_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(AccessibilityPrivateApiTest);
+};
+
+IN_PROC_BROWSER_TEST_F(AccessibilityPrivateApiTest, GetUpdateStatus) {
+  UpdateEngineClient::Status status_not_available;
+  status_not_available.status = UpdateEngineClient::UPDATE_STATUS_IDLE;
+  UpdateEngineClient::Status status_updating;
+  status_updating.status = UpdateEngineClient::UPDATE_STATUS_DOWNLOADING;
+  status_updating.download_progress = 0.5;
+  UpdateEngineClient::Status status_boot_needed;
+  status_boot_needed.status =
+      UpdateEngineClient::UPDATE_STATUS_UPDATED_NEED_REBOOT;
+
+  EXPECT_CALL(*mock_update_engine_client_, GetLastStatus())
+      .WillOnce(Return(status_not_available))
+      .WillOnce(Return(status_updating))
+      .WillOnce(Return(status_boot_needed));
+  ASSERT_TRUE(RunComponentExtensionTest(
+      "accessibility_private/get_update_status")) << message_;
+}