Bug 1027255: Add ASSERT_/EXPECT_ GTest helpers for mozilla::pkix::Result, r=mmc, a=test-only
authorBrian Smith <brian@briansmith.org>
Wed, 18 Jun 2014 23:40:34 -0700
changeset 200676 06db3c6b9ada101fd50a8291bc4dc7b7a6554c97
parent 200675 e60e210dbfc0d9e24b729f3ef9cc031854213259
child 200677 91835441918e08562ec5f3955f3af8e8335b74d4
push id486
push userasasaki@mozilla.com
push dateMon, 14 Jul 2014 18:39:42 +0000
treeherdermozilla-release@d33428174ff1 [default view] [failures only]
perfherder[talos] [build metrics] [platform microbench] (compared to previous push)
reviewersmmc, test-only
bugs1027255
milestone31.0
Bug 1027255: Add ASSERT_/EXPECT_ GTest helpers for mozilla::pkix::Result, r=mmc, a=test-only
security/pkix/test/gtest/moz.build
security/pkix/test/gtest/pkixgtest.cpp
security/pkix/test/gtest/pkixgtest.h
--- a/security/pkix/test/gtest/moz.build
+++ b/security/pkix/test/gtest/moz.build
@@ -5,16 +5,17 @@
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
 LIBRARY_NAME = 'mozillapkix_gtest'
 
 SOURCES += [
     'pkixder_input_tests.cpp',
     'pkixder_pki_types_tests.cpp',
     'pkixder_universal_types_tests.cpp',
+    'pkixgtest.cpp',
 ]
 
 LOCAL_INCLUDES += [
     '../../include',
     '../../lib',
 ]
 
 FINAL_LIBRARY='xul-gtest'
new file mode 100644
--- /dev/null
+++ b/security/pkix/test/gtest/pkixgtest.cpp
@@ -0,0 +1,62 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This code is made available to you under your choice of the following sets
+ * of licensing terms:
+ */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+/* Copyright 2014 Mozilla Contributors
+ *
+ * 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.
+ */
+
+#include "pkixgtest.h"
+
+namespace mozilla { namespace pkix { namespace test {
+
+std::ostream&
+operator<<(std::ostream& os, const ResultWithPRErrorCode& value)
+{
+  switch (value.mRv)
+  {
+    case Success:
+      os << "Success";
+      break;
+    case RecoverableError:
+      os << "RecoverableError";
+      break;
+    case SECFailure:
+      os << "FatalError";
+      break;
+    default:
+      os << "[Invalid Result: " << static_cast<int64_t>(value.mRv) << ']';
+      break;
+  }
+
+  if (value.mRv != Success) {
+    os << '(';
+    const char* name = PR_ErrorToName(value.mErrorCode);
+    if (name) {
+      os << name;
+    } else {
+      os << value.mErrorCode;
+    }
+    os << ')';
+  }
+
+  return os;
+}
+
+} } } // namespace mozilla::pkix::test
new file mode 100644
--- /dev/null
+++ b/security/pkix/test/gtest/pkixgtest.h
@@ -0,0 +1,97 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This code is made available to you under your choice of the following sets
+ * of licensing terms:
+ */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+/* Copyright 2014 Mozilla Contributors
+ *
+ * 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 mozilla_pkix__pkixgtest_h
+#define mozilla_pkix__pkixgtest_h
+
+#include <ostream>
+
+#include "gtest/gtest.h"
+#include "pkixutil.h"
+#include "prerror.h"
+#include "stdint.h"
+
+namespace mozilla { namespace pkix { namespace test {
+
+class ResultWithPRErrorCode
+{
+public:
+  ResultWithPRErrorCode(Result rv, PRErrorCode errorCode)
+    : mRv(rv)
+    , mErrorCode(errorCode)
+  {
+  }
+
+  explicit ResultWithPRErrorCode(Result rv)
+    : mRv(rv)
+    , mErrorCode(rv == Success ? 0 : PR_GetError())
+  {
+  }
+
+  bool operator==(const ResultWithPRErrorCode& other) const
+  {
+    return mRv == other.mRv && mErrorCode == other.mErrorCode;
+  }
+
+private:
+  const Result mRv;
+  const PRErrorCode mErrorCode;
+
+  friend std::ostream& operator<<(std::ostream& os,
+                                  const ResultWithPRErrorCode & value);
+
+  void operator=(const ResultWithPRErrorCode&) /*= delete*/;
+};
+
+::std::ostream& operator<<(::std::ostream&, const ResultWithPRErrorCode &);
+
+#define ASSERT_Success(rv) \
+  ASSERT_EQ(::mozilla::pkix::test::ResultWithPRErrorCode( \
+                ::mozilla::pkix::Success, 0), \
+            ::mozilla::pkix::test::ResultWithPRErrorCode(rv))
+#define EXPECT_Success(rv) \
+  EXPECT_EQ(::mozilla::pkix::test::ResultWithPRErrorCode( \
+                ::mozilla::pkix::Success, 0), \
+            ::mozilla::pkix::test::ResultWithPRErrorCode(rv))
+
+#define ASSERT_RecoverableError(expectedError, rv) \
+  ASSERT_EQ(::mozilla::pkix::test::ResultWithPRErrorCode( \
+                 ::mozilla::pkix::RecoverableError, expectedError), \
+            ::mozilla::pkix::test::ResultWithPRErrorCode(rv))
+#define EXPECT_RecoverableError(expectedError, rv) \
+  EXPECT_EQ(::mozilla::pkix::test::ResultWithPRErrorCode( \
+                 ::mozilla::pkix::RecoverableError, expectedError), \
+            ::mozilla::pkix::test::ResultWithPRErrorCode(rv))
+
+#define ASSERT_FatalError(expectedError, rv) \
+  ASSERT_EQ(::mozilla::pkix::test::ResultWithPRErrorCode( \
+                 ::mozilla::pkix::FatalError, expectedError), \
+            ::mozilla::pkix::test::ResultWithPRErrorCode(rv))
+#define EXPECT_FatalError(expectedError, rv) \
+  EXPECT_EQ(::mozilla::pkix::test::ResultWithPRErrorCode( \
+                 ::mozilla::pkix::FatalError, expectedError), \
+            ::mozilla::pkix::test::ResultWithPRErrorCode(rv))
+
+} } } // namespace mozilla::pkix::test
+
+#endif // mozilla_pkix__pkixgtest_h