use of org.forgerock.openam.cts.api.tokens.Token in project OpenAM by OpenRock.
the class CoreTokenAdapterTest method shouldUseTaskQueueForUpdate.
@Test
public void shouldUseTaskQueueForUpdate() throws CoreTokenException {
// Given
Token token = new Token("badger", TokenType.SESSION);
// When
adapter.updateOrCreate(token);
// Then
verify(mockTaskDispatcher).update(eq(token), any(ResultHandler.class));
}
use of org.forgerock.openam.cts.api.tokens.Token in project OpenAM by OpenRock.
the class LdapAdapterTest method shouldNoNothingIfNoModificaitonsOnUpdate.
@Test
public void shouldNoNothingIfNoModificaitonsOnUpdate() throws Exception {
// Given
String tokenId = "badger";
Token first = new Token(tokenId, TokenType.OAUTH);
Token second = new Token(tokenId, TokenType.OAUTH);
Connection mockConnection = mock(Connection.class);
LdapDataLayerConfiguration config = mock(LdapDataLayerConfiguration.class);
when(config.getTokenStoreRootSuffix()).thenReturn(DN.valueOf("ou=unit-test"));
LDAPDataConversion dataConversion = new LDAPDataConversion();
LdapTokenAttributeConversion conversion = new LdapTokenAttributeConversion(dataConversion, config);
adapter = new LdapAdapter(conversion, mockQueryVisitor, mockQueryFactory);
// When
adapter.update(mockConnection, first, second);
// Then
verify(mockConnection, never()).modify(any(ModifyRequest.class));
}
use of org.forgerock.openam.cts.api.tokens.Token in project OpenAM by OpenRock.
the class CoreTokenAdapterTest method shouldCreateToken.
@Test
public void shouldCreateToken() throws LdapException, CoreTokenException {
// Given
Token token = new Token("badger", TokenType.SESSION);
// When
adapter.create(token);
// Then
verify(mockTaskDispatcher).create(eq(token), any(ResultHandler.class));
}
use of org.forgerock.openam.cts.api.tokens.Token in project OpenAM by OpenRock.
the class LdapAdapterTest method shouldReturnNullWhenObjectNotFound.
@Test
public void shouldReturnNullWhenObjectNotFound() throws Exception {
// Given
String tokenId = "badger";
DN testDN = DN.rootDN();
SearchRequest request = LDAPRequests.newSingleEntrySearchRequest(testDN);
LdapException exception = LdapException.newLdapException(ResultCode.NO_SUCH_OBJECT);
given(mockConnection.searchSingleEntry(request)).willThrow(exception);
given(mockConversion.generateTokenDN(anyString())).willReturn(testDN);
// When
Token result = adapter.read(mockConnection, tokenId);
// Then
assertThat(result).isNull();
}
use of org.forgerock.openam.cts.api.tokens.Token in project OpenAM by OpenRock.
the class LdapAdapterTest method shouldQuery.
@Test
public void shouldQuery() throws Exception {
// Given
final QueryBuilder<Connection, Filter> mockBuilder = mock(QueryBuilder.class);
given(mockBuilder.withFilter(any(Filter.class))).willAnswer(new Answer<QueryBuilder<Connection, Filter>>() {
@Override
public QueryBuilder<Connection, Filter> answer(InvocationOnMock invocation) throws Throwable {
return mockBuilder;
}
});
given(mockBuilder.execute(any(Connection.class))).willReturn(Arrays.asList((Collection<Token>) Arrays.asList(new Token("weasel", TokenType.OAUTH))).iterator());
given(mockQueryFactory.createInstance()).willReturn(mockBuilder);
QueryFilterVisitor<Filter, Void, CoreTokenField> visitor = mock(QueryFilterVisitor.class);
given(mockQueryFactory.createFilterConverter()).willReturn(visitor);
given(visitor.visitBooleanLiteralFilter(null, true)).willReturn(Filter.alwaysTrue());
// When
TokenFilter filter = new TokenFilterBuilder().withQuery(QueryFilter.<CoreTokenField>alwaysTrue()).build();
Collection<Token> result = adapter.query(mockConnection, filter);
// Then
verify(mockBuilder).withFilter(any(Filter.class));
verify(mockBuilder).execute(mockConnection);
assertThat(result.size()).isEqualTo(1);
assertThat(result.iterator().next().getTokenId()).isEqualTo("weasel");
}
Aggregations