/*
* Copyright 2010-2012 Luca Garulli (l.garulli--at--orientechnologies.com)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.orientechnologies.orient.test.database.auto;
import com.orientechnologies.orient.client.db.ODatabaseHelper;
import com.orientechnologies.orient.core.db.ODatabaseRecordThreadLocal;
import com.orientechnologies.orient.core.db.document.ODatabaseDocumentTx;
import com.orientechnologies.orient.core.exception.OSchemaException;
import com.orientechnologies.orient.core.exception.OValidationException;
import com.orientechnologies.orient.core.metadata.schema.OClass;
import com.orientechnologies.orient.core.metadata.schema.OSchema;
import com.orientechnologies.orient.core.metadata.schema.OType;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.core.sql.OCommandSQL;
import com.orientechnologies.orient.core.sql.OCommandSQLParsingException;
import com.orientechnologies.orient.core.sql.query.OSQLSynchQuery;
import com.orientechnologies.orient.core.storage.OStorage;
import com.orientechnologies.orient.enterprise.channel.binary.OResponseProcessingException;
import org.testng.Assert;
import org.testng.annotations.Optional;
import org.testng.annotations.Parameters;
import org.testng.annotations.Test;
import java.io.IOException;
import java.util.List;
@Test(groups = "schema")
public class SchemaTest extends DocumentDBBaseTest {
@Parameters(value = "url")
public SchemaTest(@Optional String url) {
super(url);
}
public void createSchema() throws IOException {
database = new ODatabaseDocumentTx(url);
if (ODatabaseHelper.existsDatabase(database, "plocal"))
database.open("admin", "admin");
else
database.create();
if (database.getMetadata().getSchema().existsClass("Account"))
return;
Assert.assertNotNull(database.getMetadata().getSchema().getClass("ORIDs"));
createBasicTestSchema();
database.close();
}
@Test(dependsOnMethods = "createSchema")
public void checkSchema() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
OSchema schema = database.getMetadata().getSchema();
assert schema != null;
assert schema.getClass("Profile") != null;
assert schema.getClass("Profile").getProperty("nick").getType() == OType.STRING;
assert schema.getClass("Profile").getProperty("name").getType() == OType.STRING;
assert schema.getClass("Profile").getProperty("surname").getType() == OType.STRING;
assert schema.getClass("Profile").getProperty("registeredOn").getType() == OType.DATETIME;
assert schema.getClass("Profile").getProperty("lastAccessOn").getType() == OType.DATETIME;
assert schema.getClass("Whiz") != null;
assert schema.getClass("whiz").getProperty("account").getType() == OType.LINK;
assert schema.getClass("whiz").getProperty("account").getLinkedClass().getName().equalsIgnoreCase("Account");
assert schema.getClass("WHIZ").getProperty("date").getType() == OType.DATE;
assert schema.getClass("WHIZ").getProperty("text").getType() == OType.STRING;
assert schema.getClass("WHIZ").getProperty("text").isMandatory();
assert schema.getClass("WHIZ").getProperty("text").getMin().equals("1");
assert schema.getClass("WHIZ").getProperty("text").getMax().equals("140");
assert schema.getClass("whiz").getProperty("replyTo").getType() == OType.LINK;
assert schema.getClass("Whiz").getProperty("replyTo").getLinkedClass().getName().equalsIgnoreCase("Account");
database.close();
}
@Test(dependsOnMethods = "checkSchema")
public void checkSchemaApi() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
OSchema schema = database.getMetadata().getSchema();
try {
Assert.assertNull(schema.getClass("Animal33"));
} catch (OSchemaException e) {
}
database.close();
}
@Test(dependsOnMethods = "checkSchemaApi")
public void checkClusters() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
for (OClass cls : database.getMetadata().getSchema().getClasses()) {
if (!cls.isAbstract())
assert database.getClusterNameById(cls.getDefaultClusterId()) != null;
}
database.close();
}
@Test(dependsOnMethods = "createSchema")
public void checkDatabaseSize() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
Assert.assertTrue(database.getSize() > 0);
database.close();
}
@Test(dependsOnMethods = "createSchema")
public void checkTotalRecords() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
Assert.assertTrue(database.getStorage().countRecords() > 0);
database.close();
}
@Test(expectedExceptions = OValidationException.class)
public void checkErrorOnUserNoPasswd() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
database.getMetadata().getSecurity().createUser("error", null, (String) null);
database.close();
}
@Test
public void testMultiThreadSchemaCreation() throws InterruptedException {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
Thread thread = new Thread(new Runnable() {
@Override
public void run() {
ODatabaseRecordThreadLocal.INSTANCE.set(database);
ODocument doc = new ODocument("NewClass");
database.save(doc);
doc.delete();
database.getMetadata().getSchema().dropClass("NewClass");
database.close();
}
});
thread.start();
thread.join();
}
@Test
public void createAndDropClassTestApi() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
final String testClassName = "dropTestClass";
final int clusterId;
OClass dropTestClass = database.getMetadata().getSchema().createClass(testClassName);
clusterId = dropTestClass.getDefaultClusterId();
database.getMetadata().getSchema().reload();
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNotNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), clusterId);
Assert.assertNotNull(database.getClusterNameById(clusterId));
database.close();
database = null;
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNotNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), clusterId);
Assert.assertNotNull(database.getClusterNameById(clusterId));
database.getMetadata().getSchema().dropClass(testClassName);
database.getMetadata().getSchema().reload();
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), -1);
Assert.assertNull(database.getClusterNameById(clusterId));
database.close();
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), -1);
Assert.assertNull(database.getClusterNameById(clusterId));
database.close();
}
@Test
public void createAndDropClassTestCommand() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
final String testClassName = "dropTestClass";
final int clusterId;
OClass dropTestClass = database.getMetadata().getSchema().createClass(testClassName);
clusterId = dropTestClass.getDefaultClusterId();
database.getMetadata().getSchema().reload();
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNotNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), clusterId);
Assert.assertNotNull(database.getClusterNameById(clusterId));
database.close();
database = null;
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNotNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), clusterId);
Assert.assertNotNull(database.getClusterNameById(clusterId));
database.command(new OCommandSQL("drop class " + testClassName)).execute();
database.reload();
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), -1);
Assert.assertNull(database.getClusterNameById(clusterId));
database.close();
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
dropTestClass = database.getMetadata().getSchema().getClass(testClassName);
Assert.assertNull(dropTestClass);
Assert.assertEquals(database.getStorage().getClusterIdByName(testClassName), -1);
Assert.assertNull(database.getClusterNameById(clusterId));
database.close();
}
@Test(dependsOnMethods = "createSchema")
public void customAttributes() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
// TEST CUSTOM PROPERTY CREATION
database.getMetadata().getSchema().getClass("Profile").getProperty("nick").setCustom("stereotype", "icon");
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("stereotype"), "icon");
// TEST CUSTOM PROPERTY EXISTS EVEN AFTER REOPEN
database.close();
database.open("admin", "admin");
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("stereotype"), "icon");
// TEST CUSTOM PROPERTY REMOVAL
database.getMetadata().getSchema().getClass("Profile").getProperty("nick").setCustom("stereotype", null);
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("stereotype"), null);
// TEST CUSTOM PROPERTY UPDATE
database.getMetadata().getSchema().getClass("Profile").getProperty("nick").setCustom("stereotype", "polygon");
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("stereotype"),
"polygon");
// TEST CUSTOM PROPERTY UDPATED EVEN AFTER REOPEN
database.close();
database.open("admin", "admin");
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("stereotype"),
"polygon");
database.close();
// TEST CUSTOM PROPERTY WITH =
database.open("admin", "admin");
database.getMetadata().getSchema().getClass("Profile").getProperty("nick").setCustom("equal", "this = that");
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("equal"),
"this = that");
database.close();
// TEST CUSTOM PROPERTY WITH = AFTER REOPEN
database.open("admin", "admin");
Assert.assertEquals(database.getMetadata().getSchema().getClass("Profile").getProperty("nick").getCustom("equal"),
"this = that");
database.close();
}
@Test(dependsOnMethods = "createSchema")
public void alterAttributes() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
OClass company = database.getMetadata().getSchema().getClass("Company");
OClass superClass = company.getSuperClass();
Assert.assertNotNull(superClass);
boolean found = false;
for (OClass c : superClass.getBaseClasses()) {
if (c.equals(company)) {
found = true;
break;
}
}
Assert.assertEquals(found, true);
company.setSuperClass(null);
Assert.assertNull(company.getSuperClass());
for (OClass c : superClass.getBaseClasses()) {
Assert.assertNotSame(c, company);
}
database.command(new OCommandSQL("alter class " + company.getName() + " superclass " + superClass.getName())).execute();
database.getMetadata().getSchema().reload();
company = database.getMetadata().getSchema().getClass("Company");
superClass = company.getSuperClass();
Assert.assertNotNull(company.getSuperClass());
found = false;
for (OClass c : superClass.getBaseClasses()) {
if (c.equals(company)) {
found = true;
break;
}
}
Assert.assertEquals(found, true);
database.close();
}
@Test
public void invalidClusterWrongClusterId() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
try {
database.command(new OCommandSQL("create class Antani cluster 212121")).execute();
Assert.fail();
} catch (Exception e) {
if (e instanceof OResponseProcessingException)
e = (Exception) e.getCause();
Assert.assertTrue(e instanceof OCommandSQLParsingException);
} finally {
database.close();
}
}
@Test
public void invalidClusterWrongClusterName() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
try {
database.command(new OCommandSQL("create class Antani cluster blaaa")).execute();
Assert.fail();
} catch (Exception e) {
if (e instanceof OResponseProcessingException)
e = (Exception) e.getCause();
Assert.assertTrue(e instanceof OCommandSQLParsingException);
} finally {
database.close();
}
}
@Test
public void invalidClusterWrongKeywords() {
database = new ODatabaseDocumentTx(url);
database.open("admin", "admin");
try {
database.command(new OCommandSQL("create class Antani the pen is on the table")).execute();
Assert.fail();
} catch (Exception e) {
if (e instanceof OResponseProcessingException)
e = (Exception) e.getCause();
Assert.assertTrue(e instanceof OCommandSQLParsingException);
} finally {
database.close();
}
}
@Test
public void testRenameClass() {
ODatabaseDocumentTx databaseDocumentTx = new ODatabaseDocumentTx(url);
databaseDocumentTx.open("admin", "admin");
OClass oClass = databaseDocumentTx.getMetadata().getSchema().createClass("RenameClassTest");
ODocument document = new ODocument("RenameClassTest");
document.save();
document.reset();
document.setClassName("RenameClassTest");
document.save();
List<ODocument> result = databaseDocumentTx.query(new OSQLSynchQuery<ODocument>("select from RenameClassTest"));
Assert.assertEquals(result.size(), 2);
oClass.set(OClass.ATTRIBUTES.NAME, "RenameClassTest2");
result = databaseDocumentTx.query(new OSQLSynchQuery<ODocument>("select from RenameClassTest2"));
Assert.assertEquals(result.size(), 2);
}
public void testMinimumClustersAndClusterSelection() {
ODatabaseDocumentTx databaseDocumentTx = new ODatabaseDocumentTx(url);
databaseDocumentTx.open("admin", "admin");
databaseDocumentTx.command(new OCommandSQL("alter database minimumclusters 3")).execute();
try {
databaseDocumentTx.command(new OCommandSQL("create class multipleclusters")).execute();
databaseDocumentTx.close();
databaseDocumentTx.open("admin", "admin");
databaseDocumentTx.reload();
Assert.assertFalse(databaseDocumentTx.existsCluster("multipleclusters"));
for (int i = 0; i < 3; ++i) {
Assert.assertTrue(databaseDocumentTx.existsCluster("multipleclusters_" + i));
}
for (int i = 0; i < 6; ++i) {
new ODocument("multipleclusters").field("num", i).save();
}
// CHECK THERE ARE 2 RECORDS IN EACH CLUSTER (ROUND-ROBIN STRATEGY)
for (int i = 0; i < 3; ++i) {
Assert.assertEquals(
databaseDocumentTx.countClusterElements(databaseDocumentTx.getClusterIdByName("multipleclusters_" + i)), 2);
}
// DELETE ALL THE RECORDS
int deleted = databaseDocumentTx.command(new OCommandSQL("delete from cluster:multipleclusters_2")).execute();
Assert.assertEquals(deleted, 2);
// CHANGE CLASS STRATEGY to BALANCED
databaseDocumentTx.command(new OCommandSQL("alter class multipleclusters clusterselection balanced")).execute();
databaseDocumentTx.reload();
databaseDocumentTx.getMetadata().getSchema().reload();
for (int i = 0; i < 2; ++i) {
new ODocument("multipleclusters").field("num", i).save();
}
Assert.assertEquals(databaseDocumentTx.countClusterElements(databaseDocumentTx.getClusterIdByName("multipleclusters_2")), 2);
} finally {
// RESTORE DEFAULT
databaseDocumentTx.command(new OCommandSQL("alter database minimumclusters 1")).execute();
}
}
public void testExchangeCluster() {
ODatabaseDocumentTx databaseDocumentTx = new ODatabaseDocumentTx(url);
databaseDocumentTx.open("admin", "admin");
try {
databaseDocumentTx.command(new OCommandSQL("CREATE CLASS TestRenameClusterOriginal")).execute();
swapClusters(databaseDocumentTx, 1);
swapClusters(databaseDocumentTx, 2);
swapClusters(databaseDocumentTx, 3);
} finally {
databaseDocumentTx.close();
}
}
private void swapClusters(ODatabaseDocumentTx databaseDocumentTx, int i) {
databaseDocumentTx.command(new OCommandSQL("CREATE CLASS TestRenameClusterNew extends TestRenameClusterOriginal")).execute();
databaseDocumentTx.command(new OCommandSQL("INSERT INTO TestRenameClusterNew (iteration) VALUES(" + i + ")")).execute();
databaseDocumentTx.command(new OCommandSQL("ALTER CLASS TestRenameClusterOriginal removecluster TestRenameClusterOriginal"))
.execute();
databaseDocumentTx.command(new OCommandSQL("ALTER CLASS TestRenameClusterNew removecluster TestRenameClusterNew")).execute();
databaseDocumentTx.command(new OCommandSQL("DROP CLASS TestRenameClusterNew")).execute();
databaseDocumentTx.command(new OCommandSQL("ALTER CLASS TestRenameClusterOriginal addcluster TestRenameClusterNew")).execute();
databaseDocumentTx.command(new OCommandSQL("DROP CLUSTER TestRenameClusterOriginal")).execute();
databaseDocumentTx.command(new OCommandSQL("ALTER CLUSTER TestRenameClusterNew name TestRenameClusterOriginal")).execute();
List<ODocument> result = databaseDocumentTx.query(new OSQLSynchQuery<ODocument>("select * from TestRenameClusterOriginal"));
Assert.assertEquals(result.size(), 1);
ODocument document = result.get(0);
Assert.assertEquals(document.field("iteration"), i);
}
}