raise ProgrammingError for invalid id

varun-edachali-dbx · varun-edachali-dbx · commit a026d31007cf · 2025-06-26T10:33:22.000Z
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/unit/test_sea_backend.py b/tests/unit/test_sea_backend.py
@@ -486,7 +486,7 @@ def test_command_management(
         )
 
         # Test get_query_state with invalid ID
-        with pytest.raises(ValueError) as excinfo:
+        with pytest.raises(ProgrammingError) as excinfo:
             sea_client.get_query_state(thrift_command_id)
         assert "Not a valid SEA command ID" in str(excinfo.value)
 

Original file line number	Diff line number	Diff line change
`@@ -486,7 +486,7 @@ def test_command_management(`
`486`	`486`	`)`
`487`	`487`
`488`	`488`	`# Test get_query_state with invalid ID`
`489`		`- with pytest.raises(ValueError) as excinfo:`
	`489`	`+ with pytest.raises(ProgrammingError) as excinfo:`
`490`	`490`	`sea_client.get_query_state(thrift_command_id)`
`491`	`491`	`assert "Not a valid SEA command ID" in str(excinfo.value)`
`492`	`492`