Update write burst test to use session