56 lines
1.2 KiB
Java
56 lines
1.2 KiB
Java
package banking;
|
|
|
|
import dev.banking.model.BankAccount;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
import java.util.concurrent.CountDownLatch;
|
|
import java.util.concurrent.ExecutorService;
|
|
import java.util.concurrent.Executors;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
|
|
public class TestB {
|
|
|
|
@Test
|
|
void concurrentDepositWithdraw() throws Exception {
|
|
|
|
BankAccount account =
|
|
new BankAccount(1, 1_000_000);
|
|
|
|
ExecutorService executor =
|
|
Executors.newFixedThreadPool(100);
|
|
|
|
CountDownLatch latch =
|
|
new CountDownLatch(100);
|
|
|
|
for (int i = 0; i < 50; i++) {
|
|
|
|
executor.submit(() -> {
|
|
|
|
for (int j = 0; j < 10000; j++) {
|
|
account.deposit(1);
|
|
}
|
|
|
|
latch.countDown();
|
|
});
|
|
|
|
executor.submit(() -> {
|
|
|
|
for (int j = 0; j < 10000; j++) {
|
|
account.withdraw(1);
|
|
}
|
|
|
|
latch.countDown();
|
|
});
|
|
}
|
|
|
|
latch.await();
|
|
|
|
executor.shutdown();
|
|
|
|
assertEquals(
|
|
1_000_000,
|
|
account.getBalance()
|
|
);
|
|
}
|
|
} |