Inject clock for Tmch validity using InjectRule instead of manually

Basically a less patchy version of []

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=202138086
This commit is contained in:
guyben 2018-06-26 09:09:21 -07:00 committed by Ben McIlwain
parent 3550045636
commit b4cb3e0a6f
8 changed files with 14 additions and 20 deletions

View file

@ -108,8 +108,11 @@ public abstract class FlowTestCase<F extends Flow> extends ShardableTestCase {
sessionMetadata.setClientId("TheRegistrar");
sessionMetadata.setServiceExtensionUris(ProtocolDefinition.getVisibleServiceExtensionUris());
ofy().saveWithoutBackup().entity(new ClaimsListSingleton()).now();
inject.setStaticField(Ofy.class, "clock", clock); // For transactional flows.
}
// For transactional flows
inject.setStaticField(Ofy.class, "clock", clock);
// For SignedMark signature validity
inject.setStaticField(TmchCertificateAuthority.class, "clock", clock);
}
protected void removeServiceExtensionUri(String uri) {
sessionMetadata.setServiceExtensionUris(