mojo/common/message_pump_mojo_unittest.cc - Issue 1262173005: Straightens outs DEPS in mojo/common

Side by Side Diff: mojo/common/message_pump_mojo_unittest.cc

Issue 1262173005: Straightens outs DEPS in mojo/common (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master

Patch Set: add dep on system_for_component Created 5 years, 4 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

OLD	NEW
	(Empty)
1 // Copyright 2013 The Chromium Authors. All rights reserved.

2 // Use of this source code is governed by a BSD-style license that can be

3 // found in the LICENSE file.

4

5 #include "mojo/common/message_pump_mojo.h"

6

7 #include "base/message_loop/message_loop_test.h"

8 #include "base/run_loop.h"

9 #include "mojo/common/message_pump_mojo_handler.h"

10 #include "testing/gtest/include/gtest/gtest.h"

11 #include "third_party/mojo/src/mojo/public/cpp/system/core.h"

12

13 namespace mojo {

14 namespace common {

15 namespace test {

16

17 scoped_ptr<base::MessagePump> CreateMojoMessagePump() {

18 return scoped_ptr<base::MessagePump>(new MessagePumpMojo());

19 }

20

21 RUN_MESSAGE_LOOP_TESTS(Mojo, &CreateMojoMessagePump);

22

23 class CountingMojoHandler : public MessagePumpMojoHandler {

24 public:

25 CountingMojoHandler() : success_count_(0), error_count_(0) {}

26

27 void OnHandleReady(const Handle& handle) override {

28 ReadMessageRaw(static_cast<const MessagePipeHandle&>(handle),

29 NULL,

30 NULL,

31 NULL,

32 NULL,

33 MOJO_READ_MESSAGE_FLAG_NONE);

34 ++success_count_;

35 }

36 void OnHandleError(const Handle& handle, MojoResult result) override {

37 ++error_count_;

38 }

39

40 int success_count() { return success_count_; }

41 int error_count() { return error_count_; }

42

43 private:

44 int success_count_;

45 int error_count_;

46

47 DISALLOW_COPY_AND_ASSIGN(CountingMojoHandler);

48 };

49

50 class CountingObserver : public MessagePumpMojo::Observer {

51 public:

52 void WillSignalHandler() override { will_signal_handler_count++; }

53 void DidSignalHandler() override { did_signal_handler_count++; }

54

55 int will_signal_handler_count = 0;

56 int did_signal_handler_count = 0;

57 };

58

59 TEST(MessagePumpMojo, RunUntilIdle) {

60 base::MessageLoop message_loop(MessagePumpMojo::Create());

61 CountingMojoHandler handler;

62 MessagePipe handles;

63 MessagePumpMojo::current()->AddHandler(&handler,

64 handles.handle0.get(),

65 MOJO_HANDLE_SIGNAL_READABLE,

66 base::TimeTicks());

67 WriteMessageRaw(

68 handles.handle1.get(), NULL, 0, NULL, 0, MOJO_WRITE_MESSAGE_FLAG_NONE);

69 WriteMessageRaw(

70 handles.handle1.get(), NULL, 0, NULL, 0, MOJO_WRITE_MESSAGE_FLAG_NONE);

71 base::RunLoop run_loop;

72 run_loop.RunUntilIdle();

73 EXPECT_EQ(2, handler.success_count());

74 }

75

76 TEST(MessagePumpMojo, Observer) {

77 base::MessageLoop message_loop(MessagePumpMojo::Create());

78

79 CountingObserver observer;

80 MessagePumpMojo::current()->AddObserver(&observer);

81

82 CountingMojoHandler handler;

83 MessagePipe handles;

84 MessagePumpMojo::current()->AddHandler(&handler,

85 handles.handle0.get(),

86 MOJO_HANDLE_SIGNAL_READABLE,

87 base::TimeTicks());

88 WriteMessageRaw(

89 handles.handle1.get(), NULL, 0, NULL, 0, MOJO_WRITE_MESSAGE_FLAG_NONE);

90 base::RunLoop run_loop;

91 run_loop.RunUntilIdle();

92 EXPECT_EQ(1, handler.success_count());

93 EXPECT_EQ(1, observer.will_signal_handler_count);

94 EXPECT_EQ(1, observer.did_signal_handler_count);

95 MessagePumpMojo::current()->RemoveObserver(&observer);

96

97 WriteMessageRaw(

98 handles.handle1.get(), NULL, 0, NULL, 0, MOJO_WRITE_MESSAGE_FLAG_NONE);

99 base::RunLoop run_loop2;

100 run_loop2.RunUntilIdle();

101 EXPECT_EQ(2, handler.success_count());

102 EXPECT_EQ(1, observer.will_signal_handler_count);

103 EXPECT_EQ(1, observer.did_signal_handler_count);

104 }

105

106 TEST(MessagePumpMojo, UnregisterAfterDeadline) {

107 base::MessageLoop message_loop(MessagePumpMojo::Create());

108 CountingMojoHandler handler;

109 MessagePipe handles;

110 MessagePumpMojo::current()->AddHandler(

111 &handler,

112 handles.handle0.get(),

113 MOJO_HANDLE_SIGNAL_READABLE,

114 base::TimeTicks::Now() - base::TimeDelta::FromSeconds(1));

115 for (int i = 0; i < 2; ++i) {

116 base::RunLoop run_loop;

117 run_loop.RunUntilIdle();

118 }

119 EXPECT_EQ(1, handler.error_count());

120 }

121

122 } // namespace test

123 } // namespace common

124 } // namespace mojo

OLD	NEW

« no previous file with comments | « mojo/common/message_pump_mojo_handler.h ('k') | mojo/common/time_helper.h » ('j') | no next file with comments »