| 43 | }; |
| 44 | |
| 45 | TEST_F(EnsureWritableVectorTest, flat) { |
| 46 | SelectivityVector rows(1'000); |
| 47 | |
| 48 | VectorPtr result; |
| 49 | BaseVector::ensureWritable(rows, BIGINT(), pool(), result); |
| 50 | ASSERT_TRUE(result); |
| 51 | ASSERT_EQ(rows.size(), result->size()); |
| 52 | ASSERT_EQ(TypeKind::BIGINT, result->typeKind()); |
| 53 | ASSERT_EQ(VectorEncoding::Simple::FLAT, result->encoding()); |
| 54 | ASSERT_TRUE(result->mayChangeContentUnderSameAddress()); |
| 55 | |
| 56 | auto* flatResult = result->asFlatVector<int64_t>(); |
| 57 | for (vector_size_t i = 0; i < rows.size(); i++) { |
| 58 | if (i % 5 == 0) { |
| 59 | flatResult->setNull(i, true); |
| 60 | } else { |
| 61 | flatResult->set(i, i); |
| 62 | } |
| 63 | } |
| 64 | |
| 65 | // Singly referenced vector with singly referenced buffers should be reused |
| 66 | // as-is |
| 67 | auto* rawNulls = flatResult->nulls().get(); |
| 68 | auto* rawValues = flatResult->values().get(); |
| 69 | BaseVector::ensureWritable(rows, BIGINT(), pool(), result); |
| 70 | ASSERT_EQ(flatResult, result.get()); |
| 71 | ASSERT_EQ(rawNulls, flatResult->nulls().get()); |
| 72 | ASSERT_EQ(rawValues, flatResult->values().get()); |
| 73 | |
| 74 | // Resize upwards singly-referenced vector with singly referenced buffers |
| 75 | rows.resize(2'000); |
| 76 | BaseVector::ensureWritable(rows, BIGINT(), pool(), result); |
| 77 | ASSERT_EQ(rows.size(), result->size()); |
| 78 | ASSERT_EQ(flatResult, result.get()); |
| 79 | rawNulls = flatResult->nulls().get(); |
| 80 | rawValues = flatResult->values().get(); |
| 81 | |
| 82 | for (vector_size_t i = 0; i < rows.size(); i++) { |
| 83 | if (i % 5 == 0) { |
| 84 | flatResult->setNull(i, true); |
| 85 | } else { |
| 86 | flatResult->set(i, i); |
| 87 | } |
| 88 | } |
| 89 | |
| 90 | // Resize downwards |
| 91 | rows.resize(1'024); |
| 92 | BaseVector::ensureWritable(rows, BIGINT(), pool(), result); |
| 93 | ASSERT_EQ(2'000, result->size()); |
| 94 | ASSERT_EQ(rawNulls, flatResult->nulls().get()); |
| 95 | ASSERT_EQ(rawValues, flatResult->values().get()); |
| 96 | |
| 97 | // Add second reference to the vector -> new vector should be allocated |
| 98 | auto resultCopy = result; |
| 99 | |
| 100 | BaseVector::ensureWritable(rows, BIGINT(), pool(), result); |
| 101 | ASSERT_NE(flatResult, result.get()); |
| 102 | ASSERT_TRUE(result->mayChangeContentUnderSameAddress()); |
nothing calls this directly
no test coverage detected