diff options
Diffstat (limited to 'src/mongo/db/dbmessage_test.cpp')
-rw-r--r-- | src/mongo/db/dbmessage_test.cpp | 155 |
1 files changed, 77 insertions, 78 deletions
diff --git a/src/mongo/db/dbmessage_test.cpp b/src/mongo/db/dbmessage_test.cpp index 867a52d9885..c0d40f49bde 100644 --- a/src/mongo/db/dbmessage_test.cpp +++ b/src/mongo/db/dbmessage_test.cpp @@ -33,111 +33,110 @@ #include "mongo/unittest/unittest.h" namespace mongo { - using std::string; +using std::string; - // Test if the reserved field is short of 4 bytes - TEST(DBMessage1, ShortFlags) { - BufBuilder b; - string ns("test"); +// Test if the reserved field is short of 4 bytes +TEST(DBMessage1, ShortFlags) { + BufBuilder b; + string ns("test"); - b.appendChar( 1 ); + b.appendChar(1); - Message toSend; - toSend.setData( dbDelete , b.buf() , b.len() ); + Message toSend; + toSend.setData(dbDelete, b.buf(), b.len()); - ASSERT_THROWS(DbMessage d1(toSend), UserException); - } + ASSERT_THROWS(DbMessage d1(toSend), UserException); +} - // Test a short NS missing a trailing null - TEST(DBMessage1, BadNS) { - BufBuilder b; +// Test a short NS missing a trailing null +TEST(DBMessage1, BadNS) { + BufBuilder b; - b.appendNum( static_cast<int>(1) ); - b.appendChar( 'b' ); - b.appendChar( 'a' ); - b.appendChar( 'd' ); - // Forget to append \0 + b.appendNum(static_cast<int>(1)); + b.appendChar('b'); + b.appendChar('a'); + b.appendChar('d'); + // Forget to append \0 - Message toSend; - toSend.setData( dbDelete , b.buf() , b.len() ); + Message toSend; + toSend.setData(dbDelete, b.buf(), b.len()); - ASSERT_THROWS(DbMessage d1(toSend), UserException); - } + ASSERT_THROWS(DbMessage d1(toSend), UserException); +} - // Test a valid kill message and try an extra pull - TEST(DBMessage1, GoodKill) { - BufBuilder b; +// Test a valid kill message and try an extra pull +TEST(DBMessage1, GoodKill) { + BufBuilder b; - b.appendNum( static_cast<int>(1) ); - b.appendNum( static_cast<int>(3) ); + b.appendNum(static_cast<int>(1)); + b.appendNum(static_cast<int>(3)); - Message toSend; - toSend.setData( dbKillCursors , b.buf() , b.len() ); + Message toSend; + toSend.setData(dbKillCursors, b.buf(), b.len()); - DbMessage d1(toSend); - ASSERT_EQUALS(3, d1.pullInt()); + DbMessage d1(toSend); + ASSERT_EQUALS(3, d1.pullInt()); - ASSERT_THROWS(d1.pullInt(), UserException); - } + ASSERT_THROWS(d1.pullInt(), UserException); +} - // Try a bad read of a type too large - TEST(DBMessage1, GoodKill2) { - BufBuilder b; +// Try a bad read of a type too large +TEST(DBMessage1, GoodKill2) { + BufBuilder b; - b.appendNum( static_cast<int>(1) ); - b.appendNum( static_cast<int>(3) ); + b.appendNum(static_cast<int>(1)); + b.appendNum(static_cast<int>(3)); - Message toSend; - toSend.setData( dbKillCursors , b.buf() , b.len() ); + Message toSend; + toSend.setData(dbKillCursors, b.buf(), b.len()); - DbMessage d1(toSend); - ASSERT_THROWS(d1.pullInt64(), UserException); - } + DbMessage d1(toSend); + ASSERT_THROWS(d1.pullInt64(), UserException); +} - // Test a basic good insert, and an extra read - TEST(DBMessage1, GoodInsert) { - BufBuilder b; - string ns("test"); +// Test a basic good insert, and an extra read +TEST(DBMessage1, GoodInsert) { + BufBuilder b; + string ns("test"); - b.appendNum( static_cast<int>(1) ); - b.appendStr(ns); - b.appendNum( static_cast<int>(3) ); - b.appendNum( static_cast<int>(39) ); + b.appendNum(static_cast<int>(1)); + b.appendStr(ns); + b.appendNum(static_cast<int>(3)); + b.appendNum(static_cast<int>(39)); - Message toSend; - toSend.setData( dbInsert , b.buf() , b.len() ); + Message toSend; + toSend.setData(dbInsert, b.buf(), b.len()); - DbMessage d1(toSend); - ASSERT_EQUALS(3, d1.pullInt()); - ASSERT_EQUALS(39, d1.pullInt()); - ASSERT_THROWS(d1.pullInt(), UserException); - } + DbMessage d1(toSend); + ASSERT_EQUALS(3, d1.pullInt()); + ASSERT_EQUALS(39, d1.pullInt()); + ASSERT_THROWS(d1.pullInt(), UserException); +} - // Test a basic good insert, and an extra read - TEST(DBMessage1, GoodInsert2) { - BufBuilder b; - string ns("test"); +// Test a basic good insert, and an extra read +TEST(DBMessage1, GoodInsert2) { + BufBuilder b; + string ns("test"); - b.appendNum( static_cast<int>(1) ); - b.appendStr(ns); - b.appendNum( static_cast<int>(3) ); - b.appendNum( static_cast<int>(39) ); + b.appendNum(static_cast<int>(1)); + b.appendStr(ns); + b.appendNum(static_cast<int>(3)); + b.appendNum(static_cast<int>(39)); - BSONObj bo = BSON( "ts" << 0 ); - bo.appendSelfToBufBuilder( b ); + BSONObj bo = BSON("ts" << 0); + bo.appendSelfToBufBuilder(b); - Message toSend; - toSend.setData( dbInsert , b.buf() , b.len() ); + Message toSend; + toSend.setData(dbInsert, b.buf(), b.len()); - DbMessage d1(toSend); - ASSERT_EQUALS(3, d1.pullInt()); + DbMessage d1(toSend); + ASSERT_EQUALS(3, d1.pullInt()); - ASSERT_EQUALS(39, d1.pullInt()); - BSONObj bo2 = d1.nextJsObj(); - ASSERT_THROWS(d1.nextJsObj(), MsgAssertionException); - } + ASSERT_EQUALS(39, d1.pullInt()); + BSONObj bo2 = d1.nextJsObj(); + ASSERT_THROWS(d1.nextJsObj(), MsgAssertionException); +} - -} // mongo namespace +} // mongo namespace |