mirrors/scylladb
1
0
Fork 0
mirror of https://github.com/scylladb/scylladb.git synced 2026-05-12 19:02:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: mutation_assertions: Allow expecting fragments

Browse Source
This commit is contained in:
Tomasz Grabiec
2017-06-07 15:16:57 +02:00
parent 1f23130b07
commit 44fdee3f2e
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/mutation_assertions.hh
View File

@@ -160,6 +160,25 @@ public:
return *this;
}
streamed_mutation_assertions& produces(mutation_fragment mf) {
auto mfopt = _sm().get0();
if (!mfopt) {
BOOST_FAIL(sprint("Expected mutation fragment %s, got end of stream", mf));
}
if (!mfopt->equal(*_sm.schema(), mf)) {
BOOST_FAIL(sprint("Expected %s, but got %s", mf, *mfopt));
}
return *this;
}
streamed_mutation_assertions& produces_only(const std::deque<mutation_fragment>& fragments) {
for (auto&& f : fragments) {
produces(f);
}
produces_end_of_stream();
return *this;
}
streamed_mutation_assertions& produces_row_with_key(const clustering_key& ck) {
BOOST_TEST_MESSAGE(sprint("Expect %s", ck));
auto mfo = _sm().get0();