private void assertSetRole(String statement, Map<String, SelectedRole> expected) { SetRole setRole = (SetRole) parser.createStatement(statement); QueryStateMachine stateMachine = QueryStateMachine.begin( statement, testSessionBuilder() .setCatalog(CATALOG_NAME) .build(), URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); new SetRoleTask().execute(setRole, transactionManager, metadata, accessControl, stateMachine, ImmutableList.of()); QueryInfo queryInfo = stateMachine.getQueryInfo(Optional.empty()); assertEquals(queryInfo.getSetRoles(), expected); } }
private void assertSetRole(String statement, Map<String, SelectedRole> expected) { SetRole setRole = (SetRole) parser.createStatement(statement); QueryStateMachine stateMachine = QueryStateMachine.begin( statement, testSessionBuilder() .setCatalog(CATALOG_NAME) .build(), URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); new SetRoleTask().execute(setRole, transactionManager, metadata, accessControl, stateMachine, ImmutableList.of()); QueryInfo queryInfo = stateMachine.getQueryInfo(Optional.empty()); assertEquals(queryInfo.getSetRoles(), expected); } }
setRoles = queryInfo.getSetRoles();
setRoles = queryInfo.getSetRoles();
queryInfo.getSetSessionProperties(), queryInfo.getResetSessionProperties(), queryInfo.getSetRoles(), queryInfo.getAddedPreparedStatements(), queryInfo.getDeallocatedPreparedStatements(),
queryInfo.getSetSessionProperties(), queryInfo.getResetSessionProperties(), queryInfo.getSetRoles(), queryInfo.getAddedPreparedStatements(), queryInfo.getDeallocatedPreparedStatements(),