use of org.zalando.nakadi.security.ClientResolver in project nakadi by zalando.
the class StoragesControllerTest method before.
@Before
public void before() {
final StoragesController controller = new StoragesController(securitySettings, storageService, adminService);
final FeatureToggleService featureToggleService = mock(FeatureToggleService.class);
doReturn("nakadi").when(securitySettings).getAdminClientId();
mockMvc = standaloneSetup(controller).setMessageConverters(new StringHttpMessageConverter(), TestUtils.JACKSON_2_HTTP_MESSAGE_CONVERTER).setCustomArgumentResolvers(new ClientResolver(securitySettings, featureToggleService)).build();
}
use of org.zalando.nakadi.security.ClientResolver in project nakadi by zalando.
the class EventTypeControllerTestCase method init.
@Before
public void init() throws Exception {
final NakadiSettings nakadiSettings = new NakadiSettings(0, 0, 0, TOPIC_RETENTION_TIME_MS, 0, 60, NAKADI_POLL_TIMEOUT, NAKADI_SEND_TIMEOUT, 0, NAKADI_EVENT_MAX_BYTES, NAKADI_SUBSCRIPTION_MAX_PARTITIONS, "service", "nakadi", "I am warning you");
final PartitionsCalculator partitionsCalculator = new KafkaConfig().createPartitionsCalculator("t2.large", TestUtils.OBJECT_MAPPER, nakadiSettings);
when(timelineService.getTopicRepository((Timeline) any())).thenReturn(topicRepository);
when(timelineService.getTopicRepository((EventTypeBase) any())).thenReturn(topicRepository);
when(transactionTemplate.execute(any())).thenAnswer(invocation -> {
final TransactionCallback callback = (TransactionCallback) invocation.getArguments()[0];
return callback.doInTransaction(null);
});
final EventTypeService eventTypeService = new EventTypeService(eventTypeRepository, timelineService, partitionResolver, enrichment, subscriptionRepository, schemaEvolutionService, partitionsCalculator, featureToggleService, authorizationValidator, timelineSync, transactionTemplate, nakadiSettings, nakadiKpiPublisher, "et-log-event-type");
final EventTypeOptionsValidator eventTypeOptionsValidator = new EventTypeOptionsValidator(TOPIC_RETENTION_MIN_MS, TOPIC_RETENTION_MAX_MS);
final EventTypeController controller = new EventTypeController(eventTypeService, featureToggleService, eventTypeOptionsValidator, applicationService, adminService, nakadiSettings);
doReturn(randomUUID).when(uuid).randomUUID();
doReturn(true).when(applicationService).exists(any());
doReturn(true).when(featureToggleService).isFeatureEnabled(CHECK_PARTITIONS_KEYS);
mockMvc = standaloneSetup(controller).setMessageConverters(new StringHttpMessageConverter(), TestUtils.JACKSON_2_HTTP_MESSAGE_CONVERTER).setCustomArgumentResolvers(new ClientResolver(settings, featureToggleService)).setControllerAdvice(new ExceptionHandling()).build();
}
use of org.zalando.nakadi.security.ClientResolver in project nakadi by zalando.
the class PartitionsControllerTest method before.
@Before
public void before() throws InternalNakadiException, NoSuchEventTypeException {
eventTypeRepositoryMock = mock(EventTypeRepository.class);
topicRepositoryMock = mock(TopicRepository.class);
eventTypeCache = mock(EventTypeCache.class);
timelineService = Mockito.mock(TimelineService.class);
cursorOperationsService = Mockito.mock(CursorOperationsService.class);
when(timelineService.getActiveTimelinesOrdered(eq(UNKNOWN_EVENT_TYPE))).thenThrow(NoSuchEventTypeException.class);
when(timelineService.getActiveTimelinesOrdered(eq(TEST_EVENT_TYPE))).thenReturn(Collections.singletonList(TIMELINE));
when(timelineService.getAllTimelinesOrdered(eq(TEST_EVENT_TYPE))).thenReturn(Collections.singletonList(TIMELINE));
when(timelineService.getTopicRepository((Timeline) any())).thenReturn(topicRepositoryMock);
final CursorConverter cursorConverter = new CursorConverterImpl(eventTypeCache, timelineService);
final PartitionsController controller = new PartitionsController(timelineService, cursorConverter, cursorOperationsService, eventTypeRepositoryMock, authorizationValidator);
settings = mock(SecuritySettings.class);
final FeatureToggleService featureToggleService = Mockito.mock(FeatureToggleService.class);
mockMvc = standaloneSetup(controller).setMessageConverters(new StringHttpMessageConverter(), TestUtils.JACKSON_2_HTTP_MESSAGE_CONVERTER).setCustomArgumentResolvers(new ClientResolver(settings, featureToggleService)).setControllerAdvice(new ExceptionHandling()).build();
}
use of org.zalando.nakadi.security.ClientResolver in project nakadi by zalando.
the class EventPublishingControllerTest method setUp.
@Before
public void setUp() {
metricRegistry = new MetricRegistry();
publisher = mock(EventPublisher.class);
eventTypeMetricRegistry = new EventTypeMetricRegistry(metricRegistry);
kpiPublisher = mock(NakadiKpiPublisher.class);
settings = mock(SecuritySettings.class);
when(settings.getAuthMode()).thenReturn(OFF);
when(settings.getAdminClientId()).thenReturn("nakadi");
blacklistService = Mockito.mock(BlacklistService.class);
when(blacklistService.isProductionBlocked(any(), any())).thenReturn(false);
final FeatureToggleService featureToggleService = Mockito.mock(FeatureToggleService.class);
final EventPublishingController controller = new EventPublishingController(publisher, eventTypeMetricRegistry, blacklistService, kpiPublisher, "kpiEventTypeName");
mockMvc = standaloneSetup(controller).setMessageConverters(new StringHttpMessageConverter(), TestUtils.JACKSON_2_HTTP_MESSAGE_CONVERTER).setCustomArgumentResolvers(new ClientResolver(settings, featureToggleService)).build();
}
use of org.zalando.nakadi.security.ClientResolver in project nakadi by zalando.
the class EventStreamControllerTest method setup.
@Before
public void setup() throws NakadiException, UnknownHostException, InvalidCursorException {
EVENT_TYPE.setName(TEST_EVENT_TYPE_NAME);
timeline = buildTimelineWithTopic(TEST_TOPIC);
eventTypeRepository = mock(EventTypeRepository.class);
topicRepositoryMock = mock(TopicRepository.class);
adminService = mock(AdminService.class);
authorizationService = mock(AuthorizationService.class);
when(topicRepositoryMock.topicExists(TEST_TOPIC)).thenReturn(true);
eventStreamFactoryMock = mock(EventStreamFactory.class);
eventTypeCache = mock(EventTypeCache.class);
requestMock = mock(HttpServletRequest.class);
when(requestMock.getRemoteAddr()).thenReturn(InetAddress.getLoopbackAddress().getHostAddress());
when(requestMock.getRemotePort()).thenReturn(12345);
responseMock = mock(HttpServletResponse.class);
metricRegistry = new MetricRegistry();
streamMetrics = new MetricRegistry();
final EventConsumer.LowLevelConsumer eventConsumerMock = mock(EventConsumer.LowLevelConsumer.class);
when(topicRepositoryMock.createEventConsumer(eq(KAFKA_CLIENT_ID), any())).thenReturn(eventConsumerMock);
final ClosedConnectionsCrutch crutch = mock(ClosedConnectionsCrutch.class);
when(crutch.listenForConnectionClose(requestMock)).thenReturn(new AtomicBoolean(true));
blacklistService = Mockito.mock(BlacklistService.class);
Mockito.when(blacklistService.isConsumptionBlocked(any(), any())).thenReturn(false);
final ConsumerLimitingService consumerLimitingService = Mockito.mock(ConsumerLimitingService.class);
when(consumerLimitingService.acquireConnectionSlots(any(), any(), any())).thenReturn(ImmutableList.of());
featureToggleService = mock(FeatureToggleService.class);
timelineService = mock(TimelineService.class);
when(timelineService.getTopicRepository((Timeline) any())).thenReturn(topicRepositoryMock);
when(timelineService.getTopicRepository((EventTypeBase) any())).thenReturn(topicRepositoryMock);
when(timelineService.getTopicRepository((Storage) any())).thenReturn(topicRepositoryMock);
when(timelineService.getActiveTimelinesOrdered(any())).thenReturn(Collections.singletonList(timeline));
when(timelineService.getAllTimelinesOrdered(any())).thenReturn(Collections.singletonList(timeline));
authorizationValidator = mock(AuthorizationValidator.class);
eventTypeChangeListener = mock(EventTypeChangeListener.class);
when(eventTypeChangeListener.registerListener(any(), any())).thenReturn(mock(Closeable.class));
controller = new EventStreamController(eventTypeRepository, timelineService, TestUtils.OBJECT_MAPPER, eventStreamFactoryMock, metricRegistry, streamMetrics, crutch, blacklistService, consumerLimitingService, featureToggleService, new CursorConverterImpl(eventTypeCache, timelineService), authorizationValidator, eventTypeChangeListener, null);
settings = mock(SecuritySettings.class);
when(settings.getAuthMode()).thenReturn(OFF);
when(settings.getAdminClientId()).thenReturn("nakadi");
mockMvc = standaloneSetup(controller).setMessageConverters(new StringHttpMessageConverter(), TestUtils.JACKSON_2_HTTP_MESSAGE_CONVERTER).setCustomArgumentResolvers(new ClientResolver(settings, featureToggleService)).build();
}
Aggregations