org.meanbean.factories.net.UrlFactory Maven / Gradle / Ivy
/*-
*
* meanbean
*
* Copyright (C) 2010 - 2020 the original author or authors.
*
* 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 org.meanbean.factories.net;
import org.meanbean.factories.basic.RandomFactoryBase;
import org.meanbean.util.RandomValueGenerator;
import org.meanbean.util.RandomValueSampler;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
public class UrlFactory extends RandomFactoryBase {
private RandomValueSampler sampler;
private List schemes = Arrays.asList("http://", "https://", "ftp://");
private List tlds = Arrays.asList(".example", ".invalid", ".test");
private List paths = Arrays.asList("", "/foo", "/foo/bar/", "/foo/bar/index?a=b");
public UrlFactory(RandomValueGenerator randomValueGenerator) {
super(randomValueGenerator);
this.sampler = new RandomValueSampler(randomValueGenerator);
}
@Override
public URL create() {
String scheme = sampler.getFrom(schemes);
String domain = getRandomDomain();
String tld = sampler.getFrom(tlds);
String path = sampler.getFrom(paths);
String url = String.join("", scheme, domain, tld, path);
try {
return new URL(url);
} catch (MalformedURLException e) {
throw new IllegalStateException(url, e);
}
}
protected String getRandomDomain() {
int subdomainCount = getRandomValueGenerator().nextInt(1) + 1;
return IntStream.range(0, subdomainCount)
.mapToObj(num -> UUID.randomUUID().toString())
.collect(Collectors.joining("."));
}
}