com.indeed.proctor.consumer.ant.groups.ftl Maven / Gradle / Ivy
The newest version!
package ${packageName};
import com.indeed.proctor.common.*;
import com.indeed.proctor.common.model.Payload;
import com.indeed.proctor.common.model.TestBucket;
import com.indeed.proctor.consumer.*;
import com.indeed.proctor.consumer.logging.TestUsageObserver;
import javax.annotation.Generated;
import javax.annotation.Nullable;
import javax.annotation.Nonnull;
import java.lang.Override;
import java.util.Map;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.core.JsonProcessingException;
/*
* GENERATED source; do not edit this file directly
* This class can be extended:
* - to modify only logging, override {@link #toLoggingString()}
* - to modify determined groups, override {@link #overrideDeterminedBucketValue()}
*/
@SuppressWarnings("deprecation")
@Generated("com.indeed.proctor.consumer.gen.TestGroupsGenerator")
public class ${mainClassName} extends AbstractGroups {
public static final ${mainClassName} EMPTY = new ${mainClassName}(ProctorResult.EMPTY);
private static final ObjectMapper MAPPER = new ObjectMapper();
public ${mainClassName}(final ProctorResult proctorResult) {
super(proctorResult);
}
/**
* @param testUsageObserver if non-null, it's methods will be called whenever proctor tests are used
*/
public ${mainClassName}(final ProctorResult proctorResult, final TestUsageObserver testUsageObserver) {
super(proctorResult, testUsageObserver);
}
public static Bucket<${testEnumName}>[] getBuckets(final ${testEnumName} test) {
/** As a workaround to an ART bug, cast return value to Bucket<${testEnumName}>[] PROC-259 **/
switch (test) {
<#list testDefs as testDef>
<#if testDef.buckets?has_content>
case ${testDef.enumName}:
return (Bucket<${testEnumName}>[])${testDef.javaClassName}.values();
}
return null;
}
public enum ${testEnumName} implements com.indeed.proctor.consumer.Test {
<#list testDefs as testDef>
${testDef.enumName}("${testDef.name}", ${testDef.defaultValue})<#if testDef_has_next>,<#else>;
; // fix compilation if no tests
private final String name;
private final int fallbackValue;
private ${testEnumName}(final String name, final int fallbackValue) {
this.name = name;
this.fallbackValue = fallbackValue;
}
@Override
public String getName() {
return name;
}
@Override
public int getFallbackValue() {
return fallbackValue;
}
}
<#list testDefs as testDef>
<#if testDef.buckets?has_content>
public enum ${testDef.javaClassName} implements Bucket<${testEnumName}> {
<#list testDef.buckets as bucket>
${bucket.enumName}(${bucket.value}, "${bucket.name}")<#if bucket_has_next>,<#else>;
private final int value;
private final String name;
private final String fullName;
private ${testDef.javaClassName}(final int value, final String name) {
this.value = value;
this.name = name;
this.fullName = getTest().getName() + "-" + name;
}
@Override
public ${testEnumName} getTest() {
return ${testEnumName}.${testDef.enumName};
}
@Override
public int getValue() {
return value;
}
@Override
public String getName() {
return name;
}
@Override
public String getFullName() {
return fullName;
}
<#list testDef.buckets as bucket>
<#if testDef.defaultValue == bucket.value>
public static ${testDef.javaClassName} getFallback() {
return ${testDef.javaClassName}.${bucket.enumName};
}
}
<#list testDefs as testDef>
<#if testDef.buckets?has_content>
@Nonnull
public ${testDef.javaClassName} get${testDef.javaClassName}() {
for (final ${testDef.javaClassName} bucket : ${testDef.javaClassName}.values()) {
final String testName = Test.${testDef.enumName}.getName();
if (isBucketActive(testName, bucket.getValue(), ${testDef.defaultValue})) {
return bucket;
}
}
// Safe to throw NPE here because the code generator ensures that the default value
// is a valid bucket in the test.
throw new NullPointerException("No fallback bucket found for '${testDef.name}'");
}
/**
* @deprecated Use {@link #get${testDef.javaClassName}Value()} instead
*/
@Deprecated
public int get${testDef.javaClassName}Value(final int defaultValue) {
return getValue(${testEnumName}.${testDef.enumName}.getName(), defaultValue);
}
public int get${testDef.javaClassName}Value() {
return getValue(${testEnumName}.${testDef.enumName}.getName(), ${testEnumName}.${testDef.enumName}.getFallbackValue());
}
<#if (testDef.payloadJavaClass)??>
<#if (testDef.isMap)??>
public @Nullable ${mainClassName}Payload.${testDef.name?cap_first} get${testDef.javaClassName}Payload() {
final Payload payload = getPayload(${testEnumName}.${testDef.enumName}.getName(), ${testEnumName}.${testDef.enumName}.getFallbackValue());
if (payload == null || payload.equals(Payload.EMPTY_PAYLOAD)) {
return null;
}
return new ${mainClassName}Payload.${testDef.name?cap_first}(payload);
}
<#if testDef.buckets?has_content>
public @Nullable ${mainClassName}Payload.${testDef.name?cap_first} get${testDef.javaClassName}PayloadForBucket(final ${testDef.javaClassName} targetBucket) {
final @Nullable TestBucket bucket = getTestBucketForBucket(${testEnumName}.${testDef.enumName}.getName(), targetBucket);
if (bucket == null) {
return null;
}
return new ${mainClassName}Payload.${testDef.name?cap_first}(bucket);
}
<#else>
public @Nullable ${testDef.payloadJavaClass} get${testDef.javaClassName}Payload() {
final Payload payload = getPayload(${testEnumName}.${testDef.enumName}.getName(), ${testEnumName}.${testDef.enumName}.getFallbackValue());
return payload.${testDef.payloadAccessorName}();
}
<#if testDef.buckets?has_content>
public @Nullable ${testDef.payloadJavaClass} get${testDef.javaClassName}PayloadForBucket(final ${testDef.javaClassName} targetBucket) {
final @Nullable TestBucket bucket = getTestBucketForBucket(${testEnumName}.${testDef.enumName}.getName(), targetBucket);
if (bucket == null) {
return null;
}
final Payload payload = bucket.getPayload();
if (payload == null) {
return null;
}
return payload.${testDef.payloadAccessorName}();
}
<#if (testDef.isJson)??>
public @Nullable T get${testDef.javaClassName}Payload(final Class payloadType) {
final Payload payload = getPayload(${testEnumName}.${testDef.enumName}.getName(), ${testEnumName}.${testDef.enumName}.getFallbackValue());
if (payload == null) {
return null;
}
try {
return MAPPER.treeToValue(payload.${testDef.payloadAccessorName}(), payloadType);
} catch (JsonProcessingException e) {
return null;
}
}
<#if (testDef.description)??>
public @Nullable String get${testDef.javaClassName}Description() {
return "${testDef.description}";
}
<#list testDef.buckets as bucket>
public boolean is${testDef.javaClassName}${bucket.javaClassName}() {
final String testName = Test.${testDef.enumName}.getName();
final int bucketValue = ${testDef.javaClassName}.${bucket.enumName}.getValue();
return isBucketActive(testName, bucketValue, ${testDef.defaultValue});
}
<#if bucket_has_next || testDef_has_next>
}