Add AssignmentSource to BlimpClientContextImpl

blimp/client/test/test_blimp_client_context_delegate.h

 // Copyright 2016 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.
 
 #ifndef BLIMP_CLIENT_TEST_TEST_BLIMP_CLIENT_CONTEXT_DELEGATE_H_
 #define BLIMP_CLIENT_TEST_TEST_BLIMP_CLIENT_CONTEXT_DELEGATE_H_
 
 #include "base/macros.h"
 #include "blimp/client/public/blimp_client_context_delegate.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace blimp {
 namespace client {
 class BlimpContents;
 
 // Helper class to use in tests when a BlimpClientContextDelegate is needed.
 class TestBlimpClientContextDelegate : public BlimpClientContextDelegate {
  public:
   TestBlimpClientContextDelegate();
   ~TestBlimpClientContextDelegate() override;
 
   // BlimpClientContextDelegate implementation.
   MOCK_METHOD1(AttachBlimpContentsHelpers, void(BlimpContents*));
+  MOCK_METHOD2(OnAssignmentConnectionAttempted,
+               void(AssignmentRequestResult, const Assignment&));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestBlimpClientContextDelegate);
 };
 
 }  // namespace client
 }  // namespace blimp
 
 #endif  // BLIMP_CLIENT_TEST_TEST_BLIMP_CLIENT_CONTEXT_DELEGATE_H_