UpgradeCatalog250 extends AbstractUpgradeCatalog { @Override protected void executeDMLUpdates() throws AmbariException, SQLException { addNewConfigurationsFromXml(); updateAMSConfigs(); updateStormAlerts(); updateLogSearchAlert(); removeAlertDuplicates(); updateHadoopEnvConfigs(); updateKafkaConfigs(); updateHIVEInteractiveConfigs(); updateHiveLlapConfigs(); updateTablesForZeppelinViewRemoval(); updateZeppelinConfigs(); updateAtlasConfigs(); updateLogSearchConfigs(); updateAmbariInfraConfigs(); updateYarnSite(); updateRangerUrlConfigs(); addManageServiceAutoStartPermissions(); addManageAlertNotificationsPermissions(); updateKerberosDescriptorArtifacts(); fixHBaseMasterCPUUtilizationAlertDefinition(); updateTezHistoryUrlBase(); } @Inject  UpgradeCatalog250(Injector injector); @Override String getTargetVersion(); @Override String getSourceVersion(); static final String COMPONENT_TABLE; static final String COMPONENT_VERSION_TABLE; static final String COMPONENT_VERSION_PK; static final String COMPONENT_VERSION_FK_COMPONENT; static final String COMPONENT_VERSION_FK_REPO_VERSION; }
@Test public void testExecuteDMLUpdates() throws Exception { Method updateAmsConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateAMSConfigs"); Method updateHadoopEnvConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateHadoopEnvConfigs"); Method updateKafkaConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateKafkaConfigs"); Method updateHiveLlapConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateHiveLlapConfigs"); Method updateHIVEInteractiveConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateHIVEInteractiveConfigs"); Method addManageServiceAutoStartPermissions = UpgradeCatalog250.class.getDeclaredMethod("addManageServiceAutoStartPermissions"); Method addManageAlertNotificationsPermissions = UpgradeCatalog250.class.getDeclaredMethod("addManageAlertNotificationsPermissions"); Method addNewConfigurationsFromXml = AbstractUpgradeCatalog.class.getDeclaredMethod("addNewConfigurationsFromXml"); Method updateTablesForZeppelinViewRemoval = UpgradeCatalog250.class.getDeclaredMethod("updateTablesForZeppelinViewRemoval"); Method updateZeppelinConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateZeppelinConfigs"); Method updateAtlasConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateAtlasConfigs"); Method updateLogSearchConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateLogSearchConfigs"); Method updateLogSearchAlert = UpgradeCatalog250.class.getDeclaredMethod("updateLogSearchAlert"); Method updateAmbariInfraConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateAmbariInfraConfigs"); Method updateRangerUrlConfigs = UpgradeCatalog250.class.getDeclaredMethod("updateRangerUrlConfigs"); Method updateTezHistoryUrlBase = UpgradeCatalog250.class.getDeclaredMethod("updateTezHistoryUrlBase"); Method updateYarnSite = UpgradeCatalog250.class.getDeclaredMethod("updateYarnSite"); Method updateAlerts = UpgradeCatalog250.class.getDeclaredMethod("updateStormAlerts"); Method removeAlertDuplicates = UpgradeCatalog250.class.getDeclaredMethod("removeAlertDuplicates"); Method updateKerberosDescriptorArtifacts = AbstractUpgradeCatalog.class.getDeclaredMethod("updateKerberosDescriptorArtifacts"); Method fixHBaseMasterCPUUtilizationAlertDefinition = UpgradeCatalog250.class.getDeclaredMethod("fixHBaseMasterCPUUtilizationAlertDefinition"); UpgradeCatalog250 upgradeCatalog250 = createMockBuilder(UpgradeCatalog250.class) .addMockedMethod(updateAmsConfigs) .addMockedMethod(updateHadoopEnvConfigs) .addMockedMethod(updateKafkaConfigs) .addMockedMethod(updateHiveLlapConfigs) .addMockedMethod(addNewConfigurationsFromXml) .addMockedMethod(addManageServiceAutoStartPermissions) .addMockedMethod(addManageAlertNotificationsPermissions) .addMockedMethod(updateHIVEInteractiveConfigs) .addMockedMethod(updateTablesForZeppelinViewRemoval) .addMockedMethod(updateZeppelinConfigs) .addMockedMethod(updateAtlasConfigs) .addMockedMethod(updateLogSearchConfigs) .addMockedMethod(updateAmbariInfraConfigs) .addMockedMethod(updateRangerUrlConfigs) .addMockedMethod(updateYarnSite) .addMockedMethod(updateAlerts) .addMockedMethod(updateLogSearchAlert) .addMockedMethod(removeAlertDuplicates) .addMockedMethod(updateKerberosDescriptorArtifacts) .addMockedMethod(fixHBaseMasterCPUUtilizationAlertDefinition) .addMockedMethod(updateTezHistoryUrlBase) .createMock(); upgradeCatalog250.updateAMSConfigs(); expectLastCall().once(); upgradeCatalog250.updateHadoopEnvConfigs(); expectLastCall().once(); upgradeCatalog250.addNewConfigurationsFromXml(); expectLastCall().once(); upgradeCatalog250.updateKafkaConfigs(); expectLastCall().once(); upgradeCatalog250.updateHIVEInteractiveConfigs(); expectLastCall().once(); upgradeCatalog250.updateHiveLlapConfigs(); expectLastCall().once(); upgradeCatalog250.updateTablesForZeppelinViewRemoval(); expectLastCall().once(); upgradeCatalog250.updateZeppelinConfigs(); expectLastCall().once(); upgradeCatalog250.updateAtlasConfigs(); expectLastCall().once(); upgradeCatalog250.updateLogSearchConfigs(); expectLastCall().once(); upgradeCatalog250.updateAmbariInfraConfigs(); expectLastCall().once(); upgradeCatalog250.updateRangerUrlConfigs(); expectLastCall().once(); upgradeCatalog250.addManageServiceAutoStartPermissions(); expectLastCall().once(); upgradeCatalog250.addManageAlertNotificationsPermissions(); expectLastCall().once(); upgradeCatalog250.updateTezHistoryUrlBase(); expectLastCall().once(); upgradeCatalog250.updateYarnSite(); expectLastCall().once(); upgradeCatalog250.updateStormAlerts(); expectLastCall().once(); upgradeCatalog250.updateLogSearchAlert(); expectLastCall().once(); upgradeCatalog250.removeAlertDuplicates(); expectLastCall().once(); upgradeCatalog250.updateKerberosDescriptorArtifacts(); expectLastCall().once(); upgradeCatalog250.fixHBaseMasterCPUUtilizationAlertDefinition(); expectLastCall().once(); replay(upgradeCatalog250); upgradeCatalog250.executeDMLUpdates(); verify(upgradeCatalog250); }