FairShareSchedulingTest.java

packagecom.thealgorithms.scheduling;

importstaticorg.junit.jupiter.api.Assertions.assertEquals;

importjava.util.Map;
importorg.junit.jupiter.api.BeforeEach;
importorg.junit.jupiter.api.Test;

publicclassFairShareSchedulingTest {

privateFairShareSchedulingscheduler;

@BeforeEach
publicvoidsetup() {
scheduler = newFairShareScheduling();
 }

@Test
publicvoidtestAllocateResourcesSingleUser() {
scheduler.addUser("User1");
scheduler.addTask("User1", 5);
scheduler.allocateResources(100);
Map<String, Integer> expected = Map.of("User1", 100);
assertEquals(expected, scheduler.getAllocatedResources());
 }

@Test
publicvoidtestAllocateResourcesMultipleUsers() {
scheduler.addUser("User1");
scheduler.addUser("User2");
scheduler.addUser("User3");
scheduler.addTask("User1", 2);
scheduler.addTask("User2", 3);
scheduler.addTask("User3", 5);
scheduler.allocateResources(100);
Map<String, Integer> expected = Map.of("User1", 20, "User2", 30, "User3", 50);
assertEquals(expected, scheduler.getAllocatedResources());
 }

@Test
publicvoidtestAllocateResourcesZeroWeightUser() {
scheduler.addUser("User1");
scheduler.addUser("User2");
scheduler.addTask("User2", 5);
scheduler.allocateResources(100);
Map<String, Integer> expected = Map.of("User1", 0, "User2", 100);
assertEquals(expected, scheduler.getAllocatedResources());
 }

@Test
publicvoidtestAllocateResourcesEqualWeights() {
scheduler.addUser("User1");
scheduler.addUser("User2");
scheduler.addTask("User1", 1);
scheduler.addTask("User2", 1);
scheduler.allocateResources(100);
Map<String, Integer> expected = Map.of("User1", 50, "User2", 50);
assertEquals(expected, scheduler.getAllocatedResources());
 }
}