Add more tests for company API