Add unit test

[sweemeng]

Design test for this, use mock because we should mock actual API call. There is rate limit.