Fix UT, missing mock.
Fix UT, missing mock.

matthewdupre/felix,nbartos/calico,kasisnu/calico,Metaswitch/calico,neiljerram/felix,anortef/calico,beddari/calico,TrimBiggs/calico,neiljerram/felix,anortef/calico,Metaswitch/calico,neiljerram/felix,TrimBiggs/calico,alexhersh/calico,ocadotechnology/calico,nbartos/calico,beddari/calico,ocadotechnology/calico,kasisnu/calico,alexhersh/calico,neiljerram/felix,matthewdupre/felix