Examples with ModifyCollectionOptions com.mysql.cj.xdevapi.Schema.ModifyCollectionOptions used on opensource projects

Example 1 with ModifyCollectionOptions

use of com.mysql.cj.xdevapi.Schema.ModifyCollectionOptions 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);
    }
}