/*
* *********************************************************************************************************************
*
* TheseFoolishThings: Miscellaneous utilities
* http://tidalwave.it/projects/thesefoolishthings
*
* Copyright (C) 2009 - 2024 by Tidalwave s.a.s. (http://tidalwave.it)
*
* *********************************************************************************************************************
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
* specific language governing permissions and limitations under the License.
*
* *********************************************************************************************************************
*
* git clone https://bitbucket.org/tidalwave/thesefoolishthings-src
* git clone https://github.com/tidalwave-it/thesefoolishthings-src
*
* *********************************************************************************************************************
*/
package it.tidalwave.util.mock;
import javax.annotation.Nonnull;
import java.util.Collection;
import java.util.Collections;
import it.tidalwave.util.As;
import it.tidalwave.role.impl.AsDelegate;
import it.tidalwave.role.spi.OwnerRoleFactory;
import it.tidalwave.role.spi.OwnerRoleFactoryProvider;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import static org.mockito.Mockito.*;
/***********************************************************************************************************************
*
* A provider of static factory methods for creating mocks with {@link As} support.
*
* @author Fabrizio Giudici
*
**********************************************************************************************************************/
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public final class MockAsFactory
{
/*******************************************************************************************************************
*
* Creates a mock with Mockito that fully supports {@link As}.
*
* @param clazz the mock class
* @return the mock
* @since 3.2-ALPHA-3 (refactored)
*
******************************************************************************************************************/
@Nonnull
public static <T extends As> T mockWithAs (@Nonnull final Class<T> clazz)
{
return mockWithAs(clazz, Collections.emptyList());
}
/*******************************************************************************************************************
*
* Creates a mock with Mockito that fully supports {@link As}. This method doesn't call
* {@link OwnerRoleFactoryProvider#getInstance()}.
*
* @param clazz the mock class
* @param roles a collection of roles or factories for roles
* @return the mock
* @since 3.2-ALPHA-3 (refactored)
*
******************************************************************************************************************/
@Nonnull
public static <T extends As> T mockWithAs (@Nonnull final Class<T> clazz, @Nonnull final Collection<Object> roles)
{
final var mock = mock(clazz);
final var orf = new OwnerRoleFactory()
{
public <U> Collection<U> findRoles (@Nonnull final Class<? extends U> roleType)
{
return Collections.emptyList();
}
};
final var as = new AsDelegate(o -> orf, mock, roles);
when(mock.as(any(Class.class))).thenCallRealMethod();
when(mock.maybeAs(any(Class.class))).thenAnswer(i -> as.maybeAs((Class<?>)i.getArguments()[0]));
when(mock.asMany(any(Class.class))).thenAnswer(i -> as.asMany((Class<?>)i.getArguments()[0]));
when(mock.as(any(As.Type.class))).thenCallRealMethod();