def test_uploadImageNoVolumeUpdate(self): uploader = Uploader('image.img', self.ch) uploader.pdisk.uploadVolume = self._uploadVolume uploader.pdisk.updateVolumeAsUser = self._updateVolumeAsUser uploader._uploadImage() assert uploader.pdisk.uploadVolume.called == True assert uploader.pdisk.uploadVolume.call_count == 1 assert uploader.pdisk.uploadVolume.call_args == (('image.img',), {}) assert uploader.pdisk.uploadVolume._return_value == 'https://example.com/pdisk/uuid' assert uploader.pdisk.updateVolumeAsUser.called == False
def test_uploadImageVolumeUpdate(self): self.ch.set('imageMetadata', {'foo':'bar'}) uploader = Uploader('image.img', self.ch) uploader.pdisk.uploadVolume = self._uploadVolume uploader.pdisk.updateVolumeAsUser = self._updateVolumeAsUser uploader._uploadImage() assert uploader.pdisk.uploadVolume.called == True assert uploader.pdisk.uploadVolume.call_count == 1 assert uploader.pdisk.uploadVolume.call_args == (('image.img',), {}) assert uploader.pdisk.uploadVolume._return_value == 'https://example.com/pdisk/uuid' assert uploader.pdisk.updateVolumeAsUser.called == True assert uploader.pdisk.updateVolumeAsUser.call_count == 1 assert uploader.pdisk.updateVolumeAsUser.call_args == (({'foo':'bar'}, 'uuid'), {})