SessionContext sessionContext = new HttpRequestSessionContext(servletRequest);
SessionContext sessionContext = new HttpRequestSessionContext(servletRequest);
@Test public void testEmptyClientTags() { HttpServletRequest request1 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .build(), "remoteAddress"); HttpRequestSessionContext context1 = new HttpRequestSessionContext(request1); assertEquals(context1.getClientTags(), ImmutableSet.of()); HttpServletRequest request2 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_CLIENT_TAGS, "") .build(), "remoteAddress"); HttpRequestSessionContext context2 = new HttpRequestSessionContext(request2); assertEquals(context2.getClientTags(), ImmutableSet.of()); }
@Test public void testClientCapabilities() { HttpServletRequest request1 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_CLIENT_CAPABILITIES, "foo, bar") .build(), "remoteAddress"); HttpRequestSessionContext context1 = new HttpRequestSessionContext(request1); assertEquals(context1.getClientCapabilities(), ImmutableSet.of("foo", "bar")); HttpServletRequest request2 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .build(), "remoteAddress"); HttpRequestSessionContext context2 = new HttpRequestSessionContext(request2); assertEquals(context2.getClientCapabilities(), ImmutableSet.of()); }
@Test public void testClientCapabilities() { HttpServletRequest request1 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_CLIENT_CAPABILITIES, "foo, bar") .build(), "remoteAddress"); HttpRequestSessionContext context1 = new HttpRequestSessionContext(request1); assertEquals(context1.getClientCapabilities(), ImmutableSet.of("foo", "bar")); HttpServletRequest request2 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .build(), "remoteAddress"); HttpRequestSessionContext context2 = new HttpRequestSessionContext(request2); assertEquals(context2.getClientCapabilities(), ImmutableSet.of()); }
@Test public void testEmptyClientTags() { HttpServletRequest request1 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .build(), "remoteAddress"); HttpRequestSessionContext context1 = new HttpRequestSessionContext(request1); assertEquals(context1.getClientTags(), ImmutableSet.of()); HttpServletRequest request2 = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_CLIENT_TAGS, "") .build(), "remoteAddress"); HttpRequestSessionContext context2 = new HttpRequestSessionContext(request2); assertEquals(context2.getClientTags(), ImmutableSet.of()); }
@Test public void testCreateSession() HttpRequestSessionContext context = new HttpRequestSessionContext(TEST_REQUEST); QuerySessionSupplier sessionSupplier = new QuerySessionSupplier( createTestTransactionManager(),
@Test public void testCreateSession() HttpRequestSessionContext context = new HttpRequestSessionContext(TEST_REQUEST); QuerySessionSupplier sessionSupplier = new QuerySessionSupplier( createTestTransactionManager(),
"testRemote"); HttpRequestSessionContext context = new HttpRequestSessionContext(request); assertEquals(context.getSource(), "testSource"); assertEquals(context.getCatalog(), "testCatalog");
"testRemote"); HttpRequestSessionContext context = new HttpRequestSessionContext(request); assertEquals(context.getSource(), "testSource"); assertEquals(context.getCatalog(), "testCatalog");
@Test(expectedExceptions = WebApplicationException.class) public void testPreparedStatementsHeaderDoesNotParse() { HttpServletRequest request = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_SOURCE, "testSource") .put(PRESTO_CATALOG, "testCatalog") .put(PRESTO_SCHEMA, "testSchema") .put(PRESTO_PATH, "testPath") .put(PRESTO_LANGUAGE, "zh-TW") .put(PRESTO_TIME_ZONE, "Asia/Taipei") .put(PRESTO_CLIENT_INFO, "null") .put(PRESTO_PREPARED_STATEMENT, "query1=abcdefg") .build(), "testRemote"); new HttpRequestSessionContext(request); } }
@Test(expectedExceptions = WebApplicationException.class) public void testPreparedStatementsHeaderDoesNotParse() { HttpServletRequest request = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_SOURCE, "testSource") .put(PRESTO_CATALOG, "testCatalog") .put(PRESTO_SCHEMA, "testSchema") .put(PRESTO_PATH, "testPath") .put(PRESTO_LANGUAGE, "zh-TW") .put(PRESTO_TIME_ZONE, "Asia/Taipei") .put(PRESTO_CLIENT_INFO, "null") .put(PRESTO_PREPARED_STATEMENT, "query1=abcdefg") .build(), "testRemote"); new HttpRequestSessionContext(request); } }
@Test(expectedExceptions = PrestoException.class) public void testInvalidTimeZone() { HttpServletRequest request = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_TIME_ZONE, "unknown_timezone") .build(), "testRemote"); HttpRequestSessionContext context = new HttpRequestSessionContext(request); QuerySessionSupplier sessionSupplier = new QuerySessionSupplier( createTestTransactionManager(), new AllowAllAccessControl(), new SessionPropertyManager(), new SqlEnvironmentConfig()); sessionSupplier.createSession(new QueryId("test_query_id"), context); }
@Test(expectedExceptions = PrestoException.class) public void testInvalidTimeZone() { HttpServletRequest request = new MockHttpServletRequest( ImmutableListMultimap.<String, String>builder() .put(PRESTO_USER, "testUser") .put(PRESTO_TIME_ZONE, "unknown_timezone") .build(), "testRemote"); HttpRequestSessionContext context = new HttpRequestSessionContext(request); QuerySessionSupplier sessionSupplier = new QuerySessionSupplier( createTestTransactionManager(), new AllowAllAccessControl(), new SessionPropertyManager(), new SqlEnvironmentConfig()); sessionSupplier.createSession(new QueryId("test_query_id"), context); }