org.mockito.internal.util.io.IOUtil Java Examples

@Test
public void testStartNotificationFailsCorrectKeystorePasswdBlankKeypasswd() throws ParserConfigurationException, SAXException, IOException {
    Logger notificationServiceLogger = (Logger) LoggerFactory.getLogger(NotificationServiceManager.class);
    ListAppender<ILoggingEvent> listAppender = new ListAppender<>();
    listAppender.start();
    notificationServiceLogger.addAppender(listAppender);

    String configFileOutput = CONFIGURATION_FILE_TEXT_BLANK_KEY_PASSWORD.replace("${test.server}", String.valueOf(mockWebServer.url("/")));
    IOUtil.writeText(configFileOutput, new File(tempConfigFilePath));

    NotificationServiceManager notificationServiceManager = new NotificationServiceManager();
    notificationServiceManager.setMaxNotificationAttempts(1);
    notificationServiceManager.loadNotificationServices(new File(tempConfigFilePath));

    List<ILoggingEvent> logsList = listAppender.list;
    boolean notificationServiceFailed = false;
    for (ILoggingEvent logMessage : logsList) {
        if (logMessage.getFormattedMessage().contains("'Key Password' validated against '' is invalid because Key Password cannot be empty")) {
            notificationServiceFailed = true;
        }
    }

    assertTrue(notificationServiceFailed);
}
 

@Test
public void testStartNotificationFailsBlankKeystorePasswdCorrectKeypasswd() throws ParserConfigurationException, SAXException, IOException {
    Logger notificationServiceLogger = (Logger) LoggerFactory.getLogger(NotificationServiceManager.class);
    ListAppender<ILoggingEvent> listAppender = new ListAppender<>();
    listAppender.start();
    notificationServiceLogger.addAppender(listAppender);

    String configFileOutput = CONFIGURATION_FILE_TEXT_BLANK_KEYSTORE_PASSWORD.replace("${test.server}", String.valueOf(mockWebServer.url("/")));
    IOUtil.writeText(configFileOutput, new File(tempConfigFilePath));

    NotificationServiceManager notificationServiceManager = new NotificationServiceManager();
    notificationServiceManager.setMaxNotificationAttempts(1);
    notificationServiceManager.loadNotificationServices(new File(tempConfigFilePath));

    List<ILoggingEvent> logsList = listAppender.list;
    boolean notificationServiceFailed = false;
    for (ILoggingEvent logMessage : logsList) {
        if (logMessage.getFormattedMessage().contains("'Keystore Password' validated against '' is invalid because Keystore Password cannot be empty")) {
            notificationServiceFailed = true;
        }
    }

    assertTrue(notificationServiceFailed);
}
 

@Test
public void testStartNotificationSucceedsNoKeyPasswd() throws ParserConfigurationException, SAXException, IOException {
    Logger notificationServiceLogger = (Logger) LoggerFactory.getLogger(NotificationServiceManager.class);
    ListAppender<ILoggingEvent> listAppender = new ListAppender<>();
    listAppender.start();
    notificationServiceLogger.addAppender(listAppender);

    String configFileOutput = CONFIGURATION_FILE_TEXT_NO_KEY_PASSWORD.replace("${test.server}", String.valueOf(mockWebServer.url("/")));
    IOUtil.writeText(configFileOutput, new File(tempConfigFilePath));

    NotificationServiceManager notificationServiceManager = new NotificationServiceManager();
    notificationServiceManager.setMaxNotificationAttempts(1);
    notificationServiceManager.loadNotificationServices(new File(tempConfigFilePath));

    List<ILoggingEvent> logsList = listAppender.list;
    boolean notificationServiceFailed = false;
    for (ILoggingEvent logMessage : logsList) {
        if (logMessage.getFormattedMessage().contains("is not valid for the following reasons")) {
            notificationServiceFailed = true;
        }
    }

    assertFalse(notificationServiceFailed);
}
 

@Test
public void testStartNotificationSucceedsNoKeystorePasswd() throws ParserConfigurationException, SAXException, IOException {
    Logger notificationServiceLogger = (Logger) LoggerFactory.getLogger(NotificationServiceManager.class);
    ListAppender<ILoggingEvent> listAppender = new ListAppender<>();
    listAppender.start();
    notificationServiceLogger.addAppender(listAppender);

    String configFileOutput = CONFIGURATION_FILE_TEXT_NO_KEYSTORE_PASSWORD.replace("${test.server}", String.valueOf(mockWebServer.url("/")));
    IOUtil.writeText(configFileOutput, new File(tempConfigFilePath));

    NotificationServiceManager notificationServiceManager = new NotificationServiceManager();
    notificationServiceManager.setMaxNotificationAttempts(1);
    notificationServiceManager.loadNotificationServices(new File(tempConfigFilePath));

    List<ILoggingEvent> logsList = listAppender.list;
    boolean notificationServiceFailed = false;
    for (ILoggingEvent logMessage : logsList) {
        if (logMessage.getFormattedMessage().contains("is not valid for the following reasons")) {
            notificationServiceFailed = true;
        }
    }

    assertFalse(notificationServiceFailed);
}
 

@Before
public void startServer() throws IOException, TlsException {
    tempConfigFilePath = "./target/TestHttpNotificationService-config.xml";

    Files.deleteIfExists(Paths.get(tempConfigFilePath));

    mockWebServer = new MockWebServer();

    TlsConfiguration tlsConfiguration = new TlsConfiguration("./src/test/resources/keystore.jks", "passwordpassword", null, "JKS",
            "./src/test/resources/truststore.jks", "passwordpassword", "JKS", CertificateUtils.getHighestCurrentSupportedTlsProtocolVersion());
    final SSLContext sslContext = SslContextFactory.createSslContext(tlsConfiguration, SslContextFactory.ClientAuth.REQUIRED);
    mockWebServer.useHttps(sslContext.getSocketFactory(), false);

    String configFileOutput = CONFIGURATION_FILE_TEXT.replace("${test.server}", String.valueOf(mockWebServer.url("/")));
    IOUtil.writeText(configFileOutput, new File(tempConfigFilePath));
}
 

@Test
public void portDetect() throws UnknownHostException, IOException {
	int port = NetUtil.findRandomAvailablePort(20000, 20100);
	assertThat(port).isBetween(20000, 20100);
	System.out.println("random port:" + port);

	assertThat(NetUtil.isPortAvailable(port)).isTrue();

	int port2 = NetUtil.findAvailablePortFrom(port);
	assertThat(port2).isEqualTo(port);

	int port3 = NetUtil.findRandomAvailablePort();

	assertThat(port3).isBetween(NetUtil.PORT_RANGE_MIN, NetUtil.PORT_RANGE_MAX);
	System.out.println("random port:" + port3);

	// 尝试占住一个端口
	ServerSocket serverSocket = null;
	try {
		serverSocket = ServerSocketFactory.getDefault().createServerSocket(port, 1,
				InetAddress.getByName("localhost"));

		assertThat(NetUtil.isPortAvailable(port)).isFalse();

		int port4 = NetUtil.findAvailablePortFrom(port);
		assertThat(port4).isEqualTo(port + 1);

		try {
			int port5 = NetUtil.findRandomAvailablePort(port, port);
			fail("should fail before");
		} catch (Throwable t) {
			assertThat(t).isInstanceOf(IllegalStateException.class);
		}

	} finally {
		IOUtil.close(serverSocket);
	}

}
 

@BeforeClass
public static void startServer() throws IOException {
    tempConfigFilePath = "./target/TestHttpNotificationService-config.xml";

    Files.deleteIfExists(Paths.get(tempConfigFilePath));

    mockWebServer = new MockWebServer();

    String configFileOutput = CONFIGURATION_FILE_TEXT.replace("${test.server}", String.valueOf(mockWebServer.url("/")));
    IOUtil.writeText(configFileOutput, new File(tempConfigFilePath));
}
 

@Override
public void traverse() {
    String readFilters = getCommandLineParser()
            .getPluginDescriptor(GATKReadFilterPluginDescriptor.class)
            .getResolvedInstances()
            .stream()
            .map(a -> a.getClass().getSimpleName())
            .collect(Collectors.joining("\n"));
    IOUtil.writeText(readFilters, new File(output));
}
 

@Override
public void traverse() {
    String annotations = makeVariantAnnotations().stream()
            .map(a -> a.getClass().getSimpleName())
            .collect(Collectors.joining("\n"));
    IOUtil.writeText(annotations, new File(output));
}
 

@Before
public void createCredentials() throws IOException {
	File tempDir = tempFolder.newFolder();
	File clientJson = new File(tempDir, "client.json");
	IOUtil.writeText("{\"client_id\":\"abcdefg \",\"client_secret\":\"geheim\"}", clientJson);
	File userJson = new File(tempDir, "user.json");
	IOUtil.writeText("{\"application_username\":\"klaus \",\"application_password\":\"geheim\"}", userJson);
	System.setProperty(CREDENTIALS_DIR, tempDir.getAbsolutePath());
}
 

public void run() {
    try {
        try {
            if (!cmd.hasOption(SILENT_OPTION)) {
                System.out.printf("%sLoading schema from %s:%s%s", ANSI_RED, host, port, ANSI_RESET);
                System.out.flush();
            }
            String cfids = cmd.getOptionValue(CFID_OPTION, "");
            cluster = CassandraUtils.loadTablesFromRemote(host, port, cfids);
        } finally {
            System.out.print(ANSI_RESET);
        }
            for (String target : cmd.getArgs()) {
                File targetFile = new File(target);
                if (!targetFile.exists()) {
                    if (!cmd.hasOption(SILENT_OPTION)) {
                        System.out.print("\r" + ANSI_RESET + Strings.repeat(" ", 60) + "\r");
                    }
                    System.err.println("Non-existant target provided: " + target);
                } else {
                    if (!cmd.hasOption(SILENT_OPTION)) {
                        System.out.print("\r" + ANSI_RESET + Strings.repeat(" ", 60) + "\r");
                        System.out.println("\u001B[1;34m" + targetFile.getAbsolutePath());
                        System.out.println(ANSI_CYAN + Strings.repeat("=", targetFile.getAbsolutePath().length()));
                        System.out.print(ANSI_RESET);
                    }
                    MutationReplayer replayer = null;
                    if (cmd.hasOption(REPLAY_OPTION)) {
                        replayer = new MutationReplayer(cluster);
                    }
                    walkMutations(targetFile, System.out, replayer);
                }
            }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        IOUtil.closeQuietly(cluster);
    }
}
 

@Test
public void portDetect() throws UnknownHostException, IOException {
	int port = NetUtil.findRandomAvailablePort(20000, 20100);
	assertThat(port).isBetween(20000, 20100);
	System.out.println("random port:" + port);

	assertThat(NetUtil.isPortAvailable(port)).isTrue();

	int port2 = NetUtil.findAvailablePortFrom(port);
	assertThat(port2).isEqualTo(port);

	int port3 = NetUtil.findRandomAvailablePort();

	assertThat(port3).isBetween(NetUtil.PORT_RANGE_MIN, NetUtil.PORT_RANGE_MAX);
	System.out.println("random port:" + port3);

	// 尝试占住一个端口
	ServerSocket serverSocket = null;
	try {
		serverSocket = ServerSocketFactory.getDefault().createServerSocket(port, 1,
				InetAddress.getByName("localhost"));

		assertThat(NetUtil.isPortAvailable(port)).isFalse();

		int port4 = NetUtil.findAvailablePortFrom(port);
		assertThat(port4).isEqualTo(port + 1);

		try {
			int port5 = NetUtil.findRandomAvailablePort(port, port);
			fail("should fail before");
		} catch (Throwable t) {
			assertThat(t).isInstanceOf(IllegalStateException.class);
		}

	} finally {
		IOUtil.close(serverSocket);
	}

}
 

@Test
public void runAccessTokenRefresher() throws IOException, InterruptedException {
	File tempDir = tempFolder.newFolder();
	File clientJson = new File(tempDir, "client.json");
	IOUtil.writeText("{\"client_id\":\"abcdefg \",\"client_secret\":\"geheim\"}", clientJson);
	File userJson = new File(tempDir, "user.json");
	IOUtil.writeText("{\"application_username\":\"klaus \",\"application_password\":\"geheim\"}", userJson);
	System.setProperty(CREDENTIALS_DIR, tempDir.getAbsolutePath());

	ClientCredentialsProvider ccp = new JsonFileBackedClientCredentialsProvider(clientJson);
	UserCredentialsProvider ucp = new JsonFileBackedUserCredentialsProvider(userJson);
	HttpProviderFactory hpf = Mockito.mock(HttpProviderFactory.class);

	AccessTokensBuilder accessTokenBuilder = Tokens.createAccessTokensWithUri(uri)
	                                               .usingClientCredentialsProvider(ccp)
	                                               .usingUserCredentialsProvider(ucp)
	                                               .usingHttpProviderFactory(hpf)
	                                               .manageToken("TR_TEST")
	                                               .done()
	                                               .manageToken("TR_TEST_1")
	                                               .done()
	                                               .manageToken("TR_TEST_2")
                                                      .done();

	HttpProvider httpProvider = Mockito.mock(HttpProvider.class);

	Mockito.when(hpf.create(Mockito.any(ClientCredentials.class), Mockito.any(UserCredentials.class),
			Mockito.any(URI.class), Mockito.any(HttpConfig.class))).thenReturn(httpProvider);

	Mockito.when(httpProvider.createToken(Mockito.any(AccessTokenConfiguration.class)))
               .thenReturn(new AccessToken("123456789", "BEARER", 2, new Date(System.currentTimeMillis() + 15000)))
               .thenThrow(new RuntimeException("DO NOT FAIL ALL CONFIGURATIONS"))
               .thenReturn(new AccessToken("123456789", "BEARER", 2, new Date(System.currentTimeMillis() + 15000)))
               .thenReturn(new AccessToken("123456789", "BEARER", 2, new Date(System.currentTimeMillis() + 15000)))
               .thenThrow(new RuntimeException("DO NOT FAIL ALL CONFIGURATIONS"))
               .thenReturn(new AccessToken("123456789", "BEARER", 2, new Date(System.currentTimeMillis() + 15000)));
	accessTokens = accessTokenBuilder.start();
	Assertions.assertThat(accessTokens).isNotNull();

	TimeUnit.SECONDS.sleep(30);

	Mockito.verify(httpProvider, Mockito.atLeastOnce()).createToken(Mockito.any(AccessTokenConfiguration.class));

}
 

@Override
public BElement deserialize(InputStream in) {
    return BReference.of(new SimplePojo(IOUtil.readLines(in).iterator().next()));
}
 

@Override
public BElement deserialize(InputStream in) {
    return BElement.ofAny(IOUtil.readLines(in).iterator().next());
}
 

@Override
public BElement deserialize(InputStream in) {
    return BObject.of("name", IOUtil.readLines(in).iterator().next());
}
 

@Override
public BElement deserialize(InputStream in) {
    return BElement.ofAny(IOUtil.readLines(in).iterator().next());
}