Examples with ProviderState au.com.dius.pact.core.model.ProviderState used on opensource projects

Example 1 with ProviderState

use of au.com.dius.pact.core.model.ProviderState in project pact-jvm by DiUS.

the class MessagePactProviderRule method apply.

/* (non-Javadoc)
	 * @see org.junit.rules.ExternalResource#apply(org.junit.runners.model.Statement, org.junit.runner.Description)
	 */
@Override
public Statement apply(final Statement base, final Description description) {
    return new Statement() {

        @Override
        public void evaluate() throws Throwable {
            PactVerifications pactVerifications = description.getAnnotation(PactVerifications.class);
            if (pactVerifications != null) {
                evaluatePactVerifications(pactVerifications, base, description);
                return;
            }
            PactVerification pactDef = description.getAnnotation(PactVerification.class);
            // no pactVerification? execute the test normally
            if (pactDef == null) {
                base.evaluate();
                return;
            }
            Message providedMessage = null;
            Map<String, Message> pacts;
            if (StringUtils.isNoneEmpty(pactDef.fragment())) {
                Optional<Method> possiblePactMethod = findPactMethod(pactDef);
                if (possiblePactMethod.isEmpty()) {
                    base.evaluate();
                    return;
                }
                pacts = new HashMap<>();
                Method method = possiblePactMethod.get();
                Pact pact = method.getAnnotation(Pact.class);
                MessagePactBuilder builder = new MessagePactBuilder().consumer(Objects.toString(ep.parseExpression(pact.consumer(), DataType.RAW))).hasPactWith(provider);
                messagePact = (MessagePact) method.invoke(testClassInstance, builder);
                for (Message message : messagePact.getMessages()) {
                    pacts.put(message.getProviderStates().stream().map(ProviderState::getName).collect(Collectors.joining()), message);
                }
            } else {
                pacts = parsePacts();
            }
            if (pactDef.value().length == 2 && !pactDef.value()[1].trim().isEmpty()) {
                providedMessage = pacts.get(pactDef.value()[1].trim());
            } else if (!pacts.isEmpty()) {
                providedMessage = pacts.values().iterator().next();
            }
            if (providedMessage == null) {
                base.evaluate();
                return;
            }
            setMessage(providedMessage, description);
            Metrics.INSTANCE.sendMetrics(new MetricEvent.ConsumerTestRun(messagePact.getMessages().size(), "junit"));
            try {
                base.evaluate();
                PactFolder pactFolder = testClassInstance.getClass().getAnnotation(PactFolder.class);
                PactDirectory pactDirectory = testClassInstance.getClass().getAnnotation(PactDirectory.class);
                if (pactFolder != null) {
                    messagePact.write(pactFolder.value(), PactSpecVersion.V3);
                } else if (pactDirectory != null) {
                    messagePact.write(pactDirectory.value(), PactSpecVersion.V3);
                } else {
                    messagePact.write(BuiltToolConfig.INSTANCE.getPactDirectory(), PactSpecVersion.V3);
                }
            } catch (Throwable t) {
                throw t;
            }
        }
    };
}

Example 2 with ProviderState

use of au.com.dius.pact.core.model.ProviderState in project pact-jvm by DiUS.

the class MessagePactProviderRule method parsePacts.

@SuppressWarnings("unchecked")
private Map<String, Message> parsePacts() {
    if (providerStateMessages == null) {
        providerStateMessages = new HashMap<>();
        for (Method m : testClassInstance.getClass().getMethods()) {
            if (conformsToSignature(m)) {
                Pact pact = m.getAnnotation(Pact.class);
                if (pact != null) {
                    String provider = Objects.toString(ep.parseExpression(pact.provider(), DataType.RAW));
                    if (provider != null && !provider.trim().isEmpty()) {
                        MessagePactBuilder builder = new MessagePactBuilder().consumer(pact.consumer()).hasPactWith(provider);
                        List<Message> messages;
                        try {
                            messagePact = (MessagePact) m.invoke(testClassInstance, builder);
                            messages = messagePact.getMessages();
                        } catch (Exception e) {
                            throw new RuntimeException("Failed to invoke pact method", e);
                        }
                        for (Message message : messages) {
                            if (message.getProviderStates().isEmpty()) {
                                providerStateMessages.put("", message);
                            } else {
                                for (ProviderState state : message.getProviderStates()) {
                                    providerStateMessages.put(state.getName(), message);
                                }
                            }
                        }
                    }
                }
            }
        }
    }
    return providerStateMessages;
}