com.aire.ux.test.spring.SpringTestVaadinServletFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aire-test-spring Show documentation
Sunshower UX Testing Libraries
The newest version!
package com.aire.ux.test.spring;

import com.aire.ux.test.InstantiatorFactory;
import com.aire.ux.test.VaadinServletFactory;
import com.aire.ux.test.vaadin.Frames;
import com.github.mvysny.kaributesting.v10.Routes;
import com.github.mvysny.kaributesting.v10.mock.MockInstantiator;
import com.github.mvysny.kaributesting.v10.mock.MockedUI;
import com.github.mvysny.kaributesting.v10.spring.MockSpringServlet;
import com.github.mvysny.kaributesting.v10.spring.MockSpringServletService;
import com.vaadin.flow.component.UI;
import com.vaadin.flow.di.Instantiator;
import com.vaadin.flow.function.DeploymentConfiguration;
import com.vaadin.flow.server.ServiceException;
import com.vaadin.flow.server.VaadinServlet;
import com.vaadin.flow.server.VaadinServletService;
import com.vaadin.flow.spring.SpringInstantiator;
import java.util.Optional;
import java.util.ServiceLoader;
import java.util.function.Supplier;
import lombok.val;
import org.springframework.test.context.junit.jupiter.SpringExtension;

public class SpringTestVaadinServletFactory implements VaadinServletFactory {

  public int getOrder() {
    return 50;
  }

  @Override
  public Supplier getUIFactory() {
    return () -> {
      try {
        return SpringExtension.getApplicationContext(Frames.currentContext()).getBean(UI.class);
      } catch (Exception ex) {
        return new MockedUI();
      }
    };
  }

  @Override
  public Optional createServlet(Routes routes) {
    val context = SpringExtension.getApplicationContext(Frames.currentContext());
    return Optional.of(
        new MockSpringServlet(routes, context, () -> getUIFactory().get()) {

          @Override
          protected VaadinServletService createServletService(
              DeploymentConfiguration deploymentConfiguration) throws ServiceException {
            val service =
                new MockSpringServletService(this, deploymentConfiguration, ctx, uiFactory) {
                  @Override
                  public Instantiator getInstantiator() {
                    val factories =
                        ServiceLoader.load(
                                InstantiatorFactory.class,
                                Thread.currentThread().getContextClassLoader())
                            .iterator();
                    Instantiator current =
                        new MockInstantiator(new SpringInstantiator(this, context));
                    while (factories.hasNext()) {
                      current = factories.next().create(current);
                    }
                    return current;
                  }
                };
            service.init();
            routes.register(service.getContext());
            return service;
          }
        });
  }
}