diff options
-rw-r--r-- | linker/linked_list.h | 6 | ||||
-rw-r--r-- | linker/tests/linked_list_test.cpp | 17 |
2 files changed, 23 insertions, 0 deletions
diff --git a/linker/linked_list.h b/linker/linked_list.h index 8003dbf84..0572e635b 100644 --- a/linker/linked_list.h +++ b/linker/linked_list.h @@ -127,7 +127,13 @@ class LinkedList { } else { p->next = next; } + + if (tail_ == e) { + tail_ = p; + } + Allocator::free(e); + e = next; } else { p = e; diff --git a/linker/tests/linked_list_test.cpp b/linker/tests/linked_list_test.cpp index 09ad68766..12348d9be 100644 --- a/linker/tests/linked_list_test.cpp +++ b/linker/tests/linked_list_test.cpp @@ -133,6 +133,23 @@ TEST(linked_list, remove_if_then_pop) { ASSERT_TRUE(list.pop_front() == nullptr); } +TEST(linked_list, remove_if_last_then_push_back) { + test_list_t list; + + list.push_back("a"); + list.push_back("b"); + list.push_back("c"); + list.push_back("d"); + + list.remove_if([](const char* c) { + return *c == 'c' || *c == 'd'; + }); + + ASSERT_EQ("ab", test_list_to_string(list)); + list.push_back("d"); + ASSERT_EQ("abd", test_list_to_string(list)); +} + TEST(linked_list, copy_to_array) { test_list_t list; const size_t max_size = 128; |