Commitd62cabb

authored

feat: support slot_ms in QueryPlanEntry (#1831)

1 parenta208b6a commitd62cabbCopy full SHA for d62cabb

File tree

-0

lines changed

-0

lines changed

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -2263,6 +2263,11 @@ def steps(self):`
`2263`	`2263`	`forstepinself._properties.get("steps", [])`
`2264`	`2264`	`]`
`2265`	`2265`
	`2266`	`+@property`
	`2267`	`+defslot_ms(self):`
	`2268`	`+"""Optional[int]: Slot-milliseconds used by the stage."""`
	`2269`	`+return_helpers._int_or_none(self._properties.get("slotMs"))`
	`2270`	`+`
`2266`	`2271`
`2267`	`2272`	`classTimelineEntry(object):`
`2268`	`2273`	`"""TimelineEntry represents progress of a query job at a particular`

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -261,6 +261,7 @@ class TestQueryPlanEntry(_Base):`
`261`	`261`	`STATUS="STATUS"`
`262`	`262`	`SHUFFLE_OUTPUT_BYTES=1024`
`263`	`263`	`SHUFFLE_OUTPUT_BYTES_SPILLED=1`
	`264`	`+SLOT_MS=25`
`264`	`265`
`265`	`266`	`START_RFC3339_MICROS="2018-04-01T00:00:00.000000Z"`
`266`	`267`	`END_RFC3339_MICROS="2018-04-01T00:00:04.000000Z"`
`@@ -305,6 +306,7 @@ def test_from_api_repr_empty(self):`
`305`	`306`	`self.assertIsNone(entry.shuffle_output_bytes)`
`306`	`307`	`self.assertIsNone(entry.shuffle_output_bytes_spilled)`
`307`	`308`	`self.assertEqual(entry.steps, [])`
	`309`	`+self.assertIsNone(entry.slot_ms)`
`308`	`310`
`309`	`311`	`deftest_from_api_repr_normal(self):`
`310`	`312`	`fromgoogle.cloud.bigquery.jobimportQueryPlanEntryStep`
`@@ -348,6 +350,7 @@ def test_from_api_repr_normal(self):`
`348`	`350`	`"substeps":TestQueryPlanEntryStep.SUBSTEPS,`
`349`	`351`	`}`
`350`	`352`	`],`
	`353`	`+"slotMs":self.SLOT_MS,`
`351`	`354`	`}`
`352`	`355`	`klass=self._get_target_class()`
`353`	`356`
`@@ -366,6 +369,7 @@ def test_from_api_repr_normal(self):`
`366`	`369`	`self.assertEqual(entry.records_written,self.RECORDS_WRITTEN)`
`367`	`370`	`self.assertEqual(entry.status,self.STATUS)`
`368`	`371`	`self.assertEqual(entry.steps,steps)`
	`372`	`+self.assertEqual(entry.slot_ms,self.SLOT_MS)`
`369`	`373`
`370`	`374`	`deftest_start(self):`
`371`	`375`	`fromgoogle.cloud._helpersimport_RFC3339_MICROS`

Comments

(0)