use normalised state in e2e test

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

tests/e2e/test_driver.py

@@ -30,6 +30,7 @@
     OperationalError,
     RequestError,
 )
+from databricks.sql.backend.types import CommandState
 from tests.e2e.common.predicates import (
     pysql_has_version,
     pysql_supports_arrow,
@@ -939,7 +940,7 @@ def test_result_set_close(self):
 
                 result_set.close()
 
-                assert result_set.op_state == result_set.backend.CLOSED_OP_STATE
+                assert result_set.op_state == CommandState.CLOSED
                 assert result_set.op_state != initial_op_state
 
                 # Closing the result set again should be a no-op and not raise exceptions