Adding unit tests for states.