[Chore]: Update test to run mock unit tests

[homanp]

Description

Currently tests are running the production APIs. Would be good if these were mock tests all the way.