Index: remoting/host/desktop_process_unittest.cc |
diff --git a/remoting/host/desktop_process_unittest.cc b/remoting/host/desktop_process_unittest.cc |
index bc509882c1ad8e249544cdaae6d5dc7d4370dba2..0c019be9fe345bad8c5fea8e03508b203d198484 100644 |
--- a/remoting/host/desktop_process_unittest.cc |
+++ b/remoting/host/desktop_process_unittest.cc |
@@ -30,6 +30,7 @@ |
using testing::_; |
using testing::AnyNumber; |
+using testing::AtMost; |
using testing::InSequence; |
using testing::Return; |
@@ -198,13 +199,17 @@ DesktopEnvironment* DesktopProcessTest::CreateDesktopEnvironment() { |
EXPECT_CALL(*desktop_environment, CreateAudioCapturerPtr()) |
.Times(0); |
EXPECT_CALL(*desktop_environment, CreateInputInjectorPtr()) |
- .Times(AnyNumber()) |
- .WillRepeatedly(Invoke(this, &DesktopProcessTest::CreateInputInjector)); |
+ .Times(AtMost(1)) |
+ .WillOnce(Invoke(this, &DesktopProcessTest::CreateInputInjector)); |
EXPECT_CALL(*desktop_environment, CreateScreenControlsPtr()) |
- .Times(AnyNumber()); |
+ .Times(AtMost(1)); |
EXPECT_CALL(*desktop_environment, CreateVideoCapturerPtr()) |
- .Times(AnyNumber()) |
- .WillRepeatedly(Invoke(this, &DesktopProcessTest::CreateVideoCapturer)); |
+ .Times(AtMost(1)) |
+ .WillOnce(Invoke(this, &DesktopProcessTest::CreateVideoCapturer)); |
+ EXPECT_CALL(*desktop_environment, GetCapabilities()) |
+ .Times(AtMost(1)); |
+ EXPECT_CALL(*desktop_environment, SetCapabilities(_)) |
+ .Times(AtMost(1)); |
// Notify the test that the desktop environment has been created. |
network_listener_.OnDesktopEnvironmentCreated(); |