[EasyUnlock] Port PermitMessage class to native code.

components/proximity_auth/connection_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/proximity_auth/connection.h"
 
 #include "components/proximity_auth/connection_observer.h"
 #include "components/proximity_auth/wire_message.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
+using testing::DoAll;
 using testing::NiceMock;
 using testing::Return;
+using testing::SetArgPointee;
 using testing::StrictMock;
 
 namespace proximity_auth {
 namespace {
 
 class MockConnection : public Connection {
  public:
   MockConnection() : Connection(RemoteDevice()) {}
   ~MockConnection() {}
 
   MOCK_METHOD1(SetPaused, void(bool paused));
   MOCK_METHOD0(Connect, void());
   MOCK_METHOD0(Disconnect, void());
   MOCK_METHOD0(CancelConnectionAttempt, void());
   MOCK_METHOD1(SendMessageImplProxy, void(WireMessage* message));
-  MOCK_METHOD0(HasReceivedCompleteMessage, bool());
-  MOCK_METHOD0(DeserializeWireMessageProxy, WireMessage*());
+  MOCK_METHOD1(DeserializeWireMessageProxy,
+               WireMessage*(bool* is_incomplete_message));
 
   // Gmock only supports copyable types, so create simple wrapper methods for
   // ease of mocking.
   virtual void SendMessageImpl(scoped_ptr<WireMessage> message) OVERRIDE {
     SendMessageImplProxy(message.get());
   }
 
-  virtual scoped_ptr<WireMessage> DeserializeWireMessage() OVERRIDE {
-    return make_scoped_ptr(DeserializeWireMessageProxy());
+  virtual scoped_ptr<WireMessage> DeserializeWireMessage(
+      bool* is_incomplete_message) OVERRIDE {
+    return make_scoped_ptr(DeserializeWireMessageProxy(is_incomplete_message));
   }
 
   using Connection::status;
   using Connection::SetStatus;
   using Connection::OnDidSendMessage;
   using Connection::OnBytesReceived;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockConnection);
 };
@@ skipping 14 matching lines @@
                     const WireMessage& message,
                     bool success));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockConnectionObserver);
 };
 
 // Unlike WireMessage, offers a public constructor.
 class TestWireMessage : public WireMessage {
  public:
-  TestWireMessage() {}
+  TestWireMessage() : WireMessage(std::string(), std::string()) {}
   virtual ~TestWireMessage() {}
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestWireMessage);
 };
 
 }  // namespace
 
 TEST(ProximityAuthConnectionTest, IsConnected) {
   StrictMock<MockConnection> connection;
@@ skipping 98 matching lines @@
 }
 
 TEST(ProximityAuthConnectionTest,
      OnBytesReceived_NotifiesObserversOnValidMessage) {
   NiceMock<MockConnection> connection;
   connection.SetStatus(Connection::CONNECTED);
 
   StrictMock<MockConnectionObserver> observer;
   connection.AddObserver(&observer);
 
-  ON_CALL(connection, HasReceivedCompleteMessage()).WillByDefault(Return(true));
-  ON_CALL(connection, DeserializeWireMessageProxy())
-      .WillByDefault(Return(new TestWireMessage));
+  ON_CALL(connection, DeserializeWireMessageProxy(_))
+      .WillByDefault(DoAll(SetArgPointee<0>(false),
+                           Return(new TestWireMessage)));
   EXPECT_CALL(observer, OnMessageReceived(Ref(connection), _));
   connection.OnBytesReceived(std::string());
 }
 
 TEST(ProximityAuthConnectionTest,
      OnBytesReceived_DoesntNotifyObserversIfNotConnected) {
   StrictMock<MockConnection> connection;
   connection.SetStatus(Connection::IN_PROGRESS);
 
   StrictMock<MockConnectionObserver> observer;
   connection.AddObserver(&observer);
 
   EXPECT_CALL(observer, OnMessageReceived(_, _)).Times(0);
   connection.OnBytesReceived(std::string());
 }
 
 TEST(ProximityAuthConnectionTest,
      OnBytesReceived_DoesntNotifyObserversIfMessageIsIncomplete) {
   NiceMock<MockConnection> connection;
   connection.SetStatus(Connection::CONNECTED);
 
   StrictMock<MockConnectionObserver> observer;
   connection.AddObserver(&observer);
 
-  ON_CALL(connection, HasReceivedCompleteMessage())
-      .WillByDefault(Return(false));
+  ON_CALL(connection, DeserializeWireMessageProxy(_))
+      .WillByDefault(DoAll(SetArgPointee<0>(true),
+                           Return(static_cast<WireMessage*>(NULL))));
   EXPECT_CALL(observer, OnMessageReceived(_, _)).Times(0);
   connection.OnBytesReceived(std::string());
 }
 
 TEST(ProximityAuthConnectionTest,
      OnBytesReceived_DoesntNotifyObserversIfMessageIsInvalid) {
   NiceMock<MockConnection> connection;
   connection.SetStatus(Connection::CONNECTED);
 
   StrictMock<MockConnectionObserver> observer;
   connection.AddObserver(&observer);
 
-  ON_CALL(connection, HasReceivedCompleteMessage()).WillByDefault(Return(true));
-  ON_CALL(connection, DeserializeWireMessageProxy())
-      .WillByDefault(Return(static_cast<WireMessage*>(NULL)));
+  ON_CALL(connection, DeserializeWireMessageProxy(_))
+      .WillByDefault(DoAll(SetArgPointee<0>(false),
+                           Return(static_cast<WireMessage*>(NULL))));
   EXPECT_CALL(observer, OnMessageReceived(_, _)).Times(0);
   connection.OnBytesReceived(std::string());
 }
 
 }  // namespace proximity_auth