@Test
public void testFlatMapMaxConcurrent() {
final int m = 4;
final AtomicInteger subscriptionCount = new AtomicInteger();
Observable<Integer> source =
Observable.range(1, 10)
.flatMap(
new Func1<Integer, Observable<Integer>>() {
@Override
public Observable<Integer> call(Integer t1) {
return compose(Observable.range(t1 * 10, 2), subscriptionCount, m)
.subscribeOn(Schedulers.computation());
}
},
m);
TestSubscriber<Integer> ts = new TestSubscriber<Integer>();
source.subscribe(ts);
ts.awaitTerminalEvent();
ts.assertNoErrors();
Set<Integer> expected =
new HashSet<Integer>(
Arrays.asList(
10, 11, 20, 21, 30, 31, 40, 41, 50, 51, 60, 61, 70, 71, 80, 81, 90, 91, 100, 101));
Assert.assertEquals(expected.size(), ts.getOnNextEvents().size());
Assert.assertTrue(expected.containsAll(ts.getOnNextEvents()));
}
@Test
public void shouldGetRightsForAUserOnHomeFacilityAndProgram() throws Exception {
Long userId = 1L;
Facility facility = new Facility(2L);
Program program = new Program(3L);
List<Right> expected = asList(CREATE_REQUISITION);
when(facilityService.getHomeFacility(userId)).thenReturn(facility);
when(roleRightsRepository.getRightsForUserOnHomeFacilityAndProgram(userId, program))
.thenReturn(expected);
Set<Right> result =
roleRightsService.getRightsForUserAndFacilityProgram(userId, facility, program);
assertThat(result.containsAll(expected), is(true));
verify(roleRightsRepository).getRightsForUserOnHomeFacilityAndProgram(userId, program);
}
@Test
public void shouldGetRightsForAUserOnSupervisedFacilityAndProgram() throws Exception {
Long userId = 1L;
Facility facility = new Facility(2L);
Program program = new Program(3L);
List<Right> expected = asList(CREATE_REQUISITION);
SupervisoryNode supervisoryNode = new SupervisoryNode(4L);
List<SupervisoryNode> supervisoryNodes = asList(supervisoryNode);
when(supervisoryNodeService.getFor(facility, program)).thenReturn(supervisoryNode);
when(supervisoryNodeService.getAllParentSupervisoryNodesInHierarchy(supervisoryNode))
.thenReturn(supervisoryNodes);
when(roleRightsRepository.getRightsForUserOnSupervisoryNodeAndProgram(
userId, supervisoryNodes, program))
.thenReturn(expected);
Set<Right> result =
roleRightsService.getRightsForUserAndFacilityProgram(userId, facility, program);
verify(roleRightsRepository)
.getRightsForUserOnSupervisoryNodeAndProgram(userId, supervisoryNodes, program);
assertThat(result.containsAll(expected), is(true));
}