GrantTest.java

/*-
 * #%L
 * JSQLParser library
 * %%
 * Copyright (C) 2004 - 2019 JSQLParser
 * %%
 * Dual licensed under GNU LGPL 2.1 or Apache License 2.0
 * #L%
 */
package net.sf.jsqlparser.statement.grant;

import java.io.StringReader;
import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.parser.CCJSqlParserManager;
import static net.sf.jsqlparser.test.TestUtils.*;
import static net.sf.jsqlparser.test.TestUtils.assertSqlCanBeParsedAndDeparsed;
import static org.junit.jupiter.api.Assertions.assertEquals;
import org.junit.jupiter.api.Test;

public class GrantTest {

    private final CCJSqlParserManager parserManager = new CCJSqlParserManager();

    @Test
    public void testGrantPrivilege() throws JSQLParserException {
        String statement = "GRANT SELECT ON t1 TO u";
        Grant grant = (Grant) parserManager.parse(new StringReader(statement));
        assertEquals("t1", grant.getObjectName());

        assertEquals(1, grant.getPrivileges().size());
        assertEquals("SELECT", grant.getPrivileges().get(0));

        assertEquals(1, grant.getUsers().size());
        assertEquals("u", grant.getUsers().get(0));

        assertEquals(statement, grant.toString());
        assertEquals(null, grant.getRole());

        Grant created = new Grant().addPrivileges("SELECT").withObjectName("t1").addUsers("u");
        assertDeparse(created, statement);
        assertEqualsObjectTree(grant, created);
    }

    @Test
    public void testGrantPrivileges() throws JSQLParserException {
        String statement = "GRANT SELECT, INSERT ON t1 TO u, u2";
        Grant grant = (Grant) parserManager.parse(new StringReader(statement));
        assertEquals("t1", grant.getObjectName());
        assertEquals(2, grant.getPrivileges().size());
        assertEquals(true, grant.getPrivileges().stream().anyMatch(s -> s.equals("SELECT")));
        assertEquals(true, grant.getPrivileges().stream().anyMatch(s -> s.equals("INSERT")));

        assertEquals(2, grant.getUsers().size());
        assertEquals(true, grant.getUsers().stream().anyMatch(s -> s.equals("u")));
        assertEquals(true, grant.getUsers().stream().anyMatch(s -> s.equals("u2")));

        assertEquals(statement, grant.toString());
        assertEquals(null, grant.getRole());

        Grant created = new Grant().addPrivileges(asList("SELECT", "INSERT")).withObjectName("t1")
                .addUsers(asList("u", "u2"));
        assertDeparse(created, statement);
        assertEqualsObjectTree(grant, created);
    }

    @Test
    public void testGrantRole() throws JSQLParserException {
        String statement = "GRANT role1 TO u, u2";
        Grant grant = (Grant) parserManager.parse(new StringReader(statement));

        assertEquals(null, grant.getObjectName());
        assertEquals(null, grant.getPrivileges());

        assertEquals(2, grant.getUsers().size());
        assertEquals(true, grant.getUsers().stream().anyMatch(s -> s.equals("u")));
        assertEquals(true, grant.getUsers().stream().anyMatch(s -> s.equals("u2")));

        assertEquals("role1", grant.getRole());
        assertEquals(statement, grant.toString());

        Grant created = new Grant().withRole("role1")
                .addUsers(asList("u", "u2"));
        assertDeparse(created, statement);
        assertEqualsObjectTree(grant, created);
    }

    @Test
    public void testGrantQueryWithPrivileges() throws JSQLParserException {
        assertSqlCanBeParsedAndDeparsed("GRANT SELECT, INSERT, UPDATE, DELETE ON T1 TO ADMIN_ROLE");
    }

    @Test
    public void testGrantQueryWithRole() throws JSQLParserException {
        assertSqlCanBeParsedAndDeparsed("GRANT ROLE_1 TO TEST_ROLE_1, TEST_ROLE_2");
    }

    @Test
    public void testGrantSchemaParsingIssue1080() throws JSQLParserException {
        assertSqlCanBeParsedAndDeparsed("GRANT SELECT ON schema_name.table_name TO XYZ");
    }

    @Test
    void testPublicKeywordIssue2230() throws JSQLParserException {
        String sqlStr = "grant select on da380_now to public;";
        assertSqlCanBeParsedAndDeparsed(sqlStr, true);
    }
}