diff --git a/src/main/java/edu/group5/app/model/organization/OrganizationRepository.java b/src/main/java/edu/group5/app/model/organization/OrganizationRepository.java index 27b128f..a9e5bb6 100644 --- a/src/main/java/edu/group5/app/model/organization/OrganizationRepository.java +++ b/src/main/java/edu/group5/app/model/organization/OrganizationRepository.java @@ -3,38 +3,48 @@ import edu.group5.app.model.Repository; import java.util.HashMap; +import java.util.Map; import java.util.Objects; /** - * Handles business logic associated with organizations + * Handles the business logic associated with organizations */ -public class OrganizationRepo extends Repository { - private final HashMap content; +public class OrganizationRepository extends Repository { + private final Map content; /** * Creates a new Organization Repository * - * @param content holds all current organizations in the repository + * @param content holds all current organizations in the repository; must not be null * @throws NullPointerException if content is null */ - public OrganizationRepo(HashMap content) { + public OrganizationRepository(Map content) { this.content = Objects.requireNonNull(content, "content cannot be null"); super(content); } + /** + * Returns the organizations within the repository. + * @return all organizations within the repository. + */ + @Override + public Map getContent() { + return content; + } + /** * Gets all trusted organizations in the repository * @return all organizations with trusted = true */ - public HashMap getTrustedOrganizations() { - HashMap trustedOrgs = new HashMap<>(); + public Map getTrustedOrganizations() { + Map trustedOrganizations = new HashMap<>(); content.forEach((orgNr, org) -> { if (org.trusted()) { - trustedOrgs.put(orgNr, org); + trustedOrganizations.put(orgNr, org); } }); - return trustedOrgs; + return trustedOrganizations; } }