How to use TestDeviceSpecificationsBuilder method of com.testsigma.specification.TestDeviceSpecificationsBuilder class

Best Testsigma code snippet using com.testsigma.specification.TestDeviceSpecificationsBuilder.TestDeviceSpecificationsBuilder

Source:TestDeviceService.java

...18import com.testsigma.repository.TestDeviceRepository;19import com.testsigma.repository.TestDeviceSuiteRepository;20import com.testsigma.specification.SearchCriteria;21import com.testsigma.specification.SearchOperation;22import com.testsigma.specification.TestDeviceSpecificationsBuilder;23import lombok.RequiredArgsConstructor;24import lombok.extern.log4j.Log4j2;25import org.springframework.beans.factory.annotation.Autowired;26import org.springframework.context.annotation.Lazy;27import org.springframework.data.domain.Page;28import org.springframework.data.domain.PageRequest;29import org.springframework.data.domain.Pageable;30import org.springframework.data.jpa.domain.Specification;31import org.springframework.data.repository.query.Param;32import org.springframework.stereotype.Service;33import java.io.IOException;34import java.sql.Timestamp;35import java.util.*;36import java.util.stream.Collectors;37@Log4j238@Service39@RequiredArgsConstructor(onConstructor = @__({@Autowired, @Lazy}))40public class TestDeviceService extends XMLExportService<TestDevice> {41  private final TestDeviceRepository testDeviceRepository;42  private final TestDeviceSuiteService suiteMappingService;43  private final TestDeviceSuiteRepository testDeviceSuiteRepository;44  private final ExportTestDeviceMapper mapper;45  private final TestPlanService testPlanService;46  public List<TestDevice> findByTargetMachine(Long agentId) {47    return testDeviceRepository.findTestDeviceByAgentId(agentId);48  }49  public List<TestDevice> findByTestPlanIdAndDisable(Long testPlanId, Boolean disable) {50    return testDeviceRepository.findByTestPlanIdAndDisable(testPlanId, disable);51  }52  public List<TestDevice> findByTestPlanId(Long testPlanId) {53    return testDeviceRepository.findByTestPlanId(testPlanId);54  }55  public TestDevice find(Long id) throws TestsigmaDatabaseException {56    return testDeviceRepository.findById(id).orElseThrow(57      () -> new TestsigmaDatabaseException("Could not find resource with id:" + id));58  }59  public Page<TestDevice> findAll(Specification<TestDevice> spec, Pageable pageable) {60    return this.testDeviceRepository.findAll(spec, pageable);61  }62  public TestDevice create(TestDevice testDevice) {63    testDevice.setCreatedDate(new Timestamp(Calendar.getInstance().getTimeInMillis()));64    testDevice = this.testDeviceRepository.save(testDevice);65    this.handleEnvironmentSuiteMappings(testDevice);66    return testDevice;67  }68  public TestDevice update(TestDevice testDevice) {69    testDevice.setUpdatedDate(new Timestamp(Calendar.getInstance().getTimeInMillis()));70    List<Long> suiteIds = testDevice.getSuiteIds();71    testDevice = this.testDeviceRepository.save(testDevice);72    testDevice.setSuiteIds(suiteIds);73    this.handleEnvironmentSuiteMappings(testDevice);74    //this.suiteMappingService.save(executionEnvironment);75    return testDevice;76  }77  public void delete(Set<Long> executionEnvironmentIds) {78    this.testDeviceRepository.deleteAllByIds(executionEnvironmentIds);79  }80  public void handleEnvironmentSuiteMappings(TestDevice testDevice) {81    int position = 0;82    List<TestDeviceSuite> newMappings = new ArrayList<>();83    List<TestDeviceSuite> updatedMappings = new ArrayList<>();84    this.cleanupOrphanSuiteMappings(testDevice);85    if (testDevice.getSuiteIds().size() > 0) {86      List<TestDeviceSuite> mappings = this.testDeviceSuiteRepository.findByTestDeviceIdAndSuiteIds(testDevice.getId(), testDevice.getSuiteIds());87      for (Long suiteId : testDevice.getSuiteIds()) {88        TestDeviceSuite testDeviceSuite = new TestDeviceSuite();89        Optional<TestDeviceSuite> existing = mappings.stream().filter(mapping -> mapping.getSuiteId().equals(suiteId)).findFirst();90        position++;91        if (existing.isPresent()) {92          testDeviceSuite = existing.get();93          if (!testDeviceSuite.getPosition().equals(position)) {94            testDeviceSuite.setPosition(position);95            testDeviceSuite = this.suiteMappingService.update(testDeviceSuite);96            updatedMappings.add(testDeviceSuite);97          }98        } else {99          testDeviceSuite.setTestDeviceId(testDevice.getId());100          testDeviceSuite.setSuiteId(suiteId);101          testDeviceSuite.setPosition(position);102          testDeviceSuite = this.suiteMappingService.add(testDeviceSuite);103          newMappings.add(testDeviceSuite);104        }105      }106    }107    testDevice.setUpdatedSuiteIds(updatedMappings);108    testDevice.setAddedSuiteIds(newMappings);109  }110  private void cleanupOrphanSuiteMappings(TestDevice testDevice) {111    List<Long> suiteIdsFromUI = testDevice.getSuiteIds();112    List<TestDeviceSuite> existingSuites = suiteMappingService.findAllByTestDeviceId(testDevice.getId());113    List<Long> existingSuiteIds = existingSuites.stream().map(TestDeviceSuite::getSuiteId).collect(Collectors.toList());114    if (suiteIdsFromUI == null) {115      suiteIdsFromUI = existingSuiteIds;116      testDevice.setSuiteIds(existingSuiteIds);117    }118    existingSuiteIds.removeAll(suiteIdsFromUI);119    if (existingSuiteIds.size() > 0)120      this.deleteAllByTestDeviceAndSuiteIds(testDevice, existingSuiteIds);121  }122  private void deleteAllByTestDeviceAndSuiteIds(TestDevice testDevice, List<Long> existingSuiteIds) {123    List<TestDeviceSuite> mappings = this.testDeviceSuiteRepository.findByTestDeviceIdAndSuiteIds(testDevice.getId(), existingSuiteIds);124    testDevice.setRemovedSuiteIds(mappings);125    this.suiteMappingService.deleteAll(mappings);126  }127  public List<TestDevice> findAllByAgentDeviceIds(List<Long> removedAgentDeviceIds) {128    return this.testDeviceRepository.findAllByDeviceIdIn(removedAgentDeviceIds);129  }130  public void export(BackupDTO backupDTO) throws IOException, ResourceNotFoundException {131    if (!backupDTO.getIsTestDeviceEnabled()) return;132    log.debug("backup process for execution environment initiated");133    writeXML("test_devices", backupDTO, PageRequest.of(0, 25));134    log.debug("backup process for execution environment completed");135  }136  @Override137  protected List<TestDeviceXMLDTO> mapToXMLDTOList(List<TestDevice> list) {138    return mapper.mapEnvironments(list);139  }140  public Specification<TestDevice> getExportXmlSpecification(BackupDTO backupDTO) {141    List<TestPlan> testPlanList = testPlanService.findAllByWorkspaceVersionId(backupDTO.getWorkspaceVersionId());142    List<Long> testPlanIds = testPlanList.stream().map(execution -> execution.getId()).collect(Collectors.toList());143    SearchCriteria criteria = new SearchCriteria("testPlanId", SearchOperation.IN, testPlanIds);144    List<SearchCriteria> params = new ArrayList<>();145    params.add(criteria);146    TestDeviceSpecificationsBuilder testDeviceSpecificationsBuilder = new TestDeviceSpecificationsBuilder();147    testDeviceSpecificationsBuilder.params = params;148    return testDeviceSpecificationsBuilder.build();149  }150  public void resentAppUploadIdToNull(Long appUploadId){151    this.testDeviceRepository.resentAppUploadIdToNull(appUploadId);152  }153  public void resetAgentIdToNull(Long agentId) {154    this.testDeviceRepository.resetAgentIdToNull(agentId);155  }156  public List<TestDevice> findAllByTestSuiteId(Long id) {157    return testDeviceRepository.findAllByTestSuiteId(id);158  }159}...

Source:AgentService.java

...21import com.testsigma.repository.AgentRepository;22import com.testsigma.specification.AgentSpecificationsBuilder;23import com.testsigma.specification.SearchCriteria;24import com.testsigma.specification.SearchOperation;25import com.testsigma.specification.TestDeviceSpecificationsBuilder;26import com.testsigma.util.HttpClient;27import com.testsigma.web.request.AgentRequest;28import lombok.NonNull;29import lombok.RequiredArgsConstructor;30import lombok.extern.log4j.Log4j2;31import org.apache.http.Header;32import org.apache.http.HttpHeaders;33import org.apache.http.message.BasicHeader;34import org.json.JSONObject;35import org.springframework.beans.factory.annotation.Autowired;36import org.springframework.context.ApplicationEventPublisher;37import org.springframework.context.annotation.Lazy;38import org.springframework.data.domain.Page;39import org.springframework.data.domain.PageRequest;40import org.springframework.data.domain.Pageable;41import org.springframework.data.jpa.domain.Specification;42import org.springframework.stereotype.Service;43import java.io.IOException;44import java.sql.Timestamp;45import java.util.ArrayList;46import java.util.Date;47import java.util.List;48import java.util.UUID;49import java.util.stream.Collectors;50@Log4j251@Service52@RequiredArgsConstructor(onConstructor = @__({@Autowired, @Lazy}))53public class AgentService extends XMLExportService<Agent> {54  private final AgentRepository agentRepository;55  private final AgentMapper mapper;56  private final ApplicationEventPublisher applicationEventPublisher;57  private final AgentDeviceService agentDeviceService;58  private final TestPlanService testPlanService;59  private final TestDeviceService testDeviceService;60  private final AgentMapper exportAgentMapper;61  private final HttpClient httpClient;62  private final ApplicationConfig applicationConfig;63  private final JWTTokenService jwtTokenService;64  public Agent find(Long id) throws ResourceNotFoundException {65    return agentRepository.findById(id).orElseThrow(() -> new ResourceNotFoundException("Agent is not found with id:" + id));66  }67  public Agent create(@NonNull Agent agent) {68    agent.setUpdatedDate(new Timestamp(new Date().getTime()));69    agent.setCreatedDate(new Timestamp(new Date().getTime()));70    agent.setUniqueId(UUID.randomUUID().toString());71    agent = agentRepository.save(agent);72    return agent;73  }74  public Agent create(@NonNull AgentRequest agentRequest) throws TestsigmaException {75    Agent agent = mapper.map(agentRequest);76    agent = create(agent);77    return agent;78  }79  public void createLocalAgent(Agent agent) throws TestsigmaException {80    agent = create(agent);81    String url = applicationConfig.getLocalAgentUrl() +"/api/v1/" + agent.getUniqueId() + "/register?jwtApiKey="82      + agent.generateJwtApiKey(jwtTokenService.getServerUuid());83    httpClient.put(url, getHeaders(), new JSONObject(), new TypeReference<>() {84    });85  }86  public Page<Agent> findAll(Specification<Agent> specification, Pageable pageable) {87    return agentRepository.findAll(specification, pageable);88  }89  public void destroy(@NonNull Agent agent) {90    List<AgentDevice> agentDevices = agentDeviceService.findAllByAgent(agent.getId());91    agentDevices.forEach(agentDevice -> agentDeviceService.publishEvent(agentDevice, EventType.DELETE));92    agentRepository.delete(agent);93    publishEvent(agent, EventType.DELETE);94  }95  public boolean isAgentActive(Long agentId) throws ResourceNotFoundException {96    Agent agent = find(agentId);97    long lastUpdatedTime = agent.getUpdatedDate().getTime();98    long currentTime = java.lang.System.currentTimeMillis();99    return currentTime - lastUpdatedTime <= 10 * 60 * 1000;100  }101  public Agent findByUniqueId(@NonNull String uniqueId) throws ResourceNotFoundException {102    Agent agent = null;103    try {104      agent = agentRepository.findByUniqueId(uniqueId);105    } catch (Exception e) {106      log.error(e.getMessage(), e);107    }108    if (agent == null) {109      throw new ResourceNotFoundException("Agent is not found");110    }111    return agent;112  }113  public Agent update(@NonNull AgentRequest agentRequest, String uniqueId) throws ResourceNotFoundException {114    boolean isRegistered = false;115    Agent db = findByUniqueId(uniqueId);116    if (db.getOsType() != null) {117      isRegistered = true;118    }119    mapper.map(agentRequest, db);120    db.setUpdatedDate(new Timestamp(java.lang.System.currentTimeMillis()));121    db = agentRepository.save(db);122    if (!isRegistered && db.getOsType() != null) {123      publishEvent(db, EventType.CREATE);124    }125    return db;126  }127  public void publishEvent(Agent agent, EventType eventType) {128    AgentEvent<Agent> event = createEvent(agent, eventType);129    log.info("Publishing event - " + event.toString());130    applicationEventPublisher.publishEvent(event);131  }132  public AgentEvent<Agent> createEvent(Agent agent, EventType eventType) {133    AgentEvent<Agent> event = new AgentEvent<>();134    event.setEventData(agent);135    event.setEventType(eventType);136    return event;137  }138  public void export(BackupDTO backupDTO) throws IOException, ResourceNotFoundException {139    if (!backupDTO.getIsAgentEnabled()) return;140    log.debug("backup process for agent initiated");141    writeXML("agent", backupDTO, PageRequest.of(0, 25));142    log.debug("backup process for agent completed");143  }144  @Override145  protected List<AgentXMLDTO> mapToXMLDTOList(List<Agent> list) {146    return exportAgentMapper.mapAgents(list);147  }148  @Override149  public Specification<Agent> getExportXmlSpecification(BackupDTO backupDTO) throws ResourceNotFoundException {150    List<TestPlan> testPlanList = testPlanService.findAllByWorkspaceVersionId(backupDTO.getWorkspaceVersionId());151    List<Long> testPlanIds = testPlanList.stream().map(testPlan -> testPlan.getId()).collect(Collectors.toList());152    SearchCriteria criteria = new SearchCriteria("testPlanId", SearchOperation.IN, testPlanIds);153    List<SearchCriteria> params = new ArrayList<>();154    params.add(criteria);155    TestDeviceSpecificationsBuilder testDeviceSpecificationsBuilder = new TestDeviceSpecificationsBuilder();156    testDeviceSpecificationsBuilder.params = params;157    Page<TestDevice> page = testDeviceService.findAll(testDeviceSpecificationsBuilder.build(), PageRequest.of(0, 100));158    List<Long> agentIds = page.getContent().stream().map(testDevice -> {159      if (testDevice.getAgent() != null) {160        return testDevice.getAgent().getId();161      }162      return null;163    }).collect(Collectors.toList());164    AgentSpecificationsBuilder agentSpecificationsBuilder = new AgentSpecificationsBuilder();165    SearchCriteria systems = new SearchCriteria("id", SearchOperation.IN, agentIds);166    params = new ArrayList<>();167    params.add(systems);168    agentSpecificationsBuilder.params = params;169    return agentSpecificationsBuilder.build();...

Source:TestDeviceSpecificationsBuilder.java

...7package com.testsigma.specification;8import com.testsigma.model.TestDevice;9import org.springframework.data.jpa.domain.Specification;10import java.util.ArrayList;11public class TestDeviceSpecificationsBuilder extends BaseSpecificationsBuilder {12  private Specification<TestDevice> result;13  public TestDeviceSpecificationsBuilder() {14    super(new ArrayList<>());15  }16  public Specification<TestDevice> build() {17    if (params.size() == 0) {18      return null;19    }20    params.forEach((searchCriteria) -> {21      if (searchCriteria.getKey().equals("appUploadId")) {22        String[] appIds = searchCriteria.getValue().toString().split("#");23        result = new TestDeviceSpecification(new SearchCriteria(searchCriteria.getKey(), searchCriteria.getOperation(), appIds[0]));24        for (String appId : appIds) {25          result = result.or(new TestDeviceSpecification(new SearchCriteria(searchCriteria.getKey(), searchCriteria.getOperation(), appId)));26        }27      } else {...