| 89 | import static org.junit.Assert.fail; |
| 90 | |
| 91 | public class SqlStatementTest |
| 92 | { |
| 93 | private static QueryRunnerFactoryConglomerate conglomerate; |
| 94 | private static SpecificSegmentsQuerySegmentWalker walker; |
| 95 | private static Closer resourceCloser; |
| 96 | @ClassRule |
| 97 | public static TemporaryFolder temporaryFolder = new TemporaryFolder(); |
| 98 | private ListeningExecutorService executorService; |
| 99 | private final DefaultQueryConfig defaultQueryConfig = new DefaultQueryConfig( |
| 100 | ImmutableMap.of("DEFAULT_KEY", "DEFAULT_VALUE")); |
| 101 | |
| 102 | private PolicyEnforcer policyEnforcer; |
| 103 | private SqlStatementFactory sqlStatementFactory; |
| 104 | |
| 105 | @BeforeClass |
| 106 | public static void setUpClass() throws Exception |
| 107 | { |
| 108 | resourceCloser = Closer.create(); |
| 109 | conglomerate = QueryStackTests.createQueryRunnerFactoryConglomerate(resourceCloser); |
| 110 | |
| 111 | final QueryScheduler scheduler = new QueryScheduler( |
| 112 | 5, |
| 113 | ManualQueryPrioritizationStrategy.INSTANCE, |
| 114 | new HiLoQueryLaningStrategy(40), |
| 115 | new ServerConfig() |
| 116 | ) |
| 117 | { |
| 118 | @Override |
| 119 | public <T> Sequence<T> run(Query<?> query, Sequence<T> resultSequence) |
| 120 | { |
| 121 | return super.run( |
| 122 | query, |
| 123 | new LazySequence<>(() -> resultSequence) |
| 124 | ); |
| 125 | } |
| 126 | }; |
| 127 | |
| 128 | walker = CalciteTests.createMockWalker(conglomerate, temporaryFolder.newFolder(), scheduler); |
| 129 | resourceCloser.register(walker); |
| 130 | } |
| 131 | |
| 132 | @AfterClass |
| 133 | public static void tearDownClass() throws IOException |
| 134 | { |
| 135 | resourceCloser.close(); |
| 136 | } |
| 137 | |
| 138 | @Before |
| 139 | public void setUp() |
| 140 | { |
| 141 | executorService = MoreExecutors.listeningDecorator(Execs.multiThreaded(8, "test_sql_resource_%s")); |
| 142 | |
| 143 | policyEnforcer = NoopPolicyEnforcer.instance(); |
| 144 | this.sqlStatementFactory = buildSqlStatementFactory(); |
| 145 | } |
| 146 | |
| 147 | @After |
| 148 | public void tearDown() throws Exception |