diff --git a/src/system_wrappers/source/data_log.cc b/src/system_wrappers/source/data_log.cc index c33f24c47c..af62018719 100644 --- a/src/system_wrappers/source/data_log.cc +++ b/src/system_wrappers/source/data_log.cc @@ -283,9 +283,11 @@ void DataLog::ReturnLog() { std::string DataLog::Combine(const std::string& table_name, int table_id) { std::stringstream ss; - std::string combined_id; - ss << table_name << "_" << table_id; - ss >> combined_id; + std::string combined_id = table_name; + std::string number_suffix; + ss << "_" << table_id; + ss >> number_suffix; + combined_id += number_suffix; std::transform(combined_id.begin(), combined_id.end(), combined_id.begin(), ::tolower); return combined_id; diff --git a/src/system_wrappers/source/data_log_unittest.cc b/src/system_wrappers/source/data_log_unittest.cc index d588947f04..c64ed94d6c 100644 --- a/src/system_wrappers/source/data_log_unittest.cc +++ b/src/system_wrappers/source/data_log_unittest.cc @@ -121,6 +121,11 @@ TEST(TestDataLog, CreateReturnTest) { ASSERT_LT(DataLog::AddTable(DataLog::Combine("table failure", 1)), 0); } +TEST(TestDataLog, VerifyCombineMethod) { + EXPECT_EQ(std::string("a proper table_1"), + DataLog::Combine("a proper table", 1)); +} + TEST(TestDataLog, VerifySingleTable) { DataLog::CreateLog(); DataLog::AddTable(DataLog::Combine("table", 1));