arc/vm/vsock_proxy/proxy_service_test.cc

// Copyright 2019 The Chromium OS 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 "arc/vm/vsock_proxy/proxy_service.h"

#include <gtest/gtest.h>

#include "arc/vm/vsock_proxy/proxy_base.h"

namespace arc {
namespace {

class FakeProxy;
// Track the instance.
FakeProxy* instance = nullptr;

// Fake implementation of ProxyBase to track the state.
class FakeProxy : public ProxyBase {
 public:
  FakeProxy() {
    EXPECT_FALSE(instance);
    instance = this;
  }

  ~FakeProxy() override {
    EXPECT_EQ(instance, this);
    instance = nullptr;
  }

  bool Initialize() override {
    initialized_ = true;
    return true;
  }

  bool is_initialized() const { return initialized_; }

  VSockProxy* GetVSockProxy() override { return nullptr; }

 private:
  bool initialized_ = false;

  DISALLOW_COPY_AND_ASSIGN(FakeProxy);
};

// Fake implementation of ProxyFactory to inject FakeProxy.
class FakeProxyFactory : public ProxyService::ProxyFactory {
 public:
  FakeProxyFactory() = default;
  ~FakeProxyFactory() override = default;

  std::unique_ptr<ProxyBase> Create() override {
    return std::make_unique<FakeProxy>();
  }

 private:
  DISALLOW_COPY_AND_ASSIGN(FakeProxyFactory);
};

TEST(ProxyServiceTest, Run) {
  {
    ProxyService service(std::make_unique<FakeProxyFactory>());
    ASSERT_TRUE(service.Start());
    ASSERT_TRUE(instance);
    EXPECT_TRUE(instance->is_initialized());
  }
  // Destroying the |service| should destroy the proxy and stop the dedicated
  // thread. Then unblocked.
  EXPECT_FALSE(instance);
}

}  // namespace
}  // namespace arc