@Test
public void testVDBConnectionType() throws Exception {
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
VDB vdb = admin.getVDB("bqt", 1);
Model model = vdb.getModels().get(0);
admin.updateSource("bqt", 1, "Source", "h2", "java:jboss/datasources/ExampleDS");
try {
// should not be able to remove from non-multisource
admin.removeSource("bqt", 1, model.getName(), "Source");
fail();
} catch (AdminException e) {
}
assertEquals(ConnectionType.BY_VERSION, vdb.getConnectionType());
Connection conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
conn.close();
admin.changeVDBConnectionType("bqt", 1, ConnectionType.NONE);
try {
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
fail("should have failed to connect as no new connections allowed");
} catch (Exception e) {
// pass
}
admin.deploy("bqt2.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt2.vdb")));
admin.updateSource("bqt", 2, "Source", "h2", "java:jboss/datasources/ExampleDS");
conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
conn.close();
admin.changeVDBConnectionType("bqt", 2, ConnectionType.ANY);
conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
conn.close();
vdb = admin.getVDB("bqt", 2);
model = vdb.getModels().get(0);
assertEquals(model.getSourceConnectionJndiName("Source"), "java:jboss/datasources/ExampleDS");
assertEquals(model.getSourceTranslatorName("Source"), "h2");
assertEquals(ConnectionType.ANY, vdb.getConnectionType());
}
@Test
public void testDataRoleMapping() throws Exception {
admin.deploy("bqt2.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt2.vdb")));
VDB vdb = admin.getVDB("bqt", 2);
Model model = vdb.getModels().get(0);
admin.assignToModel(
"bqt", 2, model.getName(), "Source", "h2", "java:jboss/datasources/ExampleDS");
vdb = admin.getVDB("bqt", 2);
assertTrue(vdb.isValid());
List<DataPolicy> policies = vdb.getDataPolicies();
assertEquals(1, policies.size());
DataPolicy dp = policies.get(0);
assertEquals("roleOne", dp.getName());
assertEquals(2, dp.getPermissions().size());
assertTrue(dp.isAllowCreateTemporaryTables());
assertTrue(dp.isAnyAuthenticated());
List<String> roleNames = dp.getMappedRoleNames();
assertArrayEquals(new String[] {"ROLE1", "ROLE2"}, roleNames.toArray());
admin.removeDataRoleMapping("bqt", 2, "roleOne", "ROLE1");
vdb = admin.getVDB("bqt", 2);
policies = vdb.getDataPolicies();
dp = policies.get(0);
roleNames = dp.getMappedRoleNames();
assertArrayEquals(new String[] {"ROLE2"}, roleNames.toArray());
admin.addDataRoleMapping("bqt", 2, "roleOne", "ROLE3");
vdb = admin.getVDB("bqt", 2);
policies = vdb.getDataPolicies();
dp = policies.get(0);
roleNames = dp.getMappedRoleNames();
assertArrayEquals(new String[] {"ROLE2", "ROLE3"}, roleNames.toArray());
admin.setAnyAuthenticatedForDataRole("bqt", 2, "roleOne", false);
vdb = admin.getVDB("bqt", 2);
policies = vdb.getDataPolicies();
dp = policies.get(0);
assertFalse(dp.isAnyAuthenticated());
}
@Test
public void testSystemPropertiesInVDBXML() throws Exception {
String vdbName = "test";
String testVDB =
"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n"
+ "<vdb name=\"test\" version=\"1\">\n"
+ " <property name=\"UseConnectorMetadata\" value=\"${teiid.vdb.UseConnectorMetadata:none}\" />\n"
+ " <model name=\"loopy\">\n"
+ " <source name=\"loop\" translator-name=\"loopy\" />\n"
+ " </model>\n"
+ "</vdb>";
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
JavaArchive jar = getLoopyArchive();
admin.deploy("loopy.jar", jar.as(ZipExporter.class).exportAsInputStream());
// normal load
admin.deploy("test-vdb.xml", new ByteArrayInputStream(testVDB.getBytes()));
AdminUtil.waitForVDBLoad(admin, vdbName, 1, 3);
VDB vdb = admin.getVDB(vdbName, 1);
String value = vdb.getPropertyValue("UseConnectorMetadata");
// see the arquillian.zml file in resources in the JVM proeprties section for the expected value
assertEquals("custom", value);
admin.undeploy("loopy.jar");
admin.undeploy("test-vdb.xml");
}
@Test
public void testErrorDeployment() throws Exception {
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
admin.deploy(
"error-vdb.xml", new FileInputStream(UnitTestUtil.getTestDataFile("error-vdb.xml")));
AdminUtil.waitForVDBLoad(admin, "error", 1, 3);
VDB vdb = admin.getVDB("error", 1);
assertEquals(Status.FAILED, vdb.getStatus());
}
private boolean deployVdb() throws AdminException, FileNotFoundException {
boolean vdbOneDeployed;
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
AdminUtil.waitForVDBLoad(admin, "bqt", 1, 3);
vdbOneDeployed = true;
VDB vdb = admin.getVDB("bqt", 1);
Model model = vdb.getModels().get(0);
admin.assignToModel(
"bqt", 1, model.getName(), "Source", "h2", "java:jboss/datasources/ExampleDS");
return vdbOneDeployed;
}
@Test
public void testVDBDeployment() throws Exception {
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
Collection<String> dsNames = admin.getDataSourceNames();
if (dsNames.contains("Oracle11_PushDS")) {
admin.deleteDataSource("Oracle11_PushDS");
}
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
vdbs = admin.getVDBs();
assertFalse(vdbs.isEmpty());
VDB vdb = admin.getVDB("bqt", 1);
assertFalse(vdb.isValid());
assertTrue(AdminUtil.waitForVDBLoad(admin, "bqt", 1, 3));
assertFalse(vdb.isValid());
Properties props = new Properties();
props.setProperty("connection-url", "jdbc:h2:mem:test;DB_CLOSE_DELAY=-1");
props.setProperty("user-name", "sa");
props.setProperty("password", "sa");
props.setProperty("connection-properties", "foo=bar,blah=blah");
admin.createDataSource("Oracle11_PushDS", "h2", props);
vdb = admin.getVDB("bqt", 1);
assertTrue(vdb.isValid());
assertTrue(vdb.getStatus().equals(Status.ACTIVE));
dsNames = admin.getDataSourceNames();
assertTrue(dsNames.contains("Oracle11_PushDS"));
admin.deleteDataSource("Oracle11_PushDS");
vdb = admin.getVDB("bqt", 1);
assertFalse(vdb.isValid());
}