Add tests for get_state