| Index: chrome/browser/safe_browsing/client_side_detection_host_unittest.cc
|
| diff --git a/chrome/browser/safe_browsing/client_side_detection_host_unittest.cc b/chrome/browser/safe_browsing/client_side_detection_host_unittest.cc
|
| index cc60e17d3847a72ac54ed79809c057c56f3fe484..b095af20e53e26c3bb048506923529a83a556976 100644
|
| --- a/chrome/browser/safe_browsing/client_side_detection_host_unittest.cc
|
| +++ b/chrome/browser/safe_browsing/client_side_detection_host_unittest.cc
|
| @@ -91,10 +91,12 @@
|
| base::MessageLoopForUI::current()->Quit();
|
| }
|
|
|
| -ACTION_P(InvokeDoneCallback, verdict) {
|
| - scoped_ptr<ClientPhishingRequest> request(::std::tr1::get<1>(args));
|
| - request->CopyFrom(*verdict);
|
| - ::std::tr1::get<2>(args).Run(true, request.Pass());
|
| +// It's kind of insane that InvokeArgument doesn't work with callbacks, but it
|
| +// doesn't seem like it.
|
| +ACTION_TEMPLATE(InvokeCallbackArgument,
|
| + HAS_1_TEMPLATE_PARAMS(int, k),
|
| + AND_2_VALUE_PARAMS(p0, p1)) {
|
| + ::std::tr1::get<k>(args).Run(p0, p1);
|
| }
|
|
|
| ACTION_P(InvokeMalwareCallback, verdict) {
|
| @@ -480,7 +482,8 @@
|
| verdict.set_is_phishing(true);
|
|
|
| EXPECT_CALL(*mock_extractor, ExtractFeatures(_, _, _))
|
| - .WillOnce(InvokeDoneCallback(&verdict));
|
| + .WillOnce(DoAll(DeleteArg<1>(),
|
| + InvokeCallbackArgument<2>(true, &verdict)));
|
| EXPECT_CALL(*csd_service_,
|
| SendClientReportPhishingRequest(
|
| Pointee(PartiallyEqualVerdict(verdict)), _))
|
| @@ -512,7 +515,8 @@
|
| verdict.set_is_phishing(true);
|
|
|
| EXPECT_CALL(*mock_extractor, ExtractFeatures(_, _, _))
|
| - .WillOnce(InvokeDoneCallback(&verdict));
|
| + .WillOnce(DoAll(DeleteArg<1>(),
|
| + InvokeCallbackArgument<2>(true, &verdict)));
|
| EXPECT_CALL(*csd_service_,
|
| SendClientReportPhishingRequest(
|
| Pointee(PartiallyEqualVerdict(verdict)), _))
|
| @@ -545,7 +549,8 @@
|
| verdict.set_is_phishing(true);
|
|
|
| EXPECT_CALL(*mock_extractor, ExtractFeatures(_, _, _))
|
| - .WillOnce(InvokeDoneCallback(&verdict));
|
| + .WillOnce(DoAll(DeleteArg<1>(),
|
| + InvokeCallbackArgument<2>(true, &verdict)));
|
| EXPECT_CALL(*csd_service_,
|
| SendClientReportPhishingRequest(
|
| Pointee(PartiallyEqualVerdict(verdict)), _))
|
| @@ -599,7 +604,8 @@
|
| verdict.set_is_phishing(true);
|
|
|
| EXPECT_CALL(*mock_extractor, ExtractFeatures(_, _, _))
|
| - .WillOnce(InvokeDoneCallback(&verdict));
|
| + .WillOnce(DoAll(DeleteArg<1>(),
|
| + InvokeCallbackArgument<2>(true, &verdict)));
|
| EXPECT_CALL(*csd_service_,
|
| SendClientReportPhishingRequest(
|
| Pointee(PartiallyEqualVerdict(verdict)), _))
|
|
|