15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/pickle.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/ui/views/extensions/browser_action_drag_data.h" 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/test/base/testing_profile.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/dragdrop/os_exchange_data.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/dragdrop/os_exchange_data_provider_win.h" 117dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "url/gurl.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ui::OSExchangeData::Provider* CloneProvider(const ui::OSExchangeData& data) { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new ui::OSExchangeDataProviderWin( 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::OSExchangeDataProviderWin::GetIDataObject(data)); 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef testing::Test BrowserActionDragDataTest; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(BrowserActionDragDataTest, ArbitraryFormat) { 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestingProfile profile; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) profile.SetID(L"id"); 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::OSExchangeData data; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data.SetURL(GURL("http://www.google.com"), L"Title"); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We only support our format, so this should not succeed. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserActionDragData drag_data; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(drag_data.Read(ui::OSExchangeData(CloneProvider(data)))); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(BrowserActionDragDataTest, BrowserActionDragDataFormat) { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestingProfile profile; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Profile* profile_ptr = &profile; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) profile.SetID(L"id"); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string extension_id = "42"; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Pickle pickle; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle.WriteBytes(&profile_ptr, sizeof(&profile)); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle.WriteString(extension_id); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle.WriteUInt64(42); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::OSExchangeData data; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data.SetPickledData(BrowserActionDragData::GetBrowserActionCustomFormat(), 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserActionDragData drag_data; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(drag_data.Read(ui::OSExchangeData(CloneProvider(data)))); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(drag_data.IsFromProfile(profile.GetOriginalProfile())); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_STREQ(extension_id.c_str(), drag_data.id().c_str()); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(42, drag_data.index()); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 57