updated unit test accordingly
updated unit test accordingly

fog/fog-azure-rm,vishrutshah/fog-azure-rm,bilal-naeem-confiz/fog-azure-rm,vishrutshah/fog-azure-rm,fog/fog-azure-rm,bilal-naeem-confiz/fog-azure-rm