guru.nidi.ramltester.core.UsageCollector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of raml-tester Show documentation
Test if a request/response matches a given raml definition.
There is a newer version: 0.9.1
/*
 * Copyright (C) 2014 Stefan Niederhauser ([email protected])
 *
 * 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 guru.nidi.ramltester.core;

import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 *
 */
abstract class UsageCollector {
    static final UsageCollector ACTION = new UsageCollector() {
        @Override
        public void collect(String key, Usage.Action action, Set result) {
            if (action.getUses() == 0) {
                result.add(key);
            }
        }
    };

    static final UsageCollector QUERY_PARAM = new UsageCollector() {
        @Override
        public void collect(String key, Usage.Action action, Set result) {
            for (final Map.Entry queryEntry : action.getQueryParameters().values()) {
                if (queryEntry.getValue() == 0) {
                    add(result, queryEntry.getKey(), key);
                }
            }
        }
    };

    static final UsageCollector FORM_PARAM = new UsageCollector() {
        @Override
        public void collect(String key, Usage.Action action, Set result) {
            for (final Map.Entry mimeTypeEntry : action.mimeTypes()) {
                for (final Map.Entry formEntry : mimeTypeEntry.getValue().getFormParameters().values()) {
                    if (formEntry.getValue() == 0) {
                        add(result, formEntry.getKey(), key + " (" + mimeTypeEntry.getKey() + ")");
                    }
                }
            }
        }
    };

    static final UsageCollector REQUEST_HEADER = new UsageCollector() {
        @Override
        public void collect(String key, Usage.Action action, Set result) {
            for (final Map.Entry requestEntry : action.getRequestHeaders().values()) {
                if (requestEntry.getValue() == 0) {
                    add(result, requestEntry.getKey(), key);
                }
            }
        }
    };

    static final UsageCollector RESPONSE_HEADER = new UsageCollector() {
        @Override
        public void collect(String key, Usage.Action action, Set result) {
            for (final Map.Entry responseEntry : action.responses()) {
                for (final Map.Entry headerEntry : responseEntry.getValue().getResponseHeaders().values()) {
                    if (headerEntry.getValue() == 0) {
                        add(result, headerEntry.getKey(), key + " -> " + responseEntry.getKey());
                    }
                }
            }
        }
    };

    static final UsageCollector RESPONSE_CODE = new UsageCollector() {
        @Override
        public void collect(String key, Usage.Action action, Set result) {
            for (final Map.Entry responseCodeEntry : action.getResponseCodes().values()) {
                if (responseCodeEntry.getValue() == 0) {
                    add(result, responseCodeEntry.getKey(), key);
                }
            }
        }
    };

    private UsageCollector() {
    }

    abstract void collect(String key, Usage.Action action, Set result);

    public Set collect(Usage usage) {
        final Set res = new HashSet<>();
        for (final Map.Entry resourceEntry : usage) {
            for (final Map.Entry actionEntry : resourceEntry.getValue()) {
                collect(actionEntry.getKey() + " " + resourceEntry.getKey(), actionEntry.getValue(), res);
            }
        }
        return res;
    }

    protected static void add(Set result, String key, String value) {
        result.add(key + " in " + value);
    }

}