diff options
| -rw-r--r-- | test/verify_test.cc | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/test/verify_test.cc b/test/verify_test.cc index 1bc7649f..981f8e24 100644 --- a/test/verify_test.cc +++ b/test/verify_test.cc @@ -245,6 +245,27 @@ public: } } + void delete_lines_after(string after, int lines_to_delete) + { + ChangeChecker cc(this); + auto lines = as_lines(); + _content = ""; + auto iter = std::find_if(lines.begin(), lines.end(), [after](string const& line) { + return line.find(after) != string::npos; + }); + int to_delete = 0; + for (auto i = lines.begin(); i != lines.end(); ++i) { + if (i == iter) { + to_delete = lines_to_delete; + _content += *i + "\n"; + } else if (to_delete == 0) { + _content += *i + "\n"; + } else { + --to_delete; + } + } + } + private: friend class ChangeChecker; |