Search in sources :

Example 1 with Block

use of org.teiid.query.sql.proc.Block in project teiid by teiid.

the class TestSQLStringVisitor method testBlock1.

@Test
public void testBlock1() {
    Delete d1 = new Delete();
    // $NON-NLS-1$
    d1.setGroup(new GroupSymbol("g"));
    CommandStatement cmdStmt = new CommandStatement(d1);
    // $NON-NLS-1$
    AssignmentStatement assigStmt = new AssignmentStatement(new ElementSymbol("a"), new Constant(new Integer(1)));
    // $NON-NLS-1$
    RaiseStatement errStmt = new RaiseStatement(new Constant("My Error"));
    Block b = new Block();
    b.addStatement(cmdStmt);
    b.addStatement(assigStmt);
    b.addStatement(errStmt);
    // $NON-NLS-1$
    helpTest(b, "BEGIN\nDELETE FROM g;\na = 1;\nRAISE 'My Error';\nEND");
}
Also used : CommandStatement(org.teiid.query.sql.proc.CommandStatement) AssignmentStatement(org.teiid.query.sql.proc.AssignmentStatement) Block(org.teiid.query.sql.proc.Block) RaiseStatement(org.teiid.query.sql.proc.RaiseStatement) Test(org.junit.Test)

Example 2 with Block

use of org.teiid.query.sql.proc.Block in project teiid by teiid.

the class TestSQLStringVisitor method testCreateUpdateProcedure1.

@Test
public void testCreateUpdateProcedure1() {
    Delete d1 = new Delete();
    // $NON-NLS-1$
    d1.setGroup(new GroupSymbol("g"));
    CommandStatement cmdStmt = new CommandStatement(d1);
    // $NON-NLS-1$
    AssignmentStatement assigStmt = new AssignmentStatement(new ElementSymbol("a"), new Constant(new Integer(1)));
    // $NON-NLS-1$
    RaiseStatement errStmt = new RaiseStatement(new Constant("My Error"));
    Block b = new Block();
    b.addStatement(cmdStmt);
    b.addStatement(assigStmt);
    b.addStatement(errStmt);
    CreateProcedureCommand cup = new CreateProcedureCommand(b);
    // $NON-NLS-1$
    helpTest(cup, "BEGIN\nDELETE FROM g;\na = 1;\nRAISE 'My Error';\nEND");
}
Also used : CommandStatement(org.teiid.query.sql.proc.CommandStatement) CreateProcedureCommand(org.teiid.query.sql.proc.CreateProcedureCommand) AssignmentStatement(org.teiid.query.sql.proc.AssignmentStatement) Block(org.teiid.query.sql.proc.Block) RaiseStatement(org.teiid.query.sql.proc.RaiseStatement) Test(org.junit.Test)

Example 3 with Block

use of org.teiid.query.sql.proc.Block in project teiid by teiid.

the class TestSQLStringVisitor method testCreateUpdateProcedure3.

@Test
public void testCreateUpdateProcedure3() {
    Delete d1 = new Delete();
    // $NON-NLS-1$
    d1.setGroup(new GroupSymbol("g"));
    CommandStatement cmdStmt = new CommandStatement(d1);
    // $NON-NLS-1$
    AssignmentStatement assigStmt = new AssignmentStatement(new ElementSymbol("a"), new Constant(new Integer(1)));
    // $NON-NLS-1$
    RaiseStatement errStmt = new RaiseStatement(new Constant("My Error"));
    Block b = new Block();
    b.addStatement(cmdStmt);
    b.addStatement(assigStmt);
    b.addStatement(errStmt);
    CreateProcedureCommand cup = new CreateProcedureCommand(b);
    // $NON-NLS-1$
    helpTest(cup, "BEGIN\nDELETE FROM g;\na = 1;\nRAISE 'My Error';\nEND");
}
Also used : CommandStatement(org.teiid.query.sql.proc.CommandStatement) CreateProcedureCommand(org.teiid.query.sql.proc.CreateProcedureCommand) AssignmentStatement(org.teiid.query.sql.proc.AssignmentStatement) Block(org.teiid.query.sql.proc.Block) RaiseStatement(org.teiid.query.sql.proc.RaiseStatement) Test(org.junit.Test)

Example 4 with Block

use of org.teiid.query.sql.proc.Block in project teiid by teiid.

the class TestIfStatement method sample2.

public static final IfStatement sample2() {
    Block ifBlock = TestBlock.sample2();
    Block elseBlock = TestBlock.sample1();
    Criteria criteria = TestSetCriteria.sample2();
    return new IfStatement(criteria, ifBlock, elseBlock);
}
Also used : IfStatement(org.teiid.query.sql.proc.IfStatement) Block(org.teiid.query.sql.proc.Block)

Example 5 with Block

use of org.teiid.query.sql.proc.Block in project teiid by teiid.

the class TestIfStatement method sample1.

// ################################## TEST HELPERS ################################
public static final IfStatement sample1() {
    Block ifBlock = TestBlock.sample1();
    Block elseBlock = TestBlock.sample2();
    Criteria criteria = TestSetCriteria.sample1();
    return new IfStatement(criteria, ifBlock, elseBlock);
}
Also used : IfStatement(org.teiid.query.sql.proc.IfStatement) Block(org.teiid.query.sql.proc.Block)

Aggregations

Block (org.teiid.query.sql.proc.Block)9 Test (org.junit.Test)6 AssignmentStatement (org.teiid.query.sql.proc.AssignmentStatement)5 CommandStatement (org.teiid.query.sql.proc.CommandStatement)4 CreateProcedureCommand (org.teiid.query.sql.proc.CreateProcedureCommand)4 RaiseStatement (org.teiid.query.sql.proc.RaiseStatement)4 IfStatement (org.teiid.query.sql.proc.IfStatement)2 LoopStatement (org.teiid.query.sql.proc.LoopStatement)2 LanguageObject (org.teiid.query.sql.LanguageObject)1 Labeled (org.teiid.query.sql.proc.Statement.Labeled)1 TriggerAction (org.teiid.query.sql.proc.TriggerAction)1 WhileStatement (org.teiid.query.sql.proc.WhileStatement)1 ElementSymbol (org.teiid.query.sql.symbol.ElementSymbol)1