/*
|
* Copyright (C) 2018 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.
|
*/
|
|
#ifndef TEST_TASK_RUNNER_THREAD_DELEGATES_H_
|
#define TEST_TASK_RUNNER_THREAD_DELEGATES_H_
|
|
#include "perfetto/tracing/ipc/service_ipc_host.h"
|
#include "src/traced/probes/probes_producer.h"
|
#include "test/fake_producer.h"
|
#include "test/task_runner_thread.h"
|
|
namespace perfetto {
|
// This is used only in daemon starting integrations tests.
|
class ServiceDelegate : public ThreadDelegate {
|
public:
|
ServiceDelegate(const std::string& producer_socket,
|
const std::string& consumer_socket)
|
: producer_socket_(producer_socket), consumer_socket_(consumer_socket) {}
|
~ServiceDelegate() override = default;
|
|
void Initialize(base::TaskRunner* task_runner) override {
|
svc_ = ServiceIPCHost::CreateInstance(task_runner);
|
unlink(producer_socket_.c_str());
|
unlink(consumer_socket_.c_str());
|
svc_->Start(producer_socket_.c_str(), consumer_socket_.c_str());
|
}
|
|
private:
|
std::string producer_socket_;
|
std::string consumer_socket_;
|
std::unique_ptr<ServiceIPCHost> svc_;
|
};
|
|
// This is used only in daemon starting integrations tests.
|
class ProbesProducerDelegate : public ThreadDelegate {
|
public:
|
ProbesProducerDelegate(const std::string& producer_socket)
|
: producer_socket_(producer_socket) {}
|
~ProbesProducerDelegate() override = default;
|
|
void Initialize(base::TaskRunner* task_runner) override {
|
producer_.reset(new ProbesProducer);
|
producer_->ConnectWithRetries(producer_socket_.c_str(), task_runner);
|
}
|
|
private:
|
std::string producer_socket_;
|
std::unique_ptr<ProbesProducer> producer_;
|
};
|
|
class FakeProducerDelegate : public ThreadDelegate {
|
public:
|
FakeProducerDelegate(const std::string& producer_socket,
|
std::function<void()> setup_callback,
|
std::function<void()> connect_callback)
|
: producer_socket_(producer_socket),
|
setup_callback_(std::move(setup_callback)),
|
connect_callback_(std::move(connect_callback)) {}
|
~FakeProducerDelegate() override = default;
|
|
void Initialize(base::TaskRunner* task_runner) override {
|
producer_.reset(new FakeProducer("android.perfetto.FakeProducer"));
|
producer_->Connect(producer_socket_.c_str(), task_runner,
|
std::move(setup_callback_),
|
std::move(connect_callback_));
|
}
|
|
FakeProducer* producer() { return producer_.get(); }
|
|
private:
|
std::string producer_socket_;
|
std::unique_ptr<FakeProducer> producer_;
|
std::function<void()> setup_callback_;
|
std::function<void()> connect_callback_;
|
};
|
} // namespace perfetto
|
|
#endif // TEST_TASK_RUNNER_THREAD_DELEGATES_H_
|
