Examples with SnowflakeTableListProperties org.talend.components.snowflake.SnowflakeTableListProperties used on opensource projects

Example 1 with SnowflakeTableListProperties

use of org.talend.components.snowflake.SnowflakeTableListProperties in project components by Talend.

the class SpringSnowflakeTestIT method testWizard.

@Test
public void testWizard() throws Throwable {
    final List<RepoProps> repoProps = new ArrayList<>();
    Repository repo = new TestRepository(repoProps);
    getComponentService().setRepository(repo);
    Set<ComponentWizardDefinition> wizards = getComponentService().getTopLevelComponentWizards();
    int connectionWizardNumber = 0;
    ComponentWizardDefinition wizardDef = null;
    for (ComponentWizardDefinition wizardDefinition : wizards) {
        if (wizardDefinition instanceof SnowflakeConnectionWizardDefinition) {
            wizardDef = wizardDefinition;
            connectionWizardNumber++;
        }
    }
    assertEquals(1, connectionWizardNumber);
    assertEquals("Snowflake Connection", wizardDef.getMenuItemName());
    ComponentWizard connectionWizard = getComponentService().getComponentWizard(SnowflakeConnectionWizardDefinition.COMPONENT_WIZARD_NAME, "nodeSnowflake");
    assertNotNull(connectionWizard);
    assertEquals("nodeSnowflake", connectionWizard.getRepositoryLocation());
    List<Form> forms = connectionWizard.getForms();
    Form connFormWizard = forms.get(0);
    assertEquals("Wizard", connFormWizard.getName());
    assertFalse(connFormWizard.isAllowBack());
    assertFalse(connFormWizard.isAllowForward());
    assertFalse(connFormWizard.isAllowFinish());
    // Main from SnowflakeTableListProperties
    assertEquals("Main", forms.get(1).getName());
    assertEquals("Snowflake Connection Settings", connFormWizard.getTitle());
    assertEquals("Complete these fields in order to connect to your Snowflake account.", connFormWizard.getSubtitle());
    SnowflakeConnectionProperties connProps = (SnowflakeConnectionProperties) connFormWizard.getProperties();
    Form advancedForm = connProps.getForm(Form.ADVANCED);
    assertTrue(((PresentationItem) connFormWizard.getWidget("advanced").getContent()).getFormtoShow() + " should be == to " + advancedForm, ((PresentationItem) connFormWizard.getWidget("advanced").getContent()).getFormtoShow() == advancedForm);
    assertEquals("Name", connProps.getProperty("name").getDisplayName());
    connProps.name.setValue("connName");
    setupProps(connProps);
    Form userPassword = (Form) connFormWizard.getWidget("userPassword").getContent();
    Property passwordSe = (Property) userPassword.getWidget("password").getContent();
    assertEquals("Password", passwordSe.getDisplayName());
    // $NON-NLS-1$
    NamedThing nameProp = connFormWizard.getWidget("name").getContent();
    assertEquals("Name", nameProp.getDisplayName());
    PropertiesTestUtils.checkAndValidate(getComponentService(), connFormWizard, "testConnection", connProps);
    assertTrue(connFormWizard.isAllowForward());
    Form modForm = forms.get(1);
    SnowflakeTableListProperties mlProps = (SnowflakeTableListProperties) modForm.getProperties();
    assertFalse(modForm.isCallAfterFormBack());
    assertFalse(modForm.isCallAfterFormNext());
    assertTrue(modForm.isCallAfterFormFinish());
    assertTrue(modForm.isCallBeforeFormPresent());
    assertFalse(modForm.isAllowBack());
    assertFalse(modForm.isAllowForward());
    assertFalse(modForm.isAllowFinish());
    mlProps = (SnowflakeTableListProperties) getComponentService().beforeFormPresent(modForm.getName(), mlProps);
    assertTrue(modForm.isAllowBack());
    assertFalse(modForm.isAllowForward());
    assertTrue(modForm.isAllowFinish());
    List<NamedThing> all = mlProps.selectedTableNames.getValue();
    assertNull(all);
    List<NamedThing> possibleValues = (List<NamedThing>) mlProps.selectedTableNames.getPossibleValues();
    LOGGER.info("possibleValues: " + possibleValues);
    assertEquals(1, possibleValues.size());
    List<NamedThing> selected = new ArrayList<>();
    selected.add(possibleValues.get(0));
    mlProps.selectedTableNames.setValue(selected);
    getComponentService().afterFormFinish(modForm.getName(), mlProps);
    LOGGER.debug(repoProps.toString());
    assertEquals(2, repoProps.size());
    int i = 0;
    for (RepoProps rp : repoProps) {
        if (i == 0) {
            assertEquals("connName", rp.name);
            SnowflakeConnectionProperties storedConnProps = (SnowflakeConnectionProperties) rp.props;
            assertEquals(USER, storedConnProps.userPassword.userId.getValue());
            assertEquals(PASSWORD, storedConnProps.userPassword.password.getValue());
        } else {
            SnowflakeTableProperties storedModule = (SnowflakeTableProperties) rp.props;
            assertEquals(selected.get(i - 1).getName(), storedModule.tableName.getValue());
            assertTrue(rp.schema.getFields().size() == NUM_COLUMNS);
            assertThat(storedModule.main.schema.getStringValue(), Matchers.is(rp.schema.toString()));
        }
        i++;
    }
}