Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(333)

Side by Side Diff: chrome/browser/ui/webui/print_preview/extension_printer_handler_unittest.cc

Issue 1163743002: Revert of Include USB printers in printer list as "provisional" devices. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2015 The Chromium Authors. All rights reserved. 1 // Copyright 2015 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include <string> 5 #include <string>
6 #include <vector> 6 #include <vector>
7 7
8 #include "base/bind.h" 8 #include "base/bind.h"
9 #include "base/files/file_util.h" 9 #include "base/files/file_util.h"
10 #include "base/files/scoped_temp_dir.h" 10 #include "base/files/scoped_temp_dir.h"
11 #include "base/json/json_string_value_serializer.h" 11 #include "base/json/json_string_value_serializer.h"
12 #include "base/macros.h" 12 #include "base/macros.h"
13 #include "base/memory/scoped_ptr.h" 13 #include "base/memory/scoped_ptr.h"
14 #include "base/run_loop.h" 14 #include "base/run_loop.h"
15 #include "base/strings/string16.h" 15 #include "base/strings/string16.h"
16 #include "base/strings/stringprintf.h"
17 #include "base/strings/utf_string_conversions.h" 16 #include "base/strings/utf_string_conversions.h"
18 #include "base/test/values_test_util.h"
19 #include "base/values.h" 17 #include "base/values.h"
20 #include "chrome/browser/extensions/test_extension_environment.h"
21 #include "chrome/browser/local_discovery/pwg_raster_converter.h" 18 #include "chrome/browser/local_discovery/pwg_raster_converter.h"
22 #include "chrome/browser/ui/webui/print_preview/extension_printer_handler.h" 19 #include "chrome/browser/ui/webui/print_preview/extension_printer_handler.h"
23 #include "chrome/test/base/testing_profile.h" 20 #include "chrome/test/base/testing_profile.h"
24 #include "device/core/device_client.h" 21 #include "content/public/test/test_browser_thread_bundle.h"
25 #include "device/usb/mock_usb_device.h"
26 #include "device/usb/mock_usb_service.h"
27 #include "extensions/browser/api/device_permissions_manager.h"
28 #include "extensions/browser/api/printer_provider/printer_provider_api.h" 22 #include "extensions/browser/api/printer_provider/printer_provider_api.h"
29 #include "extensions/browser/api/printer_provider/printer_provider_api_factory.h " 23 #include "extensions/browser/api/printer_provider/printer_provider_api_factory.h "
30 #include "extensions/browser/api/printer_provider/printer_provider_print_job.h" 24 #include "extensions/browser/api/printer_provider/printer_provider_print_job.h"
31 #include "extensions/common/extension.h"
32 #include "extensions/common/value_builder.h"
33 #include "printing/pdf_render_settings.h" 25 #include "printing/pdf_render_settings.h"
34 #include "printing/pwg_raster_settings.h" 26 #include "printing/pwg_raster_settings.h"
35 #include "printing/units.h" 27 #include "printing/units.h"
36 #include "testing/gtest/include/gtest/gtest.h" 28 #include "testing/gtest/include/gtest/gtest.h"
37 #include "ui/gfx/geometry/size.h" 29 #include "ui/gfx/geometry/size.h"
38 30
39 using device::MockUsbDevice;
40 using device::MockUsbService;
41 using extensions::DictionaryBuilder;
42 using extensions::Extension;
43 using extensions::PrinterProviderAPI; 31 using extensions::PrinterProviderAPI;
44 using extensions::PrinterProviderPrintJob; 32 using extensions::PrinterProviderPrintJob;
45 using extensions::TestExtensionEnvironment;
46 using local_discovery::PWGRasterConverter; 33 using local_discovery::PWGRasterConverter;
47 34
48 namespace { 35 namespace {
49 36
50 // Printer id used for requests in tests. 37 // Printer id used for requests in tests.
51 const char kPrinterId[] = "printer_id"; 38 const char kPrinterId[] = "printer_id";
52 39
53 // Printer list used a result for getPrinters. 40 // Printer list used a result for getPrinters.
54 const char kPrinterDescriptionList[] = 41 const char kPrinterDescriptionList[] =
55 "[{" 42 "[{"
(...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after
124 111
125 // Print ticket that has duplex parameter set. 112 // Print ticket that has duplex parameter set.
126 const char kPrintTicketWithDuplex[] = 113 const char kPrintTicketWithDuplex[] =
127 "{" 114 "{"
128 " \"version\": \"1.0\"," 115 " \"version\": \"1.0\","
129 " \"print\": {" 116 " \"print\": {"
130 " \"duplex\": {\"type\": \"LONG_EDGE\"}" 117 " \"duplex\": {\"type\": \"LONG_EDGE\"}"
131 " }" 118 " }"
132 "}"; 119 "}";
133 120
134 // An extension with permission for 1 printer it supports.
135 const char kExtension1[] =
136 "{"
137 " \"name\": \"Provider 1\","
138 " \"app\": {"
139 " \"background\": {"
140 " \"scripts\": [\"background.js\"]"
141 " }"
142 " },"
143 " \"permissions\": ["
144 " \"printerProvider\","
145 " \"usb\","
146 " {"
147 " \"usbDevices\": ["
148 " { \"vendorId\": 0, \"productId\": 1 }"
149 " ]"
150 " },"
151 " ],"
152 " \"usb_printers\": {"
153 " \"filters\": ["
154 " { \"vendorId\": 0, \"productId\": 0 },"
155 " { \"vendorId\": 0, \"productId\": 1 }"
156 " ]"
157 " }"
158 "}";
159
160 // An extension with permission for none of the printers it supports.
161 const char kExtension2[] =
162 "{"
163 " \"name\": \"Provider 2\","
164 " \"app\": {"
165 " \"background\": {"
166 " \"scripts\": [\"background.js\"]"
167 " }"
168 " },"
169 " \"permissions\": [ \"printerProvider\", \"usb\" ],"
170 " \"usb_printers\": {"
171 " \"filters\": ["
172 " { \"vendorId\": 0, \"productId\": 0 },"
173 " { \"vendorId\": 0, \"productId\": 1 }"
174 " ]"
175 " }"
176 "}";
177
178 const char kContentTypePDF[] = "application/pdf"; 121 const char kContentTypePDF[] = "application/pdf";
179 const char kContentTypePWG[] = "image/pwg-raster"; 122 const char kContentTypePWG[] = "image/pwg-raster";
180 123
181 // Print request status considered to be successful by fake PrinterProviderAPI. 124 // Print request status considered to be successful by fake PrinterProviderAPI.
182 const char kPrintRequestSuccess[] = "OK"; 125 const char kPrintRequestSuccess[] = "OK";
183 126
184 // Used as a callback to StartGetPrinters in tests. 127 // Used as a callback to StartGetPrinters in tests.
185 // Increases |*call_count| and records values returned by StartGetPrinters. 128 // Increases |*call_count| and records values returned by StartGetPrinters.
186 void RecordPrinterList(size_t* call_count, 129 void RecordPrinterList(size_t* call_count,
187 scoped_ptr<base::ListValue>* printers_out, 130 scoped_ptr<base::ListValue>* printers_out,
(...skipping 218 matching lines...) Expand 10 before | Expand all | Expand 10 after
406 pending_capability_callbacks_; 349 pending_capability_callbacks_;
407 std::vector<PrintRequestInfo> pending_print_requests_; 350 std::vector<PrintRequestInfo> pending_print_requests_;
408 351
409 DISALLOW_COPY_AND_ASSIGN(FakePrinterProviderAPI); 352 DISALLOW_COPY_AND_ASSIGN(FakePrinterProviderAPI);
410 }; 353 };
411 354
412 KeyedService* BuildTestingPrinterProviderAPI(content::BrowserContext* context) { 355 KeyedService* BuildTestingPrinterProviderAPI(content::BrowserContext* context) {
413 return new FakePrinterProviderAPI(); 356 return new FakePrinterProviderAPI();
414 } 357 }
415 358
416 class FakeDeviceClient : public device::DeviceClient {
417 public:
418 FakeDeviceClient() {}
419
420 // device::DeviceClient implementation:
421 device::UsbService* GetUsbService() override {
422 DCHECK(usb_service_);
423 return usb_service_;
424 }
425
426 void set_usb_service(device::UsbService* service) { usb_service_ = service; }
427
428 private:
429 device::UsbService* usb_service_ = nullptr;
430 };
431
432 } // namespace 359 } // namespace
433 360
434 class ExtensionPrinterHandlerTest : public testing::Test { 361 class ExtensionPrinterHandlerTest : public testing::Test {
435 public: 362 public:
436 ExtensionPrinterHandlerTest() : pwg_raster_converter_(NULL) {} 363 ExtensionPrinterHandlerTest() : pwg_raster_converter_(NULL) {}
437 ~ExtensionPrinterHandlerTest() override = default; 364 ~ExtensionPrinterHandlerTest() override = default;
438 365
439 void SetUp() override { 366 void SetUp() override {
440 extensions::PrinterProviderAPIFactory::GetInstance()->SetTestingFactory( 367 TestingProfile::Builder profile_builder;
441 env_.profile(), &BuildTestingPrinterProviderAPI); 368 profile_builder.AddTestingFactory(
369 extensions::PrinterProviderAPIFactory::GetInstance(),
370 &BuildTestingPrinterProviderAPI);
371 profile_ = profile_builder.Build();
372
442 extension_printer_handler_.reset(new ExtensionPrinterHandler( 373 extension_printer_handler_.reset(new ExtensionPrinterHandler(
443 env_.profile(), base::MessageLoop::current()->task_runner())); 374 profile_.get(), base::MessageLoop::current()->task_runner()));
444 375
445 pwg_raster_converter_ = new FakePWGRasterConverter(); 376 pwg_raster_converter_ = new FakePWGRasterConverter();
446 extension_printer_handler_->SetPwgRasterConverterForTesting( 377 extension_printer_handler_->SetPwgRasterConverterForTesting(
447 scoped_ptr<PWGRasterConverter>(pwg_raster_converter_)); 378 scoped_ptr<PWGRasterConverter>(pwg_raster_converter_));
448 device_client_.set_usb_service(&usb_service_);
449 } 379 }
450 380
451 protected: 381 protected:
452 FakePrinterProviderAPI* GetPrinterProviderAPI() { 382 FakePrinterProviderAPI* GetPrinterProviderAPI() {
453 return static_cast<FakePrinterProviderAPI*>( 383 return static_cast<FakePrinterProviderAPI*>(
454 extensions::PrinterProviderAPIFactory::GetInstance() 384 extensions::PrinterProviderAPIFactory::GetInstance()
455 ->GetForBrowserContext(env_.profile())); 385 ->GetForBrowserContext(profile_.get()));
456 } 386 }
457 387
458 MockUsbService usb_service_;
459 TestExtensionEnvironment env_;
460 scoped_ptr<ExtensionPrinterHandler> extension_printer_handler_; 388 scoped_ptr<ExtensionPrinterHandler> extension_printer_handler_;
461 389
462 FakePWGRasterConverter* pwg_raster_converter_; 390 FakePWGRasterConverter* pwg_raster_converter_;
463 391
464 private: 392 private:
465 FakeDeviceClient device_client_; 393 content::TestBrowserThreadBundle thread_bundle_;
394
395 scoped_ptr<TestingProfile> profile_;
466 396
467 DISALLOW_COPY_AND_ASSIGN(ExtensionPrinterHandlerTest); 397 DISALLOW_COPY_AND_ASSIGN(ExtensionPrinterHandlerTest);
468 }; 398 };
469 399
470 TEST_F(ExtensionPrinterHandlerTest, GetPrinters) { 400 TEST_F(ExtensionPrinterHandlerTest, GetPrinters) {
471 size_t call_count = 0; 401 size_t call_count = 0;
472 scoped_ptr<base::ListValue> printers; 402 scoped_ptr<base::ListValue> printers;
473 bool is_done = false; 403 bool is_done = false;
474 404
475 extension_printer_handler_->StartGetPrinters( 405 extension_printer_handler_->StartGetPrinters(
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after
512 std::string error; 442 std::string error;
513 scoped_ptr<base::ListValue> original_printers( 443 scoped_ptr<base::ListValue> original_printers(
514 GetJSONAsListValue(kPrinterDescriptionList, &error)); 444 GetJSONAsListValue(kPrinterDescriptionList, &error));
515 ASSERT_TRUE(original_printers) << "Error deserializing printers: " << error; 445 ASSERT_TRUE(original_printers) << "Error deserializing printers: " << error;
516 446
517 fake_api->TriggerNextGetPrintersCallback(*original_printers, true); 447 fake_api->TriggerNextGetPrintersCallback(*original_printers, true);
518 448
519 EXPECT_EQ(0u, call_count); 449 EXPECT_EQ(0u, call_count);
520 } 450 }
521 451
522 TEST_F(ExtensionPrinterHandlerTest, GetUsbPrinters) {
523 scoped_refptr<MockUsbDevice> device0 =
524 new MockUsbDevice(0, 0, "Google", "USB Printer", "");
525 usb_service_.AddDevice(device0);
526 scoped_refptr<MockUsbDevice> device1 =
527 new MockUsbDevice(0, 1, "Google", "USB Printer", "");
528 usb_service_.AddDevice(device1);
529
530 const Extension* extension_1 = env_.MakeExtension(
531 *base::test::ParseJson(kExtension1), "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
532 const Extension* extension_2 = env_.MakeExtension(
533 *base::test::ParseJson(kExtension2), "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb");
534
535 extensions::DevicePermissionsManager* permissions_manager =
536 extensions::DevicePermissionsManager::Get(env_.profile());
537 permissions_manager->AllowUsbDevice(extension_2->id(), device0);
538
539 size_t call_count = 0;
540 scoped_ptr<base::ListValue> printers;
541 bool is_done = false;
542 extension_printer_handler_->StartGetPrinters(
543 base::Bind(&RecordPrinterList, &call_count, &printers, &is_done));
544
545 FakePrinterProviderAPI* fake_api = GetPrinterProviderAPI();
546 ASSERT_TRUE(fake_api);
547 ASSERT_EQ(1u, fake_api->pending_get_printers_count());
548
549 EXPECT_EQ(1u, call_count);
550 EXPECT_FALSE(is_done);
551 EXPECT_TRUE(printers.get());
552 EXPECT_EQ(2u, printers->GetSize());
553 scoped_ptr<base::DictionaryValue> extension_1_entry(
554 DictionaryBuilder()
555 .Set("id", base::StringPrintf("provisional-usb:%s:%u",
556 extension_1->id().c_str(),
557 device0->unique_id()))
558 .Set("name", "USB Printer")
559 .Set("extensionName", "Provider 1")
560 .Set("extensionId", extension_1->id())
561 .Set("provisional", true)
562 .Build());
563 scoped_ptr<base::DictionaryValue> extension_2_entry(
564 DictionaryBuilder()
565 .Set("id", base::StringPrintf("provisional-usb:%s:%u",
566 extension_2->id().c_str(),
567 device1->unique_id()))
568 .Set("name", "USB Printer")
569 .Set("extensionName", "Provider 2")
570 .Set("extensionId", extension_2->id())
571 .Set("provisional", true)
572 .Build());
573 EXPECT_TRUE(printers->Find(*extension_1_entry) != printers->end());
574 EXPECT_TRUE(printers->Find(*extension_2_entry) != printers->end());
575
576 fake_api->TriggerNextGetPrintersCallback(base::ListValue(), true);
577
578 EXPECT_EQ(2u, call_count);
579 EXPECT_TRUE(is_done);
580 EXPECT_TRUE(printers.get());
581 EXPECT_EQ(0u, printers->GetSize()); // RecordPrinterList resets |printers|.
582 }
583
584 TEST_F(ExtensionPrinterHandlerTest, GetCapability) { 452 TEST_F(ExtensionPrinterHandlerTest, GetCapability) {
585 size_t call_count = 0; 453 size_t call_count = 0;
586 std::string destination_id; 454 std::string destination_id;
587 scoped_ptr<base::DictionaryValue> capability; 455 scoped_ptr<base::DictionaryValue> capability;
588 456
589 extension_printer_handler_->StartGetCapability( 457 extension_printer_handler_->StartGetCapability(
590 kPrinterId, 458 kPrinterId,
591 base::Bind(&RecordCapability, &call_count, &destination_id, &capability)); 459 base::Bind(&RecordCapability, &call_count, &destination_id, &capability));
592 460
593 EXPECT_EQ(0u, call_count); 461 EXPECT_EQ(0u, call_count);
(...skipping 325 matching lines...) Expand 10 before | Expand all | Expand 10 after
919 extension_printer_handler_->StartPrint( 787 extension_printer_handler_->StartPrint(
920 kPrinterId, kPWGRasterOnlyPrinterSimpleDescription, title, 788 kPrinterId, kPWGRasterOnlyPrinterSimpleDescription, title,
921 kEmptyPrintTicket, gfx::Size(100, 100), print_data, 789 kEmptyPrintTicket, gfx::Size(100, 100), print_data,
922 base::Bind(&RecordPrintResult, &call_count, &success, &status)); 790 base::Bind(&RecordPrintResult, &call_count, &success, &status));
923 791
924 EXPECT_EQ(1u, call_count); 792 EXPECT_EQ(1u, call_count);
925 793
926 EXPECT_FALSE(success); 794 EXPECT_FALSE(success);
927 EXPECT_EQ("INVALID_DATA", status); 795 EXPECT_EQ("INVALID_DATA", status);
928 } 796 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698