Apply state store wrapper for updates to all tests

[patchwork01]

Background

Split from:

• Unify method signatures for synchronous and asynchronous commit #4262

Follows on from:

• State store wrapper for testing to preserve existing update methods #4279

Description

We'd like to apply the new test wrapper for state store updates to all the tests.

Analysis

This should be a process of replicating what was done in a few tests in the linked issue, but for all the tests that perform updates to the state store in the test code.