Add workspace to desktop

ui/aura/test/aura_test_base.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 "ui/aura/test/aura_test_base.h"
 
 #if defined(OS_WIN)
 #include <ole2.h>
 #endif
 
@@ skipping 28 matching lines @@
   // Flush the message loop because we have pending release tasks
   // and these tasks if un-executed would upset Valgrind.
   message_loop_.RunAllPendingWithDispatcher(
       Desktop::GetInstance()->GetDispatcher());
 
   // Ensure that we don't use the previously-allocated static Desktop object
   // later -- on Linux, it holds a reference to our message loop's X connection.
   aura::Desktop::DeleteInstanceForTesting();
 }
 
+TestDesktopDelegate* AuraTestBase::GetTestDesktopDelegate() {
+  return static_cast<TestDesktopDelegate*>(
+      aura::Desktop::GetInstance()->delegate());
+}
+
 void AuraTestBase::SetUp() {
   testing::Test::SetUp();
   setup_called_ = true;
 }
 
 void AuraTestBase::TearDown() {
   teardown_called_ = true;
   testing::Test::TearDown();
 }
 
 }  // namespace test
 }  // namespace aura