Examples with Subscription io.gravitee.repository.management.model.Subscription used on opensource projects

Example 11 with Subscription

use of io.gravitee.repository.management.model.Subscription in project gravitee-gateway by gravitee-io.

the class CheckSubscriptionPolicyTest method shouldContinue.

@Test
public void shouldContinue() throws PolicyException, TechnicalException {
    CheckSubscriptionPolicy policy = new CheckSubscriptionPolicy();
    Request request = mock(Request.class);
    Response response = mock(Response.class);
    PolicyChain policyChain = mock(PolicyChain.class);
    ExecutionContext executionContext = mock(ExecutionContext.class);
    when(executionContext.getAttribute(CheckSubscriptionPolicy.CONTEXT_ATTRIBUTE_CLIENT_ID)).thenReturn("my-client-id");
    SubscriptionRepository subscriptionRepository = mock(SubscriptionRepository.class);
    when(executionContext.getComponent(SubscriptionRepository.class)).thenReturn(subscriptionRepository);
    Subscription subscription = mock(Subscription.class);
    when(subscription.getClientId()).thenReturn("my-client-id");
    when(subscriptionRepository.search(any(SubscriptionCriteria.class))).thenReturn(Collections.singletonList(subscription));
    policy.onRequest(request, response, policyChain, executionContext);
    verify(policyChain, times(1)).doNext(request, response);
}

Example 12 with Subscription

use of io.gravitee.repository.management.model.Subscription in project gravitee-gateway by gravitee-io.

the class CheckSubscriptionPolicy method onRequest.

@Override
protected void onRequest(Request request, Response response, PolicyChain policyChain, ExecutionContext executionContext) throws PolicyException {
    SubscriptionRepository subscriptionRepository = executionContext.getComponent(SubscriptionRepository.class);
    // Get plan and client_id from execution context
    String clientId = (String) executionContext.getAttribute(CONTEXT_ATTRIBUTE_CLIENT_ID);
    if (clientId == null || clientId.trim().isEmpty()) {
        sendError(response, policyChain, "invalid_client", "No client_id was supplied");
        return;
    }
    String plan = (String) executionContext.getAttribute(ExecutionContext.ATTR_PLAN);
    try {
        List<Subscription> subscriptions = subscriptionRepository.search(new SubscriptionCriteria.Builder().plans(Collections.singleton(plan)).clientId(clientId).status(Subscription.Status.ACCEPTED).build());
        if (subscriptions != null && !subscriptions.isEmpty()) {
            Subscription subscription = subscriptions.get(0);
            if (subscription.getClientId().equals(clientId) && (subscription.getEndingAt() == null || subscription.getEndingAt().after(Date.from(request.timestamp())))) {
                executionContext.setAttribute(ExecutionContext.ATTR_APPLICATION, subscription.getApplication());
                executionContext.setAttribute(ExecutionContext.ATTR_USER_ID, subscription.getId());
                policyChain.doNext(request, response);
                return;
            }
        }
        // As per https://tools.ietf.org/html/rfc6749#section-4.1.2.1
        sendUnauthorized(policyChain, OAUTH2_ERROR_ACCESS_DENIED);
    } catch (TechnicalException te) {
        // As per https://tools.ietf.org/html/rfc6749#section-4.1.2.1
        sendUnauthorized(policyChain, OAUTH2_ERROR_SERVER_ERROR);
    }
}

Example 13 with Subscription

use of io.gravitee.repository.management.model.Subscription in project gravitee-gateway by gravitee-io.

the class SubscriptionRepositoryWrapper method search.

@Override
public List<Subscription> search(SubscriptionCriteria criteria) throws TechnicalException {
    String key = criteria.getPlans().iterator().next() + '-' + criteria.getClientId();
    Element element = this.cache.get(key);
    return (element != null) ? Collections.singletonList((Subscription) element.getObjectValue()) : null;
}

Example 14 with Subscription

use of io.gravitee.repository.management.model.Subscription in project gravitee-gateway by gravitee-io.

the class SubscriptionRefresher method saveOrUpdate.

private void saveOrUpdate(Subscription subscription) {
    String key = subscription.getPlan() + '-' + subscription.getClientId();
    Element element = cache.get(subscription.getId());
    if (subscription.getStatus() == Subscription.Status.CLOSED && element != null) {
        cache.removeElement(element);
        String oldKey = (String) element.getObjectValue();
        Element eltSubscription = cache.get(oldKey);
        if (eltSubscription != null && ((Subscription) eltSubscription.getObjectValue()).getId().equals(subscription.getId())) {
            cache.remove(oldKey);
        }
    } else {
        LOGGER.debug("Cache a subscription: plan[{}] application[{}] client_id[{}]", subscription.getPlan(), subscription.getApplication(), subscription.getClientId());
        cache.put(new Element(subscription.getId(), key));
        cache.put(new Element(key, subscription));
    }
}

Example 15 with Subscription

use of io.gravitee.repository.management.model.Subscription in project gravitee-management-rest-api by gravitee-io.

the class SubscriptionServiceTest method shouldUpdateSubscriptionWithoutEndingDate.

@Test
public void shouldUpdateSubscriptionWithoutEndingDate() throws Exception {
    UpdateSubscriptionEntity updatedSubscription = new UpdateSubscriptionEntity();
    updatedSubscription.setId(SUBSCRIPTION_ID);
    // subscription object is not a mock since its state is updated by the call to subscriptionService.create()
    Subscription subscription = new Subscription();
    subscription.setId(SUBSCRIPTION_ID);
    subscription.setApplication(APPLICATION_ID);
    subscription.setPlan(PLAN_ID);
    subscription.setStatus(Subscription.Status.ACCEPTED);
    // Stub
    when(subscriptionRepository.findById(SUBSCRIPTION_ID)).thenReturn(Optional.of(subscription));
    when(subscriptionRepository.update(any())).thenAnswer(returnsFirstArg());
    when(planService.findById(PLAN_ID)).thenReturn(plan);
    when(plan.getApis()).thenReturn(Collections.singleton(API_ID));
    // Run
    subscriptionService.update(updatedSubscription);
    // Verify
    verify(subscriptionRepository, times(1)).update(any(Subscription.class));
    verify(apiKeyService, never()).findBySubscription(SUBSCRIPTION_ID);
}