Add integration test for states (create, update, delete)