[ios] Adds support for parsing Text-to-Speech search results.

ios/chrome/browser/ui/voice/text_to_speech_player_unittest.mm

Index: ios/chrome/browser/ui/voice/text_to_speech_player_unittest.mm
diff --git a/ios/chrome/browser/ui/voice/text_to_speech_player_unittest.mm b/ios/chrome/browser/ui/voice/text_to_speech_player_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..002e5521b210e939c6fb9ef8127a824d33da9412
--- /dev/null
+++ b/ios/chrome/browser/ui/voice/text_to_speech_player_unittest.mm
@@ -0,0 +1,169 @@
+// 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.
+
+#import "ios/chrome/browser/ui/voice/text_to_speech_player.h"
+
+#import <UIKit/UIKit.h>
+
+#include "base/mac/foundation_util.h"
+#import "base/mac/scoped_nsobject.h"
+#import "base/test/ios/wait_util.h"
+#include "base/time/time.h"
+#import "ios/chrome/browser/ui/voice/voice_search_notification_names.h"
+#include "ios/web/public/test/test_web_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gtest_mac.h"
+#include "testing/platform_test.h"
+#include "url/gurl.h"
+
+#pragma mark - TTSPlayerObserver
+
+// Test object that listens for TTS notifications.
+@interface TTSPlayerObserver : NSObject
+
+// The TextToSpeechPlayer passed on initialization.
+@property(nonatomic, retain) TextToSpeechPlayer* player;
+
+// Whether notifications have been received.
+@property(nonatomic, readonly) BOOL readyNotificationReceived;
+@property(nonatomic, readonly) BOOL willStartNotificationReceived;
+@property(nonatomic, readonly) BOOL didStopNotificationReceived;
+
+// Notification handlers.
+- (void)handleReadyNotification:(NSNotification*)notification;
+- (void)handleWillStartNotification:(NSNotification*)notification;
+- (void)handleDidStopNotification:(NSNotification*)notification;
+
+@end
+
+@implementation TTSPlayerObserver {
+  base::scoped_nsobject<TextToSpeechPlayer> _player;
+}
+
+@synthesize readyNotificationReceived = _readyNotificationReceived;
+@synthesize willStartNotificationReceived = _willStartNotificationReceived;
+@synthesize didStopNotificationReceived = _didStopNotificationReceived;
+
+- (void)dealloc {
+  [[NSNotificationCenter defaultCenter] removeObserver:self];
+  [super dealloc];
+}
+
+- (void)setPlayer:(TextToSpeechPlayer*)player {
+  NSNotificationCenter* defaultCenter = [NSNotificationCenter defaultCenter];
+  [defaultCenter removeObserver:self];
+  _player.reset([player retain]);
+  if (player) {
+    [defaultCenter addObserver:self
+                      selector:@selector(handleReadyNotification:)
+                          name:kTTSAudioReadyForPlaybackNotification
+                        object:player];
+    [defaultCenter addObserver:self
+                      selector:@selector(handleWillStartNotification:)
+                          name:kTTSWillStartPlayingNotification
+                        object:player];
+    [defaultCenter addObserver:self
+                      selector:@selector(handleDidStopNotification:)
+                          name:kTTSDidStopPlayingNotification
+                        object:player];
+  }
+}
+
+- (TextToSpeechPlayer*)player {
+  return _player;
+}
+
+- (void)handleReadyNotification:(NSNotification*)notification {
+  ASSERT_EQ(notification.object, self.player);
+  _readyNotificationReceived = YES;
+}
+
+- (void)handleWillStartNotification:(NSNotification*)notification {
+  ASSERT_EQ(notification.object, self.player);
+  _willStartNotificationReceived = YES;
+}
+
+- (void)handleDidStopNotification:(NSNotification*)notification {
+  ASSERT_EQ(notification.object, self.player);
+  _didStopNotificationReceived = YES;
+}
+
+@end
+
+#pragma mark - TextToSpeechPlayerTest
+
+class TextToSpeechPlayerTest : public PlatformTest {
+ protected:
+  void SetUp() override {
+    tts_player_.reset([[TextToSpeechPlayer alloc] init]);
+    tts_player_observer_.reset([[TTSPlayerObserver alloc] init]);
+    [tts_player_observer_ setPlayer:tts_player_];
+  }
+
+  base::scoped_nsobject<TextToSpeechPlayer> tts_player_;
+  base::scoped_nsobject<TTSPlayerObserver> tts_player_observer_;
+  web::TestWebThreadBundle web_thread_bundle_;
+};
+
+// Tests that kTTSAudioReadyForPlaybackNotification is received and that
+// TTSPlayer state is updated.
+TEST_F(TextToSpeechPlayerTest, ReadyForPlayback) {
+  base::scoped_nsobject<NSData> audio_data(
+      [[@"audio_data" dataUsingEncoding:NSUTF8StringEncoding] retain]);
+  [tts_player_ prepareToPlayAudioData:audio_data];
+  EXPECT_TRUE([tts_player_observer_ readyNotificationReceived]);
+  EXPECT_TRUE([tts_player_ isReadyForPlayback]);
+}
+
+// Tests that kTTSAudioReadyForPlaybackNotification is received and that
+// TTSPlayer's |-readyForPlayback| is NO for empty data.
+TEST_F(TextToSpeechPlayerTest, ReadyForPlaybackEmtpyData) {
+  base::scoped_nsobject<NSData> audio_data(
+      [[@"" dataUsingEncoding:NSUTF8StringEncoding] retain]);
+  [tts_player_ prepareToPlayAudioData:audio_data];
+  EXPECT_TRUE([tts_player_observer_ readyNotificationReceived]);
+  EXPECT_FALSE([tts_player_ isReadyForPlayback]);
+}
+
+// Tests that kTTSWillStartPlayingNotification is received when playback begins
+// and kTTSDidStopPlayingNotification is received when it is cancelled.
+// TODO(rohitrao): Disabled because the bots do not have a valid sound output
+// device.
+TEST_F(TextToSpeechPlayerTest, DISABLED_ValidPlaybackNotifications) {
+  NSString* path =
+      [[NSBundle mainBundle] pathForResource:@"test_sound"
+                                      ofType:@"m4a"
+                                 inDirectory:@"ios/chrome/test/data/voice"];
+  base::scoped_nsobject<NSData> audio_data(
+      [[NSData alloc] initWithContentsOfFile:path]);
+  [tts_player_ prepareToPlayAudioData:audio_data];
+  [tts_player_ beginPlayback];
+  EXPECT_TRUE([tts_player_observer_ willStartNotificationReceived]);
+  EXPECT_TRUE([tts_player_ isPlayingAudio]);
+  [tts_player_ cancelPlayback];
+  EXPECT_TRUE([tts_player_observer_ didStopNotificationReceived]);
+  EXPECT_FALSE([tts_player_ isPlayingAudio]);
+}
+
+// Tests that playback is cancelled when the application enters the background
+// while playback is occurring.
+// TODO(rohitrao): Disabled because the bots do not have a valid sound output
+// device.
+TEST_F(TextToSpeechPlayerTest, DISABLED_BackgroundNotification) {
+  NSString* path =
+      [[NSBundle mainBundle] pathForResource:@"test_sound"
+                                      ofType:@"m4a"
+                                 inDirectory:@"ios/chrome/test/data/voice"];
+  base::scoped_nsobject<NSData> audio_data(
+      [[NSData alloc] initWithContentsOfFile:path]);
+  [tts_player_ prepareToPlayAudioData:audio_data];
+  [tts_player_ beginPlayback];
+  EXPECT_TRUE([tts_player_observer_ willStartNotificationReceived]);
+  EXPECT_TRUE([tts_player_ isPlayingAudio]);
+  [[NSNotificationCenter defaultCenter]
+      postNotificationName:UIApplicationDidEnterBackgroundNotification
+                    object:[UIApplication sharedApplication]];
+  EXPECT_TRUE([tts_player_observer_ didStopNotificationReceived]);
+  EXPECT_FALSE([tts_player_ isPlayingAudio]);
+}