| 123456789101112131415161718192021222324252627282930313233343536373839404142 |
- from pathlib import Path
- p = Path(__file__).parent / 'pipe-network-service/zksy-admin/src/test/java/com/zksy/web/controller/cockpit/CockpitMonitoringDistributionControllerTest.java'
- s = p.read_text(encoding='utf-8')
- def replace(old, new):
- global s
- assert old in s, old
- s = s.replace(old, new)
- replace('distributionExposesEnvelopeSixStateDistributionDistrictIndustryTypeAndMapFilter', 'distributionExposesEnvelopeGlobalEightBucketsAndRetainsNonRenderableObjects')
- replace('value(8)', 'value(12)')
- replace('"$.data.monitoringPoints", hasSize(5)', '"$.data.monitoringPoints", hasSize(7)')
- replace('"$.data.unassociatedEquipment", hasSize(3)', '"$.data.unassociatedEquipment", hasSize(5)')
- replace('"$.data.pointStatusDistribution", hasSize(6)', '"$.data.pointStatusDistribution", hasSize(8)')
- replace('monitoringPoints[3]', 'monitoringPoints[5]')
- replace('monitoringPoints[4]', 'monitoringPoints[6]')
- replace('"$.data.pointStatusDistribution[1].count").value(2)', '"$.data.pointStatusDistribution[1].count").value(1)')
- replace('"$.data.pointStatusDistribution[3].count").value(1)', '"$.data.pointStatusDistribution[3].count").value(2)')
- replace('"$.data.pointStatusDistribution[4].count").value(2)', '"$.data.pointStatusDistribution[4].count").value(1)')
- replace('"$.data.districtDistribution[0].pointCount").value(2)', '"$.data.districtDistribution[0].pointCount").value(4)')
- replace('"$.data.districtDistribution[0].deviceCount").value(4)', '"$.data.districtDistribution[0].deviceCount").value(8)')
- replace('"$.data.industryDistribution", hasSize(3)', '"$.data.industryDistribution", hasSize(4)')
- replace('"$.data.industryDistribution[0].deviceCount").value(4)', '"$.data.industryDistribution[0].deviceCount").value(5)')
- replace('"$.data.deviceTypeDistribution", hasSize(3)', '"$.data.deviceTypeDistribution", hasSize(6)')
- replace('"$.data.deviceTypeDistribution[0].deviceCount").value(4)', '"$.data.deviceTypeDistribution[0].deviceCount").value(5)')
- replace('"$.data.mapFilter.pointCodes", hasSize(5)', '"$.data.mapFilter.pointCodes", hasSize(7)')
- replace('"$.data.mapFilter.deviceCodes", hasSize(8)', '"$.data.mapFilter.deviceCodes", hasSize(12)')
- for key in ('unknownTypeCount', 'typeKindMismatchCount', 'nonFormalIndustryCount'):
- replace(f'"$.data.excluded.{key}").value(1)', f'"$.data.excluded.{key}").value(0)')
- replace('"$.data.pointStatusDistribution[0].count").value(0)', '"$.data.pointStatusDistribution[0].count").value(1)')
- replace('"$.data.pointStatusDistribution[4].count").value(0)', '"$.data.pointStatusDistribution[4].count").value(1)')
- replace('"NO_RESOLVED_DEVICES"', '"NO_OBJECTS"')
- replace('pointStatus must be ALARM, WARNING, WORK_ORDER, NORMAL, OFFLINE or EMPTY', 'pointStatus must be ALARM, WARNING, WORK_ORDER, NORMAL, OFFLINE, EMPTY, UNKNOWN or UNASSOCIATED')
- # Narrowed facet result sizes changed because coordinate validation no longer drops devices.
- for name in ('industryFilterNarrowsDistributionAndMapFilter', 'deviceTypeFilterNarrowsDistribution'):
- start = s.index(' void ' + name)
- end = s.index(' @Test', start)
- block = s[start:end].replace('pagination.total").value(3)', 'pagination.total").value(4)').replace('unassociatedEquipment", hasSize(1)', 'unassociatedEquipment", hasSize(2)')
- s = s[:start] + block + s[end:]
- # Cross-endpoint global distribution counts only points in six-state buckets.
- start = s.index(' void webSocketDeltaRecomputesStatusCounts')
- s = s[:start] + s[start:].replace('assertEquals(2, statusCount(initial, CockpitMonitoringPointStatus.OFFLINE))', 'assertEquals(1, statusCount(initial, CockpitMonitoringPointStatus.OFFLINE))').replace('assertEquals(1, statusCount(initial, CockpitMonitoringPointStatus.NORMAL))', 'assertEquals(2, statusCount(initial, CockpitMonitoringPointStatus.NORMAL))').replace('assertEquals(1, statusCount(applied, CockpitMonitoringPointStatus.OFFLINE))', 'assertEquals(0, statusCount(applied, CockpitMonitoringPointStatus.OFFLINE))').replace('assertEquals(2, statusCount(applied, CockpitMonitoringPointStatus.NORMAL))', 'assertEquals(3, statusCount(applied, CockpitMonitoringPointStatus.NORMAL))').replace('assertEquals(1, statusCount(ignoredOld, CockpitMonitoringPointStatus.OFFLINE))', 'assertEquals(0, statusCount(ignoredOld, CockpitMonitoringPointStatus.OFFLINE))').replace('assertEquals(2, statusCount(ignoredOld, CockpitMonitoringPointStatus.NORMAL))', 'assertEquals(3, statusCount(ignoredOld, CockpitMonitoringPointStatus.NORMAL))').replace('assertEquals(1, statusCount(ignoredUnknown, CockpitMonitoringPointStatus.OFFLINE))', 'assertEquals(0, statusCount(ignoredUnknown, CockpitMonitoringPointStatus.OFFLINE))').replace('assertEquals(2, statusCount(ignoredUnknown, CockpitMonitoringPointStatus.NORMAL))', 'assertEquals(3, statusCount(ignoredUnknown, CockpitMonitoringPointStatus.NORMAL))')
- p.write_text(s, encoding='utf-8', newline='\n')
- print('Updated existing distribution assertions to global counting contract.')
|
