Add CDM FileIO tests.

media/cdm/ppapi/clear_key_cdm.cc

 // Copyright 2013 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 "media/cdm/ppapi/clear_key_cdm.h"
 
 #include <algorithm>
 #include <sstream>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/debug/trace_event.h"
 #include "base/logging.h"
 #include "base/time/time.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decrypt_config.h"
+#include "media/cdm/ppapi/cdm_file_io_test.h"
 #include "media/cdm/ppapi/cdm_video_decoder.h"
 
 #if defined(CLEAR_KEY_CDM_USE_FAKE_AUDIO_DECODER)
 #include "base/basictypes.h"
 const int64 kNoTimestamp = kint64min;
 #endif  // CLEAR_KEY_CDM_USE_FAKE_AUDIO_DECODER
 
 #if defined(CLEAR_KEY_CDM_USE_FFMPEG_DECODER)
 #include "base/at_exit.h"
 #include "base/files/file_path.h"
@@ skipping 35 matching lines @@
 const char kExternalClearKeyKeySystem[] = "org.chromium.externalclearkey";
 const char kExternalClearKeyDecryptOnlyKeySystem[] =
     "org.chromium.externalclearkey.decryptonly";
 const int64 kSecondsPerMinute = 60;
 const int64 kMsPerSecond = 1000;
 const int64 kInitialTimerDelayMs = 200;
 const int64 kMaxTimerDelayMs = 1 * kSecondsPerMinute * kMsPerSecond;
 // Heart beat message header. If a key message starts with |kHeartBeatHeader|,
 // it's a heart beat message. Otherwise, it's a key request.
 const char kHeartBeatHeader[] = "HEARTBEAT";
+// CDM file IO test result header.
+const char kFileIOTestResultHeader[] = "FILEIOTESTRESULT";
 
 // Copies |input_buffer| into a media::DecoderBuffer. If the |input_buffer| is
 // empty, an empty (end-of-stream) media::DecoderBuffer is returned.
 static scoped_refptr<media::DecoderBuffer> CopyDecoderBufferFrom(
     const cdm::InputBuffer& input_buffer) {
   if (!input_buffer.data) {
     DCHECK(!input_buffer.data_size);
     return media::DecoderBuffer::CreateEOSBuffer();
   }
 
@@ skipping 17 matching lines @@
       input_buffer.data_offset,
       subsamples));
 
   output_buffer->set_decrypt_config(decrypt_config.Pass());
   output_buffer->set_timestamp(
       base::TimeDelta::FromMicroseconds(input_buffer.timestamp));
 
   return output_buffer;
 }
 
+static std::string GetFileIOTestResultMessage(bool success) {
+  std::string message(kFileIOTestResultHeader);
+  message += success ? '1' : '0';
+  return message;
+}
+
 template<typename Type>
 class ScopedResetter {
  public:
   explicit ScopedResetter(Type* object) : object_(object) {}
   ~ScopedResetter() { object_->Reset(); }
 
  private:
   Type* const object_;
 };
 
@@ skipping 122 matching lines @@
 
   if (client_.status() != (Client::kMessage | Client::kCreated)) {
     // Use values returned to client if possible.
     host_->SendKeyError(client_.session_id().data(),
                         client_.session_id().size(),
                         static_cast<cdm::MediaKeyError>(client_.error_code()),
                         client_.system_code());
     return cdm::kSessionError;
   }
 
+  std::string session_id = client_.session_id();
+
   host_->SendKeyMessage(
-      client_.session_id().data(), client_.session_id().size(),
+      session_id.data(), session_id.size(),
       reinterpret_cast<const char*>(&client_.message()[0]),
       client_.message().size(),
       client_.destination_url().data(), client_.destination_url().size());
 
-  // Only save the latest session ID for heartbeat messages.
-  heartbeat_session_id_ = client_.session_id();
+  // Save the latest session ID for heartbeat and file IO test messages.
+  last_session_id_ = client_.session_id();
+
+  StartCdmFileIOTest();
 
   return cdm::kSuccess;
 }
 
 cdm::Status ClearKeyCdm::AddKey(const char* session_id,
                                 uint32_t session_id_size,
                                 const uint8_t* key,
                                 uint32_t key_size,
                                 const uint8_t* key_id,
                                 uint32_t key_id_size) {
@@ skipping 44 matching lines @@
     heartbeat_message = next_heartbeat_message_;
   } else {
     heartbeat_message = "ERROR: Invalid timer context found!";
   }
 
   // This URL is only used for testing the code path for defaultURL.
   // There is no service at this URL, so applications should ignore it.
   const char url[] = "http://test.externalclearkey.chromium.org";
 
   host_->SendKeyMessage(
-      heartbeat_session_id_.data(), heartbeat_session_id_.size(),
+      last_session_id_.data(), last_session_id_.size(),
       heartbeat_message.data(), heartbeat_message.size(),
       url, arraysize(url) - 1);
 
   ScheduleNextHeartBeat();
 }
 
 static void CopyDecryptResults(
     media::Decryptor::Status* status_copy,
     scoped_refptr<media::DecoderBuffer>* buffer_copy,
     media::Decryptor::Status status,
@@ skipping 279 matching lines @@
 
   int samples_generated = GenerateFakeAudioFramesFromDuration(
       timestamp_in_microseconds - CurrentTimeStampInMicroseconds(),
       audio_frames);
   total_samples_generated_ += samples_generated;
 
   return samples_generated == 0 ? cdm::kNeedMoreData : cdm::kSuccess;
 }
 #endif  // CLEAR_KEY_CDM_USE_FAKE_AUDIO_DECODER
 
+void ClearKeyCdm::StartCdmFileIOTest() {
+  cdm_file_io_test_.reset(new CdmFileIOTest(host_));

[ddorwin, 2013/12/11 21:16:17]

Doh. I'm the one that asked you to encapsulate FileIO creation. Hmm.

Possible solution:
Template a simple CdmFileIOTest class that just has a constructor and Run(). It
will also have a private CreateFileIo() function that it passes to the
constructor of the real test class (currently CdmFileIOTest), which it keeps as
a member (pimpl?).
That way, the test is not tied to CK, CK doesn't know about FileIO, and we don't
have a giant template class that is hard to debug. WDYT? (I'm okay doing this in
another CL.)

[xhwang, 2013/12/13 02:51:47]

Now I pass in a callback for CreateFileIO. But I don't understand the "CK
doesn't know about FIleIO" part. We need to talk.

+  cdm_file_io_test_->Run(
+      base::Bind(&ClearKeyCdm::OnFileIOTestComplete, base::Unretained(this)));
+}
+
+void ClearKeyCdm::OnFileIOTestComplete(bool success) {
+  DVLOG(1) << __FUNCTION__ << ": " << success;
+  std::string message = GetFileIOTestResultMessage(success);
+  host_->SendKeyMessage(last_session_id_.data(), last_session_id_.size(),
+                        message.data(), message.size(),
+                        NULL, 0);
+  cdm_file_io_test_.reset();
+}
+
 }  // namespace media