com.github.tomakehurst.wiremock.extension.ExtensionDeclarations Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of wiremock Show documentation

A web service test double for all occasions

There is a newer version: 3.9.1

/*
 * Copyright (C) 2023-2024 Thomas Akehurst
 *
 * 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 com.github.tomakehurst.wiremock.extension;

import static java.util.Arrays.asList;

import java.util.*;
import java.util.stream.Collectors;
import org.wiremock.webhooks.Webhooks;

public class ExtensionDeclarations {

  private final List classNames;
  private final List> classes;
  private final Map instances;
  private final List factories;
  private static final String WEBHOOK_MESSAGE =
      "Passing webhooks in extensions is no longer required and"
          + " may lead to compatibility issues in future";

  public ExtensionDeclarations() {
    this.classNames = new ArrayList<>();
    this.classes = new ArrayList<>();
    this.instances = new LinkedHashMap<>();
    this.factories = new ArrayList<>();
  }

  public void add(String... classNames) {
    List processedClassNames =
        Arrays.stream(classNames).filter(this::removeWebhook).collect(Collectors.toList());
    this.classNames.addAll(processedClassNames);
  }

  public void add(Extension... extensionInstances) {
    Arrays.stream(extensionInstances).forEach(e -> instances.put(e.getName(), e));
  }

  public void add(Class... classes) {
    List> processedClasses =
        Arrays.stream(classes).filter(c -> removeWebhook(c.getName())).collect(Collectors.toList());
    this.classes.addAll(processedClasses);
  }

  public void add(ExtensionFactory... factories) {
    this.factories.addAll(asList(factories));
  }

  public List getClassNames() {
    return classNames;
  }

  public List> getClasses() {
    return classes;
  }

  public Map getInstances() {
    return instances;
  }

  public List getFactories() {
    return factories;
  }

  private boolean removeWebhook(String className) {
    if (className.equals(Webhooks.class.getName())) {
      System.out.println(WEBHOOK_MESSAGE);
      return false;
    }
    return true;
  }
}