@BeforeClass public static void beforeAll() throws IOException { conf = new TransportConf("shuffle", MapConfigProvider.EMPTY); context = new TransportContext(conf, new TestRpcHandler()); secretKeyHolder = mock(SecretKeyHolder.class); when(secretKeyHolder.getSaslUser(eq("app-1"))).thenReturn("app-1"); when(secretKeyHolder.getSecretKey(eq("app-1"))).thenReturn("app-1"); when(secretKeyHolder.getSaslUser(eq("app-2"))).thenReturn("app-2"); when(secretKeyHolder.getSecretKey(eq("app-2"))).thenReturn("app-2"); when(secretKeyHolder.getSaslUser(anyString())).thenReturn("other-app"); when(secretKeyHolder.getSecretKey(anyString())).thenReturn("correct-password"); TransportServerBootstrap bootstrap = new SaslServerBootstrap(conf, secretKeyHolder); server = context.createServer(Arrays.asList(bootstrap)); }
@BeforeClass public static void beforeAll() throws IOException { conf = new TransportConf("shuffle", MapConfigProvider.EMPTY); context = new TransportContext(conf, new TestRpcHandler()); secretKeyHolder = mock(SecretKeyHolder.class); when(secretKeyHolder.getSaslUser(eq("app-1"))).thenReturn("app-1"); when(secretKeyHolder.getSecretKey(eq("app-1"))).thenReturn("app-1"); when(secretKeyHolder.getSaslUser(eq("app-2"))).thenReturn("app-2"); when(secretKeyHolder.getSecretKey(eq("app-2"))).thenReturn("app-2"); when(secretKeyHolder.getSaslUser(anyString())).thenReturn("other-app"); when(secretKeyHolder.getSecretKey(anyString())).thenReturn("correct-password"); TransportServerBootstrap bootstrap = new SaslServerBootstrap(conf, secretKeyHolder); server = context.createServer(Arrays.asList(bootstrap)); }
@BeforeClass public static void beforeAll() throws IOException { conf = new TransportConf("shuffle", MapConfigProvider.EMPTY); context = new TransportContext(conf, new TestRpcHandler()); secretKeyHolder = mock(SecretKeyHolder.class); when(secretKeyHolder.getSaslUser(eq("app-1"))).thenReturn("app-1"); when(secretKeyHolder.getSecretKey(eq("app-1"))).thenReturn("app-1"); when(secretKeyHolder.getSaslUser(eq("app-2"))).thenReturn("app-2"); when(secretKeyHolder.getSecretKey(eq("app-2"))).thenReturn("app-2"); when(secretKeyHolder.getSaslUser(anyString())).thenReturn("other-app"); when(secretKeyHolder.getSecretKey(anyString())).thenReturn("correct-password"); TransportServerBootstrap bootstrap = new SaslServerBootstrap(conf, secretKeyHolder); server = context.createServer(Arrays.asList(bootstrap)); }
@Test public void testNoSaslServer() { RpcHandler handler = new TestRpcHandler(); TransportContext context = new TransportContext(conf, handler); clientFactory = context.createClientFactory( Arrays.asList(new SaslClientBootstrap(conf, "app-1", secretKeyHolder))); try (TransportServer server = context.createServer()) { clientFactory.createClient(TestUtils.getLocalHost(), server.getPort()); } catch (Exception e) { assertTrue(e.getMessage(), e.getMessage().contains("Digest-challenge format violation")); } }
@Test public void testNoSaslServer() { RpcHandler handler = new TestRpcHandler(); TransportContext context = new TransportContext(conf, handler); clientFactory = context.createClientFactory( Arrays.asList(new SaslClientBootstrap(conf, "app-1", secretKeyHolder))); try (TransportServer server = context.createServer()) { clientFactory.createClient(TestUtils.getLocalHost(), server.getPort()); } catch (Exception e) { assertTrue(e.getMessage(), e.getMessage().contains("Digest-challenge format violation")); } }
@Test public void testNoSaslServer() { RpcHandler handler = new TestRpcHandler(); TransportContext context = new TransportContext(conf, handler); clientFactory = context.createClientFactory( Arrays.asList(new SaslClientBootstrap(conf, "app-1", secretKeyHolder))); try (TransportServer server = context.createServer()) { clientFactory.createClient(TestUtils.getLocalHost(), server.getPort()); } catch (Exception e) { assertTrue(e.getMessage(), e.getMessage().contains("Digest-challenge format violation")); } }