Examples with FileNamesDataConfig com.axway.ats.agent.core.threading.data.config.FileNamesDataConfig used on opensource projects

Example 1 with FileNamesDataConfig

use of com.axway.ats.agent.core.threading.data.config.FileNamesDataConfig in project ats-framework by Axway.

the class Test_ParameterDataProviderFactory method createFileNamesProvider.

@Test
public void createFileNamesProvider() throws AgentException, URISyntaxException {
    URL testFileURL = Test_ParameterDataProviderFactory.class.getResource("/testfolder/classloader.html");
    File testFile = new File(testFileURL.toURI());
    FileNamesDataConfig fileNamesConfig = new FileNamesDataConfig("param1", testFile.getParent(), true, ParameterProviderLevel.PER_INVOCATION);
    ParameterDataProvider dataProvider = ParameterDataProviderFactory.createDataProvider(fileNamesConfig);
    assertEquals(FileNamesParameterDataProvider.class, dataProvider.getClass());
    ArgumentValue generatedValue = dataProvider.getValue(new ArrayList<ArgumentValue>());
    assertEquals("param1", generatedValue.getName());
    assertTrue(((String) generatedValue.getValue()).endsWith(".html"));
}

Example 2 with FileNamesDataConfig

use of com.axway.ats.agent.core.threading.data.config.FileNamesDataConfig in project ats-framework by Axway.

the class ParameterDataProviderFactory method createDataProvider.

/**
     * Create a data provider based on a configuration provider
     *
     * @param parameterConfig the configuration based on which to create the provider
     * @return the newly created provider
     * @throws ParameterDataProviderNotSupportedException if the provider is not supported
     * @throws ParameterDataProviderInitalizationException if there is an error during provider initialization
     */
public static ParameterDataProvider createDataProvider(ParameterDataConfig parameterConfig) throws ParameterDataProviderNotSupportedException, ParameterDataProviderInitalizationException {
    ParameterDataProvider parameterDataProvider;
    Class<?> parameterConfigClass = parameterConfig.getClass();
    if (parameterConfigClass == RangeDataConfig.class) {
        RangeDataConfig rangeDataConfig = (RangeDataConfig) parameterConfig;
        //choose the right one, based on the arguments
        if (rangeDataConfig.getStaticValue() != null) {
            parameterDataProvider = new StringRangeParameterDataProvider(rangeDataConfig.getParameterName(), rangeDataConfig.getStaticValue(), (Integer) rangeDataConfig.getRangeStart(), (Integer) rangeDataConfig.getRangeEnd(), parameterConfig.getParameterProviderLevel());
        } else {
            parameterDataProvider = new IntegerRangeParameterDataProvider(rangeDataConfig.getParameterName(), (Integer) rangeDataConfig.getRangeStart(), (Integer) rangeDataConfig.getRangeEnd(), parameterConfig.getParameterProviderLevel());
        }
    } else if (parameterConfigClass == ListDataConfig.class) {
        ListDataConfig listDataConfig = (ListDataConfig) parameterConfig;
        parameterDataProvider = new ListParameterDataProvider(listDataConfig.getParameterName(), listDataConfig.getValues(), listDataConfig.getParameterProviderLevel());
    } else if (parameterConfigClass == FileNamesDataConfig.class) {
        FileNamesDataConfig fileNamesDataConfig = (FileNamesDataConfig) parameterConfig;
        parameterDataProvider = new FileNamesParameterDataProvider(fileNamesDataConfig.getParameterName(), fileNamesDataConfig.getFileContainers(), fileNamesDataConfig.getRecursiveSearch(), fileNamesDataConfig.getReturnFullPath(), fileNamesDataConfig.getParameterProviderLevel());
    } else if (parameterConfigClass == UsernameDataConfig.class) {
        UsernameDataConfig nameDataConfig = (UsernameDataConfig) parameterConfig;
        if (nameDataConfig.getValues() != null) {
            parameterDataProvider = new ListParameterDataProvider(nameDataConfig.getParameterName(), nameDataConfig.getValues(), nameDataConfig.getParameterProviderLevel(), UsernameDataConfig.class);
        } else {
            parameterDataProvider = new StringRangeParameterDataProvider(nameDataConfig.getParameterName(), nameDataConfig.getStaticValue(), (Integer) nameDataConfig.getRangeStart(), (Integer) nameDataConfig.getRangeEnd(), parameterConfig.getParameterProviderLevel(), UsernameDataConfig.class);
        }
    } else if (parameterConfigClass.getGenericSuperclass() == CustomParameterDataConfig.class) {
        // custom data provider
        CustomParameterDataConfig customDataConfig = (CustomParameterDataConfig) parameterConfig;
        try {
            // find the constructor of the custom data provider
            Constructor<? extends CustomParameterDataProvider> constructor = customDataConfig.getDataProviderClass().getDeclaredConstructor(String.class, Map.class, ParameterProviderLevel.class);
            // call the constructor of the custom data provider
            parameterDataProvider = (CustomParameterDataProvider) constructor.newInstance(customDataConfig.getParameterName(), customDataConfig.getControlTokens(), customDataConfig.getParameterProviderLevel());
        } catch (Exception e) {
            throw new ParameterDataProviderNotSupportedException(parameterConfigClass);
        }
    } else {
        throw new ParameterDataProviderNotSupportedException(parameterConfigClass);
    }
    //initialize the newly created provider
    parameterDataProvider.initialize();
    //and return it back to the caller
    return parameterDataProvider;
}