Override << operator for better test error messages

ackerleytng · Jul 7, 2020 · aa36997 · aa36997
1 parent 286dfea
commit aa36997
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/jwt_verify_lib/status.h b/jwt_verify_lib/status.h
@@ -209,6 +209,14 @@ enum class Status {
   JwksBioAllocError,
 };
 
+/**
+ * Overload << to output a Status to std::ostream
+ * @param os the std::ostream to write to
+ * @param status is the enum status.
+ * @return the std::ostream os
+ */
+std::ostream& operator<<(std::ostream& os, const Status& status);
+
 /**
  * Convert enum status to string.
  * @param status is the enum status.

diff --git a/src/status.cc b/src/status.cc
@@ -15,6 +15,7 @@
 #include "jwt_verify_lib/status.h"
 
 #include <map>
+#include <iostream>
 
 namespace google {
 namespace jwt_verify {
@@ -173,5 +174,12 @@ std::string getStatusString(Status status) {
   return "";
 }
 
+std::ostream& operator<<(std::ostream& os, const Status& status) {
+  return os <<
+      "Status(" <<
+      static_cast<int>(status) << ", " <<
+      getStatusString(status) << ")";
+}
+
 }  // namespace jwt_verify
 }  // namespace google