1 files changed, 93 insertions, 0 deletions
diff --git a/base/scoped_bstr_win_unittest.cc b/base/scoped_bstr_win_unittest.cc
new file mode 100644
index 0000000..a2e49ac
--- /dev/null
+++ b/base/scoped_bstr_win_unittest.cc
@@ -0,0 +1,93 @@
+// Copyright (c) 2006-2008 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/scoped_bstr_win.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+static const wchar_t kTestString1[] = L"123";
+static const wchar_t kTestString2[] = L"456789";
+uint32 test1_len = arraysize(kTestString1) - 1;
+uint32 test2_len = arraysize(kTestString2) - 1;
+
+void DumbBstrTests() {
+  ScopedBstr b;
+  EXPECT_TRUE(b == NULL);
+  EXPECT_TRUE(b.Length() == 0);
+  EXPECT_TRUE(b.ByteLength() == 0);
+  b.Reset(NULL);
+  EXPECT_TRUE(b == NULL);
+  EXPECT_TRUE(b.Release() == NULL);
+  ScopedBstr b2;
+  b.Swap(b2);
+  EXPECT_TRUE(b2 == NULL);
+}
+
+void GiveMeABstr(BSTR* ret) {
+  *ret = SysAllocString(kTestString1);
+}
+
+void BasicBstrTests() {
+  ScopedBstr b1(kTestString1);
+  EXPECT_TRUE(b1.Length() == test1_len);
+  EXPECT_TRUE(b1.ByteLength() == test1_len * sizeof(kTestString1[0]));
+
+  ScopedBstr b2;
+  b1.Swap(b2);
+  EXPECT_TRUE(b2.Length() == test1_len);
+  EXPECT_TRUE(b1.Length() == 0);
+  EXPECT_TRUE(lstrcmpW(b2, kTestString1) == 0);
+  BSTR tmp = b2.Release();
+  EXPECT_TRUE(tmp != NULL);
+  EXPECT_TRUE(lstrcmpW(tmp, kTestString1) == 0);
+  EXPECT_TRUE(b2 == NULL);
+  SysFreeString(tmp);
+
+  GiveMeABstr(b2.Receive());
+  EXPECT_TRUE(b2 != NULL);
+  b2.Reset();
+  EXPECT_TRUE(b2.AllocateBytes(100) != NULL);
+  EXPECT_TRUE(b2.ByteLength() == 100);
+  EXPECT_TRUE(b2.Length() == 100 / sizeof(kTestString1[0]));
+  lstrcpyW(static_cast<BSTR>(b2), kTestString1);
+  EXPECT_TRUE(lstrlen(b2) == test1_len);
+  EXPECT_TRUE(b2.Length() == 100 / sizeof(kTestString1[0]));
+  b2.SetByteLen(lstrlen(b2) * sizeof(kTestString2[0]));
+  EXPECT_TRUE(lstrlen(b2) == b2.Length());
+
+  EXPECT_TRUE(b1.Allocate(kTestString2) != NULL);
+  EXPECT_TRUE(b1.Length() == test2_len);
+  b1.SetByteLen((test2_len - 1) * sizeof(kTestString2[0]));
+  EXPECT_TRUE(b1.Length() == test2_len - 1);
+}
+
+}  // namespace
+
+TEST(ScopedBstrTest, ScopedBstr) {
+  DumbBstrTests();
+  BasicBstrTests();
+}
+
+#define kSourceStr L"this is a string"
+#define kSourceStrEmpty L""
+
+TEST(StackBstrTest, StackBstr) {
+  ScopedBstr system_bstr(kSourceStr);
+  StackBstrVar(kSourceStr, stack_bstr);
+  EXPECT_EQ(VarBstrCmp(system_bstr, stack_bstr, LOCALE_USER_DEFAULT, 0),
+            VARCMP_EQ);
+
+  StackBstrVar(kSourceStrEmpty, empty);
+  uint32 l1 = SysStringLen(stack_bstr);
+  uint32 l2 = SysStringLen(StackBstr(kSourceStr));
+  uint32 l3 = SysStringLen(system_bstr);
+  EXPECT_TRUE(l1 == l2);
+  EXPECT_TRUE(l2 == l3);
+  EXPECT_TRUE(SysStringLen(empty) == 0);
+
+  const wchar_t one_more_test[] = L"this is my const string";
+  EXPECT_EQ(SysStringLen(StackBstr(one_more_test)),
+            lstrlenW(one_more_test));
+}