io.qameta.allure.entity.WithLabels Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of allure-plugin-api Show documentation

Module allure-plugin-api of Allure Framework.

There is a newer version: 2.30.0

/*
 *  Copyright 2016-2023 Qameta Software OÜ
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package io.qameta.allure.entity;

import java.util.List;
import java.util.Optional;
import java.util.stream.Collector;
import java.util.stream.Collectors;

/**
 * @author Dmitry Baev [email protected]
 * Date: 31.01.16
 */
public interface WithLabels {

    List getLabels();

    void setLabels(List labels);

    default  T findAll(LabelName name, Collector collector) {
        return findAll(name.value(), collector);
    }

    default  T findAll(String name, Collector collector) {
        return getLabels().stream()
                .filter(label -> name.equals(label.getName()))
                .map(Label::getValue)
                .collect(collector);
    }

    default List findAll(LabelName name) {
        return findAll(name, Collectors.toList());
    }

    default List findAll(String name) {
        return findAll(name, Collectors.toList());
    }

    default Optional findOne(LabelName name) {
        return findOne(name.value());
    }

    default Optional findOne(String name) {
        return getLabels().stream()
                .filter(label -> name.equals(label.getName()))
                .map(Label::getValue)
                .findAny();
    }

    default void addLabelIfNotExists(LabelName name, String value) {
        addLabelIfNotExists(name.value(), value);
    }

    default void addLabelIfNotExists(String name, String value) {
        if (value == null || name == null) {
            return;
        }
        final Optional any = getLabels().stream()
                .map(Label::getName)
                .filter(name::equals)
                .findAny();
        if (!any.isPresent()) {
            addLabel(name, value);
        }
    }

    default void addLabel(String name, String value) {
        getLabels().add(new Label().setName(name).setValue(value));
    }
}