test: add unit tests for PetPolicy

src/test/java/org/example/vet1177/policy/PetPolicyTest.java

@@ -0,0 +1,170 @@
+package org.example.vet1177.policy;
+
+import org.example.vet1177.entities.Pet;
+import org.example.vet1177.entities.Role;
+import org.example.vet1177.entities.User;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import java.lang.reflect.Field;
+import java.util.UUID;
+
+import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
+
+class PetPolicyTest {
+
+    private PetPolicy policy;
+
+    private User admin;
+    private User owner;
+    private User otherOwner;
+    private User vet;
+
+    private Pet pet;
+
+    @BeforeEach
+    void setUp() throws Exception {
+        policy = new PetPolicy();
+
+        admin = new User("Admin Adminsson", "admin@vet.se", "hash", Role.ADMIN);
+        setPrivateField(admin, "id", UUID.randomUUID());
+
+        owner = new User("Anna Ägare", "anna@mail.se", "hash", Role.OWNER);
+        setPrivateField(owner, "id", UUID.randomUUID());
+
+        otherOwner = new User("Karin Annan", "karin@mail.se", "hash", Role.OWNER);
+        setPrivateField(otherOwner, "id", UUID.randomUUID());
+
+        vet = new User("Dr. Erik Vet", "erik@vet.se", "hash", Role.VET);
+        setPrivateField(vet, "id", UUID.randomUUID());
+
+        pet = new Pet();
+        pet.setOwner(owner);
+    }
+
+    //helper
+    private static void setPrivateField(Object target, String fieldName, Object value) throws Exception {
+        Field field = target.getClass().getDeclaredField(fieldName);
+        field.setAccessible(true);
+        field.set(target, value);
+    }
+
+    //Test canCreate
+
+    @Test
+    void canCreate_owner_shouldReturnTrue() {
+        assertThat(policy.canCreate(owner)).isTrue();
+    }
+
+    @Test
+    void canCreate_admin_shouldReturnTrue() {
+        assertThat(policy.canCreate(admin)).isTrue();
+    }
+
+    @Test
+    void canCreate_vet_shouldReturnFalse() {
+        assertThat(policy.canCreate(vet)).isFalse();
+    }
+
+    // canView
+    @Test
+    void canView_admin_shouldReturnTrue() {
+        assertThat(policy.canView(admin, pet)).isTrue();
+    }
+
+    @Test
+    void canView_ownerOfPet_shouldReturnTrue() {
+        assertThat(policy.canView(owner, pet)).isTrue();
+    }
+
+    @Test
+    void canView_ownerOfOtherPet_shouldReturnFalse() {
+        assertThat(policy.canView(otherOwner, pet)).isFalse();
+    }
+
+    @Test
+    void canView_vet_shouldReturnFalse() {
+        assertThat(policy.canView(vet, pet)).isFalse();
+    }
+
+    @Test
+    void canView_ownerWithNullPetOwner_shouldReturnFalse() {
+        pet.setOwner(null);
+
+        assertThat(policy.canView(owner, pet)).isFalse();
+    }
+
+    //Can viewOwnerPets
+    @Test
+    void canViewOwnerPets_admin_shouldReturnTrue() {
+        assertThat(policy.canViewOwnerPets(admin, UUID.randomUUID())).isTrue();
+    }
+
+    @Test
+    void canViewOwnerPets_ownerViewingOwnPets_shouldReturnTrue() {
+        assertThat(policy.canViewOwnerPets(owner, owner.getId())).isTrue();
+    }
+
+    @Test
+    void canViewOwnerPets_ownerViewingOthersPets_shouldReturnFalse() {
+        assertThat(policy.canViewOwnerPets(owner, otherOwner.getId())).isFalse();
+    }
+
+    @Test
+    void canViewOwnerPets_vet_shouldReturnFalse() {
+        assertThat(policy.canViewOwnerPets(vet, owner.getId())).isFalse();
+    }
+
+    // canDelete
+
+    @Test
+    void canDelete_admin_shouldReturnTrue() {
+        assertThat(policy.canDelete(admin, pet)).isTrue();
+    }
+
+    @Test
+    void canDelete_ownerOfPet_shouldReturnTrue() {
+        assertThat(policy.canDelete(owner, pet)).isTrue();
+    }
+
+    @Test
+    void canDelete_ownerOfOtherPet_shouldReturnFalse() {
+        assertThat(policy.canDelete(otherOwner, pet)).isFalse();
+    }
+
+    @Test
+    void canDelete_vet_shouldReturnFalse() {
+        assertThat(policy.canDelete(vet, pet)).isFalse();
+    }
+
+    @Test
+    void canDelete_ownerWithNullPetOwner_shouldReturnFalse() {
+        pet.setOwner(null);
+
+        assertThat(policy.canDelete(owner, pet)).isFalse();
+    }
+
+    //canUpdate
+    @Test
+    void canUpdate_ownerOfPet_shouldReturnTrue() {
+        assertThat(policy.canUpdate(owner, pet)).isTrue();
+    }
+
+    @Test
+    void canUpdate_ownerOfOtherPet_shouldReturnFalse() {
+        assertThat(policy.canUpdate(otherOwner, pet)).isFalse();
+    }
+
+    @Test
+    void canUpdate_vet_shouldReturnFalse() {
+        assertThat(policy.canUpdate(vet, pet)).isFalse();
+    }
+
+    @Test
+    void canUpdate_ownerWithNullPetOwner_shouldReturnFalse() {
+        pet.setOwner(null);
+
+        assertThat(policy.canUpdate(owner, pet)).isFalse();
+    }
+
+}