mirror of
https://github.com/google/nomulus.git
synced 2025-05-01 20:47:52 +02:00
9ab68613a0
Also changed truth.SubjectFactory to truth.Subject.Factory (plain renaming) and use method reference instead of anonymous class to create the factory when applicable. FailureMetadata, an opaque object to its users, is introduced to replace FailureStrategy in in custom Subject in order to resolve some existing flaws of FailureStrategy as well as enable new features to be added to Truth. New API is available in Truth-0.36, if there is a build/pom.xml, it's also updated to use this version. More information: See [] ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=176110788
88 lines
3 KiB
Java
88 lines
3 KiB
Java
// Copyright 2017 The Nomulus Authors. All Rights Reserved.
|
|
//
|
|
// 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 google.registry.testing;
|
|
|
|
import static com.google.common.truth.Truth.assertAbout;
|
|
import static google.registry.util.PreconditionsUtils.checkArgumentNotNull;
|
|
|
|
import com.google.common.truth.FailureMetadata;
|
|
import com.google.common.truth.Subject;
|
|
import google.registry.model.eppoutput.Result.Code;
|
|
import google.registry.monitoring.whitebox.EppMetric;
|
|
import google.registry.testing.TruthChainer.And;
|
|
import java.util.Objects;
|
|
import java.util.Optional;
|
|
|
|
/** Utility methods for asserting things about {@link EppMetric} instances. */
|
|
public class EppMetricSubject extends Subject<EppMetricSubject, EppMetric> {
|
|
|
|
public EppMetricSubject(FailureMetadata failureMetadata, EppMetric subject) {
|
|
super(failureMetadata, subject);
|
|
}
|
|
|
|
public static EppMetricSubject assertThat(EppMetric subject) {
|
|
return assertAbout(SUBJECT_FACTORY).that(subject);
|
|
}
|
|
|
|
public And<EppMetricSubject> hasClientId(String clientId) {
|
|
return hasValue(clientId, actual().getClientId(), "has clientId");
|
|
}
|
|
|
|
public And<EppMetricSubject> hasCommandName(String commandName) {
|
|
return hasValue(commandName, actual().getCommandName(), "has commandName");
|
|
}
|
|
|
|
public And<EppMetricSubject> hasEppTarget(String eppTarget) {
|
|
return hasValue(eppTarget, actual().getEppTarget(), "has eppTarget");
|
|
}
|
|
|
|
public And<EppMetricSubject> hasStatus(Code status) {
|
|
return hasValue(status, actual().getStatus(), "has status");
|
|
}
|
|
|
|
public And<EppMetricSubject> hasNoStatus() {
|
|
if (actual().getStatus().isPresent()) {
|
|
fail("has no status");
|
|
}
|
|
return new And<>(this);
|
|
}
|
|
|
|
public And<EppMetricSubject> hasTld(String tld) {
|
|
return hasValue(tld, actual().getTld(), "has tld");
|
|
}
|
|
|
|
public And<EppMetricSubject> hasNoTld() {
|
|
if (actual().getTld().isPresent()) {
|
|
fail("has no tld");
|
|
}
|
|
return new And<>(this);
|
|
}
|
|
|
|
private <E> And<EppMetricSubject> hasValue(E expected, Optional<E> actual, String verb) {
|
|
checkArgumentNotNull(expected, "Expected value cannot be null");
|
|
if (actual == null) {
|
|
fail("is non-null", expected);
|
|
} else if (!actual.isPresent()) {
|
|
fail("has value", expected);
|
|
} else if (!Objects.equals(expected, actual.get())) {
|
|
failWithBadResults(verb, expected, verb, actual);
|
|
}
|
|
return new And<>(this);
|
|
}
|
|
|
|
/** {@link Subject.Factory} for assertions about {@link EppMetric} objects. */
|
|
private static final Subject.Factory<EppMetricSubject, EppMetric> SUBJECT_FACTORY =
|
|
EppMetricSubject::new;
|
|
}
|