Modify StateTransitions test

During loaded <-> idle state transition, check for events
is done after allocating all the buffers on a port. This is now
done after allocation of every single buffer. This is done to
improve scope of testing

Bug: 63796949

Change-Id: Icf3405ac6716f9533d31189f6c4a4c4ac908da5f