테스트 코드 작성하기 with Transaction (1)

@Test
@DisplayName("Stopwatch 실행시켜보는 코드")
public void testStopWatch (){
    StopWatch stopWatch = new StopWatch(); // 스톱워치 객체 생성
    stopWatch.start(); // 스톱워치 시작
    // 비즈니스 코드 시작
    // 1초 대기
    for(int i=0; i<100; i++){
        try {
            Thread.sleep(10); //0.01초 대기
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    // 비즈니스 코드 끝
    stopWatch.stop(); // 스톱워치 스탑
    System.out.println(stopWatch.prettyPrint());
}

StopWatch '': running time = 1209436250 ns
---------------------------------------------
ns         %     Task name
---------------------------------------------
1209436250  100%

private static int temp = 100;

@Test
    @DisplayName("메인 스레드가 멀티 스레드 작업이 다 끝날 때까지 기다리는지 테스트")
    public void multiThreadTest() throws InterruptedException {
        StopWatch stopWatch = new StopWatch(); // 스톱워치 객체 생성
        stopWatch.start(); // 스톱워치 시작
        int threadCount = 20; // 멀티스레드 개수
        int requestCount = 100; // 요청 개수
        ExecutorService executorService = Executors.newFixedThreadPool(threadCount); // 멀티스레드 생성
        // 스레드는 countDown을 호출해서 requestCount를 하나씩 감소시킴
        CountDownLatch countDownLatch = new CountDownLatch(requestCount); // 멀티스레드 카운트 생성

        for (int i = 0; i < requestCount; i++) {
            executorService.submit(() -> { // submit 안에 함수는 스레드가 실행시킴
                try {
                    temp -= 1; // static 변수 temp을 1씩 감소
                } finally {
                    countDownLatch.countDown(); // requestCount 하나 감소
                } });
        }

        // 메인 스레드는 requestCount가 0이 될때까지 blocked된다.
        countDownLatch.await();
        stopWatch.stop(); // 스톱워치 스탑
        System.out.println(stopWatch.prettyPrint());
        assertEquals(0, temp);
    }

StopWatch '': running time = 1715750 ns
---------------------------------------------
ns         %     Task name
---------------------------------------------
001715750  100%

필요:0
실제   :1

@RunWith(SpringRunner.class)
@SpringBootTest(classes = {MapRepository.class, CarService.class}) // 원하는 컴포넌트만 선택
public class CarServiceWithRepoTest {

	@Autowired
	private CarService carService; // Service 객체 직접 사용

	@Test
	public void shouldReturnValidDateInTheFuture() {
    	Date date = carService.schedulePickup(new Date(), new Route());
    	assertTrue(date.getTime() > new Date().getTime());
	}
}

@DataJpaTest
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
public class TestRepository {

    @Autowired
    TicketRepository ticketRepository;
    @Autowired
    TestEntityManager testEntityManager;


    @Test
    @DisplayName("티켓 수량 감소 후 수량 체크")
    public void givenNewTicket_whenUpdated_thenSuccess() throws InterruptedException {
        StopWatch stopWatch = new StopWatch(); // 스톱워치 객체 생성
        stopWatch.start(); // 스톱워치 시작

        Ticket newTiket = new Ticket("효랑이", 100L);
        testEntityManager.persist(newTiket);
        newTiket.subtractQuantity(); // 티켓 수량 감소
        ticketRepository.save(newTiket);
        Long result = testEntityManager.find(Ticket.class, newTiket.getTicketId()).getQuantity();

        assertEquals(99L, result);
        System.out.println("티켓 수량 : "+result);
        stopWatch.stop(); // 스톱워치 스탑
        System.out.println(stopWatch.prettyPrint());
    }
}

@SpringBootTest
public class TestService {

    @Autowired
    TicketRepository ticketRepository;
    @Autowired
    TicketService ticketService;
    static int threadCount = 20; // 멀티 스레드 생성
    static int requestCount = 100;

    @BeforeEach
    public void before(){ // ProductServiceV2 객체 생성을 위한 생성자를 통한 DI
        ticketRepository.save(new Ticket("A좌석", (long) requestCount));
    }

    @AfterEach
    public void after() {
        ticketRepository.deleteAll();
    }
    @Test
    @DisplayName("멀티스레드로 transaction이 걸린 티켓에 수량 감소 후 체크")
    public void givenMultiThreadAndTransaction_whenUpdated_thenSuccess() throws InterruptedException {
        StopWatch stopWatch = new StopWatch(); // 스톱워치 객체 생성
        stopWatch.start(); // 스톱워치 시작

        ExecutorService executorService = Executors.newFixedThreadPool(threadCount);
        // 스레드는 countDown을 호출해서 requestCount를 하나씩 감소시킴
        CountDownLatch countDownLatch = new CountDownLatch(requestCount);
        Long result1 = ticketService.findByTicketName("A좌석").getQuantity();
        System.out.println("초기 티켓 수량 : "+result1);

        for (int i = 0; i < requestCount; i++) {
            executorService.submit(() -> { // submit 안에 함수는 스레드가 실행시킴
                try {
                    ticketService.subtract("A좌석"); // 티켓 수량 감소
                } catch (Exception e){
                    System.out.println(e.getMessage());
                }
                finally {
                    countDownLatch.countDown();
                } });
        }
        // 메인 스레드는 requestCount가 0이 될때까지 blocked된다.
        countDownLatch.await();

        Long result = ticketService.findByTicketName("A좌석").getQuantity();
        System.out.println("티켓 수량 : "+result);

        stopWatch.stop(); // 스톱워치 스탑
        System.out.println(stopWatch.prettyPrint());
        assertEquals(0L, result);
    }
}

@Transactional // un-commit read 발생으로 update하지 못하는 상황 발생!
public Long subtract(String ticketName){
    try{
        Optional<Ticket> ticket = ticketRepository.findByTicketName(ticketName);
        if(ticket.isEmpty()){
            throw new RuntimeException("It's a non-existent ticket.");
        }
        ticket.get().subtractQuantity();
        return ticket.get().getQuantity();
    } catch (RuntimeException ex) {
        // Handle concurrent modification
        // For example, you can retry the operation or take other appropriate actions
        throw new RuntimeException("Concurrent modification detected "+ ex.getMessage());
    }
}

티켓 수량 : 88
StopWatch '': 0.349443625 seconds
----------------------------------------
Seconds       %       Task name
----------------------------------------
0.349443625   100%