use of com.mysql.cj.xdevapi.Schema.Validation in project aws-mysql-jdbc by awslabs.
the class SchemaTest method testCreateCollectionWithOptions.
@Test
public void testCreateCollectionWithOptions() {
String collName1 = "testCreateCollection1";
String collName2 = "testCreateCollection2";
dropCollection(collName1);
dropCollection(collName2);
String sch1 = "{" + "\"id\":\"http://json-schema.org/geo\",\"$schema\":\"http://json-schema.org/draft-06/schema#\"," + "\"description\":\"A geographical coordinate\",\"type\":\"object\",\"properties\":{\"latitude\":{\"type\":\"number\"}," + "\"longitude\":{\"type\":\"number\"}},\"required\":[\"latitude\",\"longitude\"]}";
String sch2 = "{\"id\":\"http://json-schema.org/geo\",\"$schema\":\"http://json-schema.org/draft-06/schema#\"," + "\"description\":\"The geographical coordinate\",\"type\":\"object\",\"properties\":{\"latitude\":{\"type\":\"number\"}," + "\"longitude\":{\"type\":\"number\"}},\"required\":[\"latitude\",\"longitude\"]}";
String sch3 = "{\"id\":\"http://json-schema.org/idx\",\"$schema\":\"http://json-schema.org/draft-06/schema#\"," + "\"type\":\"object\",\"properties\":{\"index\":{\"type\":\"number\"}},\"required\":[\"index\"]}";
try {
if (mysqlVersionMeetsMinimum(ServerVersion.parseVersion("8.0.19"))) {
// TSFR2b.1: Call createCollection with both level and schema options are set. Ensure that given values are set on server.
Collection coll = //
this.schema.createCollection(//
collName1, new CreateCollectionOptions().setReuseExisting(false).setValidation(new Validation().setLevel(ValidationLevel.STRICT).setSchema(sch1)));
SqlResult res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
String def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertTrue(def.contains(sch1));
coll.add("{\"latitude\": 20, \"longitude\": 30}").execute();
Collection coll1 = //
this.schema.createCollection(//
collName1, new CreateCollectionOptions().setReuseExisting(true).setValidation(new Validation().setLevel(ValidationLevel.OFF).setSchema(sch1)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertTrue(def.contains(sch1));
coll1 = //
this.schema.createCollection(//
collName1, new CreateCollectionOptions().setReuseExisting(true).setValidation(new Validation().setSchema(sch2)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertFalse(def.contains(sch2));
assertTrue(def.contains(sch1));
coll1 = //
this.schema.createCollection(//
collName1, new CreateCollectionOptions().setReuseExisting(true).setValidation(new Validation().setLevel(ValidationLevel.STRICT).setSchema(sch1)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertFalse(def.contains(sch2));
assertTrue(def.contains(sch1));
coll1.add("{\"latitude\": 30, \"longitude\": 40}").execute();
// TSFR6d: Call createCollection(String collectionName) and createCollection(String collectionName, boolean reuseExisting) methods against server implementing WL#12965 and ensure they work as before.
this.schema.createCollection(collName2);
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName2 + "`").execute();
def = res.next().getString(1);
assertTrue(def.contains("NOT ENFORCED"));
dropCollection(collName2);
this.schema.createCollection(collName2, new CreateCollectionOptions().setReuseExisting(false));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName2 + "`").execute();
def = res.next().getString(1);
assertTrue(def.contains("NOT ENFORCED"));
dropCollection(collName2);
assertEquals(MysqlErrorNumbers.ER_TABLE_EXISTS_ERROR, assertThrows(XProtocolError.class, new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.createCollection(collName1);
return null;
}
}).getErrorCode());
assertEquals(MysqlErrorNumbers.ER_TABLE_EXISTS_ERROR, assertThrows(XProtocolError.class, new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.createCollection(collName1, false);
return null;
}
}).getErrorCode());
assertEquals(MysqlErrorNumbers.ER_TABLE_EXISTS_ERROR, assertThrows(XProtocolError.class, new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.createCollection(collName1, new CreateCollectionOptions().setReuseExisting(false));
return null;
}
}).getErrorCode());
Collection coll2 = this.schema.createCollection(collName1, true);
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertTrue(def.contains(sch1));
coll2 = this.schema.createCollection(collName1, new CreateCollectionOptions().setReuseExisting(true));
assertEquals(coll, coll2);
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertTrue(def.contains(sch1));
// TSFR2b.2: Call createCollection with only schema option. Ensure the default level value is set on server.
dropCollection(collName2);
this.schema.createCollection(collName2, new CreateCollectionOptions().setValidation(new Validation().setSchema(sch1)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName2 + "`").execute();
def = res.next().getString(1);
if (mysqlVersionMeetsMinimum(ServerVersion.parseVersion("8.0.20"))) {
// After the Bug#30830962 fix when schema is set and level is omitted xplugin sets it to "strict"
assertFalse(def.contains("NOT ENFORCED"));
} else {
assertTrue(def.contains("NOT ENFORCED"));
}
assertTrue(def.contains(sch1));
// TSFR2b.3: Call createCollection with only level option. Ensure the default schema and the given level are set on server.
dropCollection(collName2);
this.schema.createCollection(collName2, new CreateCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.STRICT)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName2 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertTrue(def.contains("{\"type\":\"object\"}"));
// TSFR5: Create collection with json schema and level `strict`. Try to insert document which doesn't match this schema, ensure that a server error being raised.
assertThrows(XProtocolError.class, "ERROR 5180 \\(HY000\\) Document is not valid according to the schema assigned to collection.*", new Callable<Void>() {
public Void call() throws Exception {
coll.add("{\"_id\": 1}").execute();
return null;
}
});
// TSFR2a.1: Call modifyCollection with only level option. Ensure it's changed on server.
this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.OFF)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertTrue(def.contains("NOT ENFORCED"));
assertTrue(def.contains(sch1));
this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.STRICT)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertTrue(def.contains(sch1));
// TSFR2a.2: Call modifyCollection with only schema option. Ensure it's changed on server.
// sch2 is compatible with sch1
this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation().setSchema(sch2)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertFalse(def.contains(sch1));
assertTrue(def.contains(sch2));
assertThrows(XProtocolError.class, "ERROR 5180 \\(HY000\\) Document is not valid according to the schema assigned to collection.*", new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.modifyCollection(collName1, // sch3 is incompatible with sch1
new ModifyCollectionOptions().setValidation(new Validation().setSchema(sch3)));
return null;
}
});
// TSFR2a.3: Call modifyCollection with both level and schema options. Ensure they are changed on server.
this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.OFF).setSchema(sch3)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertTrue(def.contains("NOT ENFORCED"));
assertFalse(def.contains(sch2));
assertTrue(def.contains(sch3));
this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.STRICT).setSchema(sch1)));
res = this.session.sql("SHOW CREATE TABLE `" + this.schema.getName() + "`.`" + collName1 + "`").execute();
def = res.next().getString(1);
assertFalse(def.contains("NOT ENFORCED"));
assertFalse(def.contains(sch3));
assertTrue(def.contains(sch1));
// TSFR2a.4: Call modifyCollection with neither level nor schema options are set. Ensure Connector/J throws the XProtocolError
assertThrows(XProtocolError.class, "ERROR 5020 \\(HY000\\) Arguments value used under \"validation\" must be an object with at least one field", new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.modifyCollection(collName1, null);
return null;
}
});
assertThrows(XProtocolError.class, "ERROR 5020 \\(HY000\\) Arguments value used under \"validation\" must be an object with at least one field", new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.modifyCollection(collName1, new ModifyCollectionOptions());
return null;
}
});
assertThrows(XProtocolError.class, "ERROR 5020 \\(HY000\\) Arguments value used under \"validation\" must be an object with at least one field", new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation()));
return null;
}
});
// TSFR4: Try to create collection with an invalid json schema. Ensure that a server error being raised.
assertThrows(XProtocolError.class, "ERROR 5182 \\(HY000\\) JSON validation schema .*", new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.createCollection("wrongSchema", new CreateCollectionOptions().setValidation(new Validation().setSchema("{\"id\": \"http://json-schema.org/geo\",\"$schema\":\"http://json-schema.org/draft-06/schema#\"," + //
"\"description\":\"The geographical coordinate\",\"type\":\"object\",\"properties\":{\"latitude\":{" + // wrong type
"\"type\":\"blablabla\"" + "}},\"required\":[\"latitude\",\"foo\"]}")));
return null;
}
});
} else {
// for old servers
// TSFR6b: Call createCollection(collectionName, createCollectionOptions) method against server which doesn't support validation parameter for `create_collection` X Protocol command,
// eg. MySQL 5.7. Ensure that server responds with error code 5015 and that Connector/J wraps it to WrongArgumentException with message
// "The server doesn't support the requested operation. Please update the MySQL Server and or Client library".
assertThrows(WrongArgumentException.class, Messages.getString("Schema.CreateCollection"), new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.createCollection(collName1, new CreateCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.STRICT).setSchema("{\"id\": \"http://json-schema.org/geo\",\"$schema\": \"http://json-schema.org/draft-06/schema#\"," + "\"description\": \"A geographical coordinate\",\"type\": \"object\",\"properties\":" + "{\"latitude\": {\"type\": \"number\"},\"longitude\": {\"type\": \"number\"}}," + "\"required\": [\"latitude\", \"longitude\"]}")));
return null;
}
});
// TSFR6c: Call modifyCollection(String collectionName, ModifyCollectionOptions options) method against server which doesn't implement `modify_collection_options` X Protocol command,
// eg. MySQL 5.7. Ensure that server responds with error code 5157 and that Connector/J wraps it to WrongArgumentException with message
// "The server doesn't support the requested operation. Please update the MySQL Server and or Client library".
assertThrows(WrongArgumentException.class, Messages.getString("Schema.CreateCollection"), new Callable<Void>() {
public Void call() throws Exception {
SchemaTest.this.schema.modifyCollection(collName1, new ModifyCollectionOptions().setValidation(new Validation().setLevel(ValidationLevel.OFF)));
return null;
}
});
// TSFR6a: Call createCollection(String collectionName) and createCollection(String collectionName, boolean reuseExisting) methods against servers not implementing WL#12965 and ensure they work as before.
// It's covered by testCreateCollection() test case.
}
} finally {
dropCollection(collName1);
dropCollection(collName2);
}
}
Aggregations