Function TEST

be/src/runtime/row-batch-test.cc:37–71

static scoped_ptr<Frontend> fe;

TEST(RowBatchTest, AcquireStateWithMarkFlushResources) {
  // Test that AcquireState() can be correctly called with MarkFlushResources() on the
  // source batch.
  ObjectPool pool;
  DescriptorTblBuilder builder(fe.get(), &pool);
  builder.DeclareTuple() << TYPE_INT;
  DescriptorTbl* desc_tbl = builder.Build();

  vector<bool> nullable_tuples = {false};
  vector<TTupleId> tuple_id = {static_cast<TupleId>(0)};
  RowDescriptor row_desc(*desc_tbl, tuple_id, nullable_tuples);
  MemTracker tracker;
  {
    RowBatch src(&row_desc, 1024, &tracker);
    src.AddRow();
    src.CommitLastRow();
    // Calls MarkFlushResources().
    src.MarkNeedsDeepCopy();

    // Note InitialCapacity(), not capacity(). Latter will DCHECK.
    RowBatch dest(&row_desc, src.InitialCapacity(), &tracker);
    dest.AcquireState(&src);
  }

  // Confirm the bad pattern causes an error.
  {
    RowBatch src(&row_desc, 1024, &tracker);
    src.AddRow();
    src.CommitLastRow();
    // Calls MarkFlushResources().
    src.MarkNeedsDeepCopy();
    RowBatch bad_dest(&row_desc, src.capacity(), &tracker);
    IMPALA_ASSERT_DEBUG_DEATH(bad_dest.AcquireState(&src), "");
  }
}

int main(int argc, char** argv) {
  ::testing::InitGoogleTest(&argc, argv);