squash: beef up testing in TestGetUnitState