Define FIDO U2f Device abstraction

device/u2f/u2f_device_unittest.cc

+// Copyright 2017 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 "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "u2f_apdu_command.h"
+#include "u2f_device.h"
+
+namespace device {
+
+class U2fDeviceTest : public testing::Test {};
+
+TEST_F(U2fDeviceTest, TestBuildApdu) {
+  std::vector<uint8_t> appid(U2fDevice::kAppIdDigestLen, 0x01);
+  std::vector<uint8_t> challenge(U2fDevice::kChallengeDigestLen, 0xff);
+  scoped_refptr<U2fApduCommand> cmd =
+      U2fDevice::BuildRegisterCommand(appid, challenge);
+  ASSERT_NE(nullptr, cmd);
+  EXPECT_THAT(U2fApduCommand::CreateFromMessage(cmd->GetEncodedCommand())
+                  ->GetEncodedCommand(),
+              testing::ContainerEq(cmd->GetEncodedCommand()));
+  // Expect null result with appid.size() > kAppIdDigestLen
+  appid.push_back(0xff);
+  cmd = U2fDevice::BuildRegisterCommand(appid, challenge);
+  EXPECT_EQ(nullptr, cmd);
+
+  appid.pop_back();
+  std::vector<uint8_t> key_handle(U2fDevice::kMaxKeyHandleLength);
+  cmd = U2fDevice::BuildSignCommand(appid, challenge, key_handle);
+  ASSERT_NE(nullptr, cmd);
+  EXPECT_THAT(U2fApduCommand::CreateFromMessage(cmd->GetEncodedCommand())
+                  ->GetEncodedCommand(),
+              testing::ContainerEq(cmd->GetEncodedCommand()));
+  // Expect null result with appid.size() > kMaxKeyHandleLength
+  key_handle.push_back(0x0f);
+  cmd = U2fDevice::BuildSignCommand(appid, challenge, key_handle);
+  EXPECT_EQ(nullptr, cmd);
+}
+
+}  // namespace device