mockit.internal.expectations.state.CascadingTypes Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation

JMockit is a Java toolkit for automated developer testing. It contains APIs for the creation of the objects to be tested, for mocking dependencies, and for faking external APIs; JUnit (4 & 5) and TestNG test runners are supported. It also contains an advanced code coverage tool.

There is a newer version: 1.49

Show newest version

/*
 * Copyright (c) 2006 Rogério Liesenfeld
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.expectations.state;

import java.lang.reflect.*;
import java.util.*;
import java.util.concurrent.*;
import javax.annotation.*;

import mockit.external.asm.*;
import static mockit.internal.util.Utilities.*;

public final class CascadingTypes
{
   @Nonnull private final Map mockedTypesToCascades;

   CascadingTypes() { mockedTypesToCascades = new ConcurrentHashMap(4); }

   public void add(boolean fromMockField, @Nonnull Type mockedType)
   {
      Class mockedClass = getClassType(mockedType);
      String mockedTypeDesc = JavaType.getInternalName(mockedClass);
      add(mockedTypeDesc, fromMockField, mockedType);
   }

   @Nonnull
   MockedTypeCascade add(@Nonnull String mockedTypeDesc, boolean fromMockField, @Nonnull Type mockedType)
   {
      MockedTypeCascade cascade = mockedTypesToCascades.get(mockedType);

      if (cascade == null) {
         cascade = new MockedTypeCascade(fromMockField, mockedType, mockedTypeDesc);
         mockedTypesToCascades.put(mockedType, cascade);
      }

      return cascade;
   }

   @Nonnull
   MockedTypeCascade getCascade(@Nonnull Type mockedType) { return mockedTypesToCascades.get(mockedType); }

   @Nullable
   MockedTypeCascade getCascade(@Nonnull String mockedTypeDesc, @Nullable Object mockInstance)
   {
      if (mockedTypesToCascades.isEmpty()) {
         return null;
      }

      if (mockInstance != null) {
         MockedTypeCascade cascade = findCascadeForInstance(mockInstance);

         if (cascade != null) {
            return cascade;
         }
      }

      for (MockedTypeCascade cascade : mockedTypesToCascades.values()) {
         if (cascade.mockedTypeDesc.equals(mockedTypeDesc)) {
            return cascade;
         }
      }

      return null;
   }

   @Nullable
   private MockedTypeCascade findCascadeForInstance(@Nonnull Object mockInstance)
   {
      for (MockedTypeCascade cascade : mockedTypesToCascades.values()) {
         if (cascade.hasInstance(mockInstance)) {
            return cascade;
         }
      }

      return null;
   }

   void clearNonSharedCascadingTypes()
   {
      if (!mockedTypesToCascades.isEmpty()) {
         Iterator itr = mockedTypesToCascades.values().iterator();

         while (itr.hasNext()) {
            MockedTypeCascade cascade = itr.next();

            if (cascade.fromMockField) {
               cascade.discardCascadedMocks();
            }
            else {
               itr.remove();
            }
         }
      }
   }

   public void clear() { mockedTypesToCascades.clear(); }

   void addInstance(@Nonnull Type mockedType, @Nonnull Object cascadingInstance)
   {
      MockedTypeCascade cascade = mockedTypesToCascades.get(mockedType);

      if (cascade != null) {
         cascade.addInstance(cascadingInstance);
      }
   }
}