A partner at law firm Cooley got an unexpected call late last year from a lawyer for one of the firm’s most famous clients, Elon Musk’s Tesla, with an ultimatum.

The world’s richest man wanted Cooley, which was representing Tesla in numerous lawsuits, to fire one of its attorneys or it would lose the electric-vehicle company’s business, people familiar with the matter said.