Index: media/capture/video/video_capture_device_unittest.cc |
diff --git a/media/capture/video/video_capture_device_unittest.cc b/media/capture/video/video_capture_device_unittest.cc |
index f752360a1c34c96c012bac03492825fef38318de..8cc2496d7b8e04f533b75c54bb1290fc59f3aaa2 100644 |
--- a/media/capture/video/video_capture_device_unittest.cc |
+++ b/media/capture/video/video_capture_device_unittest.cc |
@@ -63,7 +63,7 @@ |
// http://crbug.com/94134 http://crbug.com/137260 http://crbug.com/417824 |
#define MAYBE_AllocateBadSize DISABLED_AllocateBadSize |
#define MAYBE_CaptureMjpeg CaptureMjpeg |
-#define MAYBE_TakePhoto DISABLED_TakePhoto |
+#define MAYBE_TakePhoto TakePhoto |
#else |
#define MAYBE_AllocateBadSize AllocateBadSize |
#define MAYBE_CaptureMjpeg CaptureMjpeg |
@@ -71,6 +71,7 @@ |
#endif |
using ::testing::_; |
+using ::testing::AnyOf; |
mcasas
2016/09/16 03:50:37
Not used, remove.
xianglu
2016/09/16 18:22:34
Done.
|
using ::testing::SaveArg; |
namespace media { |
@@ -141,15 +142,24 @@ class MockImageCaptureClient : public base::RefCounted<MockImageCaptureClient> { |
public: |
// GMock doesn't support move-only arguments, so we use this forward method. |
void DoOnPhotoTaken(mojom::BlobPtr blob) { |
- EXPECT_STREQ("image/jpeg", blob->mime_type.c_str()); |
- ASSERT_GT(blob->data.size(), 4u); |
- // Check some bytes that univocally identify |data| as a JPEG File. |
- // https://en.wikipedia.org/wiki/JPEG_File_Interchange_Format#File_format_structure |
- EXPECT_EQ(0xFF, blob->data[0]); // First SOI byte |
- EXPECT_EQ(0xD8, blob->data[1]); // Second SOI byte |
- EXPECT_EQ(0xFF, blob->data[2]); // First JFIF-APP0 byte |
- EXPECT_EQ(0xE0, blob->data[3]); // Second JFIF-APP0 byte |
- OnCorrectPhotoTaken(); |
+ if (strcmp("image/jpeg", blob->mime_type.c_str()) == 0) { |
+ ASSERT_GT(blob->data.size(), 4u); |
+ // Check some bytes that univocally identify |data| as a JPEG File. |
+ // https://en.wikipedia.org/wiki/JPEG_File_Interchange_Format#File_format_structure |
+ EXPECT_EQ(0xFF, blob->data[0]); // First SOI byte |
+ EXPECT_EQ(0xD8, blob->data[1]); // Second SOI byte |
+ EXPECT_EQ(0xFF, blob->data[2]); // First JFIF-APP0 byte |
+ EXPECT_EQ(0xE0, blob->data[3]); // Second JFIF-APP0 byte |
+ OnCorrectPhotoTaken(); |
+ } else if (strcmp("image/png", blob->mime_type.c_str()) == 0) { |
+ ASSERT_GT(blob->data.size(), 4u); |
+ EXPECT_EQ('P', blob->data[1]); |
+ EXPECT_EQ('N', blob->data[2]); |
+ EXPECT_EQ('G', blob->data[3]); |
+ OnCorrectPhotoTaken(); |
+ } else { |
+ NOTREACHED() << "Photo format should be jpeg or png"; |
mcasas
2016/09/16 03:50:37
NOTREACHED() would crash the test executable and
p
xianglu
2016/09/16 18:22:34
Done.
|
+ } |
} |
MOCK_METHOD0(OnCorrectPhotoTaken, void(void)); |
MOCK_METHOD1(OnTakePhotoFailure, |