chrome/test/media_router/test_media_sinks_observer.h

   1 // Copyright 2015 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 <map>
   6 #include <string>
   7 #include <vector>
   8
   9 #include "chrome/browser/media/router/media_sinks_observer.h"
  10
  11 #ifndef CHROME_TEST_MEDIA_ROUTER_TEST_MEDIA_SINKS_OBSERVER_H_
  12 #define CHROME_TEST_MEDIA_ROUTER_TEST_MEDIA_SINKS_OBSERVER_H_
  13
  14 namespace media_router {
  15
  16 class MediaRouter;
  17
  18 // Test class to implement MediaSinksObserver that receives SinkQueryResults
  19 // from Media Router and is used for verification.
  20 class TestMediaSinksObserver : public MediaSinksObserver {
  21  public:
  22   TestMediaSinksObserver(MediaRouter* router, const MediaSource& source);
  23   ~TestMediaSinksObserver() override;
  24
  25   // MediaSinksObserver implementation.
  26   void OnSinksReceived(const std::vector<MediaSink>& result) override;
  27
  28   // Map of <sink_name, media_sink_object>
  29   std::map<std::string, const MediaSink> sink_map;
  30 };
  31
  32 }  // namespace media_router
  33
  34 #endif  // CHROME_TEST_MEDIA_ROUTER_TEST_MEDIA_SINKS_OBSERVER_H_