def test_get_access_extension_upgrade_info(self): #when there is no extension installed on linux vm, use the version we like publisher, version, auto_upgrade = _get_access_extension_upgrade_info( None, _LINUX_ACCESS_EXT) self.assertEqual('Microsoft.OSTCExtensions', publisher) self.assertEqual('1.4', version) self.assertEqual(None, auto_upgrade) #when there is no extension installed on windows vm, use the version we like publisher, version, auto_upgrade = _get_access_extension_upgrade_info( None, _WINDOWS_ACCESS_EXT) self.assertEqual('Microsoft.Compute', publisher) self.assertEqual('2.0', version) self.assertEqual(None, auto_upgrade) #when there is existing extension with higher version, stick to that extentions = [FakedAccessExtensionEntity(True, '3.0')] publisher, version, auto_upgrade = _get_access_extension_upgrade_info( extentions, _LINUX_ACCESS_EXT) self.assertEqual('3.0', version) self.assertEqual(None, auto_upgrade) extentions = [FakedAccessExtensionEntity(False, '10.0')] publisher, version, auto_upgrade = _get_access_extension_upgrade_info( extentions, _WINDOWS_ACCESS_EXT) self.assertEqual('10.0', version) self.assertEqual(None, auto_upgrade) #when there is existing extension with lower version, upgrade to ours extentions = [FakedAccessExtensionEntity(True, '1.0')] publisher, version, auto_upgrade = _get_access_extension_upgrade_info( extentions, _LINUX_ACCESS_EXT) self.assertEqual('1.4', version) self.assertEqual(True, auto_upgrade)
def test_get_access_extension_upgrade_info(self): # when there is no extension installed on linux vm, use the version we like publisher, version, auto_upgrade = _get_access_extension_upgrade_info( None, _LINUX_ACCESS_EXT) self.assertEqual('Microsoft.OSTCExtensions', publisher) self.assertEqual('1.4', version) self.assertEqual(None, auto_upgrade) # when there is no extension installed on windows vm, use the version we like publisher, version, auto_upgrade = _get_access_extension_upgrade_info( None, _WINDOWS_ACCESS_EXT) self.assertEqual('Microsoft.Compute', publisher) self.assertEqual('2.0', version) self.assertEqual(None, auto_upgrade) # when there is existing extension with higher version, stick to that extentions = [FakedAccessExtensionEntity(True, '3.0')] publisher, version, auto_upgrade = _get_access_extension_upgrade_info( extentions, _LINUX_ACCESS_EXT) self.assertEqual('3.0', version) self.assertEqual(None, auto_upgrade) extentions = [FakedAccessExtensionEntity(False, '10.0')] publisher, version, auto_upgrade = _get_access_extension_upgrade_info( extentions, _WINDOWS_ACCESS_EXT) self.assertEqual('10.0', version) self.assertEqual(None, auto_upgrade) # when there is existing extension with lower version, upgrade to ours extentions = [FakedAccessExtensionEntity(True, '1.0')] publisher, version, auto_upgrade = _get_access_extension_upgrade_info( extentions, _LINUX_ACCESS_EXT) self.assertEqual('1.4', version) self.assertEqual(True, auto_upgrade)