Add get_states basic test.