p36850791
/
kinetic-devel
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Adding unit test that a bag can be reopened and making sure that we write our final connection records and chunk info records at the END of the file

This commit is contained in:
Jeremy Leibs 2010-06-09 07:40:46 +00:00
parent cb3a561852
commit f6b12b2292
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/rosbag/src/bag.cpp
View File

@ -228,6 +228,8 @@ void Bag::stopWriting() {
if (chunk_open_)
stopWritingChunk();
seek(0, std::ios::end);
index_data_pos_ = file_.getOffset();
writeConnectionRecords();
writeChunkInfoRecords();

14
tools/rosbag/test/test_bag.cpp
View File

@ -459,6 +459,7 @@ TEST(rosbag, modify_view_works) {
TEST(rosbag, modify_bag_works) {
rosbag::Bag rwbag("/tmp/modify_bag_works.bag", rosbag::bagmode::Write | rosbag::bagmode::Read);
rwbag.setChunkThreshold(1);
std::vector<std::string> t0 = boost::assign::list_of("t0");
@ -509,6 +510,19 @@ TEST(rosbag, modify_bag_works) {
}
rwbag.close();
rosbag::Bag rwbag2("/tmp/modify_bag_works.bag", rosbag::bagmode::Read);
rosbag::View view2(rwbag2, rosbag::TopicQuery(t0));
rosbag::View::iterator iter3 = view2.begin();
imsg = iter3->instantiate<std_msgs::Int32>();
// Iter2 should contain 1->10
for (int i = 0; i < 10; i++) {
imsg = iter3->instantiate<std_msgs::Int32>();
ASSERT_EQ(imsg->data, i);
iter3++;
}
}
int main(int argc, char **argv) {