summaryrefslogtreecommitdiffstats
path: root/media/video/capture/screen/shared_buffer_unittest.cc
blob: 1efc798459dc8b502c9a2115470677630a1ddbad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// Copyright (c) 2012 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 "base/process_util.h"
#include "media/video/capture/screen/shared_buffer.h"
#include "testing/gtest/include/gtest/gtest.h"

const uint32 kBufferSize = 4096;
const int kPattern = 0x12345678;

const int kIdZero = 0;
const int kIdOne = 1;

namespace media {

TEST(SharedBufferTest, Basic) {
  scoped_refptr<SharedBuffer> source(new SharedBuffer(kBufferSize));

  // Make sure that the buffer is allocated, the size is recorded correctly and
  // its ID is reset.
  EXPECT_TRUE(source->ptr() != NULL);
  EXPECT_EQ(source->id(), kIdZero);
  EXPECT_EQ(source->size(), kBufferSize);

  // See if setting of the ID works.
  source->set_id(kIdOne);
  EXPECT_EQ(source->id(), kIdOne);

  base::SharedMemoryHandle copied_handle;
  EXPECT_TRUE(source->ShareToProcess(
      base::GetCurrentProcessHandle(), &copied_handle));

  scoped_refptr<SharedBuffer> dest(
      new SharedBuffer(kIdZero, copied_handle, kBufferSize));

  // Make sure that the buffer is allocated, the size is recorded correctly and
  // its ID is reset.
  EXPECT_TRUE(dest->ptr() != NULL);
  EXPECT_EQ(dest->id(), kIdZero);
  EXPECT_EQ(dest->size(), kBufferSize);

  // Verify that the memory contents are the same for the two buffers.
  int* source_ptr = reinterpret_cast<int*>(source->ptr());
  *source_ptr = kPattern;
  int* dest_ptr = reinterpret_cast<int*>(dest->ptr());
  EXPECT_EQ(*source_ptr, *dest_ptr);

  // Check that the destination buffer is still mapped even when the source
  // buffer is destroyed.
  source = NULL;
  EXPECT_EQ(0x12345678, *dest_ptr);
}

}  // namespace media