package com.baeldung.testloglevel;

import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.autoconfigure.security.servlet.SecurityAutoConfiguration;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
import org.springframework.boot.test.rule.OutputCapture;
import org.springframework.boot.test.web.client.TestRestTemplate;
import org.springframework.http.ResponseEntity;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.junit4.SpringRunner;

import static org.assertj.core.api.Assertions.assertThat;

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = WebEnvironment.DEFINED_PORT, classes = TestLogLevelApplication.class)
@EnableAutoConfiguration(exclude = SecurityAutoConfiguration.class)
@ActiveProfiles("testloglevel")
public class TestLogLevelWithProfileIntegrationTest {

	@Autowired
	private TestRestTemplate restTemplate;

	@Rule
	public OutputCapture outputCapture = new OutputCapture();

	@Test
	public void givenInfoRootLevelAndDebugLevelForOurPackage_whenCall_thenPrintDebugLogsForOurPackage() {
		ResponseEntity<String> response = restTemplate.getForEntity("http://localhost:8080/testLogLevel", String.class);

		assertThat(response.getStatusCode().value()).isEqualTo(200);
		assertThatOutputContainsLogForOurPackage("DEBUG");
	}

	@Test
	public void givenInfoRootLevelAndDebugLevelForOurPackage_whenCall_thenNoDebugLogsForOtherPackages() {
		ResponseEntity<String> response = restTemplate.getForEntity("http://localhost:8080/testLogLevel", String.class);

		assertThat(response.getStatusCode().value()).isEqualTo(200);
		assertThatOutputDoesntContainLogForOtherPackages("DEBUG");
	}

	@Test
	public void givenInfoRootLevelAndDebugLevelForOurPackage_whenCall_thenPrintInfoLogs() {
		ResponseEntity<String> response = restTemplate.getForEntity("http://localhost:8080/testLogLevel", String.class);

		assertThat(response.getStatusCode().value()).isEqualTo(200);
		assertThatOutputContainsLogForOurPackage("INFO");
		assertThatOutputContainsLogForOtherPackages("INFO");
	}

	private void assertThatOutputContainsLogForOurPackage(String level) {
		assertThat(outputCapture.toString()).containsPattern("TestLogLevelController.*" + level + ".*");
	}

	private void assertThatOutputDoesntContainLogForOtherPackages(String level) {
		assertThat(outputCapture.toString().replaceAll("(?m)^.*TestLogLevelController.*$", "")).doesNotContain(level);
	}

	private void assertThatOutputContainsLogForOtherPackages(String level) {
		assertThat(outputCapture.toString().replaceAll("(?m)^.*TestLogLevelController.*$", "")).contains(level);
	}

}