shared_memory_arbiter_impl_unittest.cc

/*
 * Copyright (C) 2017 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "src/tracing/core/shared_memory_arbiter_impl.h"

#include <bitset>
#include "perfetto/ext/base/utils.h"
#include "perfetto/ext/tracing/core/basic_types.h"
#include "perfetto/ext/tracing/core/commit_data_request.h"
#include "perfetto/ext/tracing/core/shared_memory_abi.h"
#include "perfetto/ext/tracing/core/trace_packet.h"
#include "perfetto/ext/tracing/core/trace_writer.h"
#include "perfetto/ext/tracing/core/tracing_service.h"
#include "src/base/test/gtest_test_suite.h"
#include "src/base/test/test_task_runner.h"
#include "src/tracing/core/in_process_shared_memory.h"
#include "src/tracing/core/patch_list.h"
#include "src/tracing/test/aligned_buffer_test.h"
#include "src/tracing/test/mock_producer_endpoint.h"
#include "test/gtest_and_gmock.h"

#include "protos/perfetto/trace/test_event.pbzero.h"
#include "protos/perfetto/trace/trace_packet.pbzero.h"

namespace perfetto {

_;
Between;
Invoke;
Mock;
NiceMock;
UnorderedElementsAreArray;

ShmemMode;

class SharedMemoryArbiterImplTest : public AlignedBufferTest { … };

size_t const kPageSizes[] = …;
INSTANTIATE_TEST_SUITE_P(…);

// The buffer has 14 pages (kNumPages), each will be partitioned in 14 chunks.
// The test requests 30 chunks (2 full pages + 2 chunks from a 3rd page) and
// releases them in different batches. It tests the consistency of the batches
// and the releasing order.
TEST_P(SharedMemoryArbiterImplTest, GetAndReturnChunks) { … }

TEST_P(SharedMemoryArbiterImplTest, BatchCommits) { … }

TEST_P(SharedMemoryArbiterImplTest, UseShmemEmulation) { … }

// Check that we can actually create up to kMaxWriterID TraceWriter(s).
TEST_P(SharedMemoryArbiterImplTest, WriterIDsAllocation) { … }

TEST_P(SharedMemoryArbiterImplTest, Shutdown) { … }

// Verify that getting a new chunk doesn't stall when kDrop policy is chosen.
TEST_P(SharedMemoryArbiterImplTest, BufferExhaustedPolicyDrop) { … }

TEST_P(SharedMemoryArbiterImplTest, CreateUnboundAndBind) { … }

// Startup tracing tests are run with the arbiter in either bound or unbound
// initial state. Startup tracing in bound state can still be useful, e.g. in
// integration tests or to enable tracing in the consumer process immediately
// before/after instructing the service to start a session, avoiding the
// round-trip time through the service.
enum class InitialBindingState { … };

class SharedMemoryArbiterImplStartupTracingTest
    : public SharedMemoryArbiterImplTest { … };

INSTANTIATE_TEST_SUITE_P(…);

TEST_P(SharedMemoryArbiterImplStartupTracingTest, StartupTracingUnbound) { … }

TEST_P(SharedMemoryArbiterImplStartupTracingTest, StartupTracingBound) { … }

TEST_P(SharedMemoryArbiterImplStartupTracingTest,
       AbortStartupTracingForReservationUnbound) { … }

TEST_P(SharedMemoryArbiterImplStartupTracingTest,
       AbortStartupTracingForReservationBound) { … }

// TODO(primiano): add multi-threaded tests.

}  // namespace perfetto
chromium/third_party/perfetto/src/tracing/core/shared_memory_arbiter_impl_unittest.cc
