Adding Chrome-side WebVR interface

content/browser/vr/vr_service_unittest.cc

Index: content/browser/vr/vr_service_unittest.cc
diff --git a/content/browser/vr/vr_service_unittest.cc b/content/browser/vr/vr_service_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..26721977b8c801dcdc4ce8734865280c933f74e4
--- /dev/null
+++ b/content/browser/vr/vr_service_unittest.cc
@@ -0,0 +1,101 @@
+// Copyright 2015 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/memory/scoped_ptr.h"
+#include "content/browser/vr/vr_device_provider.h"
+#include "content/browser/vr/vr_service.h"
+#include "content/browser/vr/vr_test_helpers.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+using blink::WebVRDevice;
+
+class VRServiceTest : public testing::Test {
+ protected:
+  VRServiceTest();
+  ~VRServiceTest() override;
+
+  void SetUp() override;
+
+ protected:
+  MockVRDeviceProvider* provider_;
+  VRService* service_;
+
+  DISALLOW_COPY_AND_ASSIGN(VRServiceTest);
+};
+
+VRServiceTest::VRServiceTest() {
+
+}
+
+VRServiceTest::~VRServiceTest() {
+  delete service_;
+}
+
+void VRServiceTest::SetUp() {
+  provider_ = new MockVRDeviceProvider();
+  service_ = new VRService(provider_);
+}
+
+TEST_F(VRServiceTest, InitializationTest) {
+  EXPECT_FALSE(provider_->IsInitialized());
+
+  // The VRService should initialize each of it's providers upon initialization.
+  service_->Initialize();
+  EXPECT_TRUE(provider_->IsInitialized());
+
+  // Similarly, upon Shutdown it should shut down each of it's providers.
+  service_->Shutdown();
+  EXPECT_FALSE(provider_->IsInitialized());
+
+  // Calling GetDevices should initialize the service if it hasn't been
+  // initialized yet or has been shutdown.
+  std::vector<blink::WebVRDevice> webvr_devices;
+  service_->GetVRDevices(&webvr_devices);
+  EXPECT_TRUE(provider_->IsInitialized());
+}
+
+TEST_F(VRServiceTest, GetDevicesBasicTest) {
+  service_->Initialize();
+
+  // Should successfully return zero devices when none are available.
+  std::vector<blink::WebVRDevice> webvr_devices;
+  service_->GetVRDevices(&webvr_devices);
+  EXPECT_EQ((size_t)0, webvr_devices.size());
+
+  // GetDeviceByIndex should return NULL if an invalid index in queried.
+  VRDevice* queried_device = service_->GetDevice(1);
+  EXPECT_EQ(NULL, queried_device);
+
+  scoped_ptr<MockVRDevice> device1(new MockVRDevice(provider_));
+  provider_->AddDevice(device1.get());
+  service_->GetVRDevices(&webvr_devices);
+  // Should have successfully returned one device.
+  EXPECT_EQ((size_t)1, webvr_devices.size());
+  // The WebVRDevice index should match the device id.
+  EXPECT_EQ(webvr_devices[0].index, device1->id());
+
+  scoped_ptr<MockVRDevice> device2(new MockVRDevice(provider_));
+  provider_->AddDevice(device2.get());
+  service_->GetVRDevices(&webvr_devices);
+  // Should have successfully returned two devices.
+  EXPECT_EQ((size_t)2, webvr_devices.size());
+  // NOTE: Returned WebVRDevices are not required to be in any particular order.
+
+  // Querying the WebVRDevice index should return the correct device.
+  queried_device = service_->GetDevice(device1->id());
+  EXPECT_EQ(device1.get(), queried_device);
+  queried_device = service_->GetDevice(device2->id());
+  EXPECT_EQ(device2.get(), queried_device);
+
+  provider_->RemoveDevice(device1.get());
+  service_->GetVRDevices(&webvr_devices);
+  // Should have successfully returned one device.
+  EXPECT_EQ((size_t)1, webvr_devices.size());
+  // The WebVRDevice index should match the only remaining device id.
+  EXPECT_EQ(webvr_devices[0].index, device2->id());
+}
+
+}  // namespace content