mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 05:32:52 +00:00
30 lines
937 B
C++
30 lines
937 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <DataStreams/ColumnGathererStream.h>
|
|
|
|
using DB::RowSourcePart;
|
|
|
|
static void check(const RowSourcePart & s, size_t num, bool flag)
|
|
{
|
|
EXPECT_FALSE((s.getSourceNum() != num || s.getSkipFlag() != flag) || (!flag && s.data != num));
|
|
}
|
|
|
|
TEST(ColumnGathererStream, RowSourcePartBitsTest)
|
|
{
|
|
check(RowSourcePart(0, false), 0, false);
|
|
check(RowSourcePart(0, true), 0, true);
|
|
check(RowSourcePart(1, false), 1, false);
|
|
check(RowSourcePart(1, true), 1, true);
|
|
check(RowSourcePart(RowSourcePart::MAX_PARTS, false), RowSourcePart::MAX_PARTS, false);
|
|
check(RowSourcePart(RowSourcePart::MAX_PARTS, true), RowSourcePart::MAX_PARTS, true);
|
|
|
|
RowSourcePart p{80, false};
|
|
check(p, 80, false);
|
|
p.setSkipFlag(true);
|
|
check(p, 80, true);
|
|
p.setSkipFlag(false);
|
|
check(p, 80, false);
|
|
p.setSourceNum(RowSourcePart::MAX_PARTS);
|
|
check(p, RowSourcePart::MAX_PARTS, false);
|
|
}
|