@SuppressWarnings("unchecked")
  public void testVisitJoinWithoutLock() throws Exception {
    Select select = new Select("test");
    select.setLocked(false);
    Visitor queryVis = getVisitor();
    Table tab1 = new Table("tab1");
    Column col1 = tab1.column("col1");
    Table tab2 = new Table("tab2");
    Column col2 = tab2.column("col2");
    Join join = new Join(JoinOperator.LEFT, new Table("tab1"), col1.equal(col2));

    Class[] clazz = {Select.class};
    Object[] obj = {select};
    try {
      Method method = queryVis.getClass().getDeclaredMethod("setSelect", clazz);
      method.setAccessible(true);
      method.invoke(queryVis, obj);
    } catch (Exception e) {
      fail("Something went wrong using reflection to execute private method.");
      e.printStackTrace();
    }

    queryVis.visit(join);

    assertEquals("LEFT JOIN \"tab1\" ON \"tab1\".\"col1\"=\"tab2\".\"col2\"", queryVis.toString());
  }
  @Test
  public void testSelectWithFromWithLock() {
    Visitor queryVis = getVisitor();

    Table tab1 = new Table("FOO");
    Table tab2 = new Table("BAR");
    TableAlias tab3 = new TableAlias(new Table("ABC"), "xyz");

    Select sel = new Select(tab1);
    sel.setLocked(true);
    sel.addFrom(tab2);
    sel.addFrom(tab3);

    queryVis.visit(sel);

    assertTrue(sel.hasFrom());

    Iterator<Qualifier> iter = sel.getFrom().iterator();
    assertEquals(iter.next(), tab1);
    assertEquals(iter.next(), tab2);
    assertEquals(iter.next(), tab3);
    assertFalse(iter.hasNext());

    assertEquals(
        "SELECT * " + "FROM \"FOO\" HOLDLOCK, \"BAR\" HOLDLOCK, " + "\"ABC\" \"xyz\" HOLDLOCK",
        queryVis.toString());
  }
Пример #3
0
  @Test
  public void testHandleLock() throws Exception {
    Select select = new Select("Test");
    select.setLocked(true);

    Visitor queryVis = getVisitor();
    ((MySQLQueryVisitor) queryVis).handleLock(select);

    assertEquals(" FOR UPDATE", queryVis.toString());
  }
Пример #4
0
  @Test
  public void testSelectNoConditionNoExpressionWithLock() throws Exception {
    Select select = new Select("TestTable");
    select.setLocked(true);

    Visitor queryVis = getVisitor();
    queryVis.visit(select);

    String expected = "SELECT * FROM TestTable FOR UPDATE";

    assertEquals(expected, queryVis.toString());
  }
  @Test
  public void testSelectWithJoinsDepthAndBreadthWithLock() {
    Visitor queryVis = getVisitor();

    Table tab1 = new Table("table1");
    Column col1 = new Column(tab1, "col1");

    Table tab2 = new Table("table2");
    Column col2 = new Column(tab2, "col2");

    Table tab3 = new Table("table3");
    Column col3 = new Column(tab3, "col3");

    Table tab4 = new Table("table4");
    Column col4 = new Column(tab4, "col4");

    Table tab5 = new Table("table5");
    Column col5 = new Column(tab5, "col5");

    Table tab6 = new Table("table6");
    Column col6 = new Column(tab6, "col6");

    Table tab7 = new Table("table7");
    Column col7 = new Column(tab7, "col7");

    Select select = new Select(tab1);
    select.setLocked(true);

    tab1.addInnerJoin(tab2, col1.equal(col2));
    tab2.addFullJoin(tab3, col2.equal(col3));
    tab2.addLeftJoin(tab7, col2.equal(col7));

    tab1.addInnerJoin(tab4, col1.equal(col4));
    tab4.addRightJoin(tab5, col4.equal(col5));
    tab5.addLeftJoin(tab6, col5.equal(col6));

    queryVis.visit(select);

    assertEquals(
        "SELECT * FROM "
            + "((\"table1\" HOLDLOCK INNER JOIN "
            + "((\"table2\" HOLDLOCK FULL JOIN \"table3\" HOLDLOCK ON "
            + "\"table2\".\"col2\"=\"table3\".\"col3\") "
            + "LEFT JOIN \"table7\" HOLDLOCK ON \"table2\".\"col2\"=\"table7\".\"col7\") "
            + "ON \"table1\".\"col1\"=\"table2\".\"col2\") "
            + "INNER JOIN (\"table4\" HOLDLOCK "
            + "RIGHT JOIN (\"table5\" HOLDLOCK LEFT JOIN \"table6\" HOLDLOCK "
            + "ON \"table5\".\"col5\"=\"table6\".\"col6\") "
            + "ON \"table4\".\"col4\"=\"table5\".\"col5\") "
            + "ON \"table1\".\"col1\"=\"table4\".\"col4\")",
        queryVis.toString());
  }
  @Test
  public void testSelectWithFromAndJoinsWithLock() {
    Visitor queryVis = getVisitor();

    Table tab1 = new Table("FOO");
    Column col1 = new Column(tab1, "ID");

    Table tab2 = new Table("BAR");
    Column col21 = new Column(tab2, "ID");
    Column col22 = new Column(tab2, "ABC_ID");

    TableAlias tab3 = new TableAlias(new Table("ABC"), "xyz");
    Column col3 = new Column(tab3, "ID");

    Table tab4 = new Table("FN");
    Table tab5 = new Table("ORD");
    TableAlias tab6 = new TableAlias(new Table("FN"), "abc");

    Select sel = new Select(tab1);
    sel.setLocked(true);
    tab1.addInnerJoin(tab2, col1.equal(col21));
    tab2.addLeftJoin(tab3, col22.equal(col3));

    sel.addFrom(tab4);
    sel.addFrom(tab5);
    tab5.addRightJoin(tab6);

    queryVis.visit(sel);

    assertTrue(sel.hasFrom());

    Iterator<Qualifier> iter = sel.getFrom().iterator();
    assertEquals(iter.next(), tab1);
    assertEquals(iter.next(), tab4);
    assertEquals(iter.next(), tab5);
    assertFalse(iter.hasNext());

    assertEquals(
        "SELECT * "
            + "FROM (\"FOO\" HOLDLOCK "
            + "INNER JOIN (\"BAR\" HOLDLOCK "
            + "LEFT JOIN \"ABC\" \"xyz\" HOLDLOCK "
            + "ON \"BAR\".\"ABC_ID\"=\"xyz\".\"ID\") "
            + "ON \"FOO\".\"ID\"=\"BAR\".\"ID\")"
            + ", \"FN\" HOLDLOCK, (\"ORD\" HOLDLOCK RIGHT JOIN \"FN\" \"abc\" "
            + "HOLDLOCK)",
        queryVis.toString());
  }
  @SuppressWarnings("unchecked")
  public void testHandleJoinConstructionWithLock() {
    Select select = new Select("test");
    select.setLocked(true);
    Visitor queryVis = getVisitor();
    Table table = new Table("tab1");
    Column col1 = table.column("col1");
    Table table2 = new Table("tab2");
    Column col2 = table2.column("col2");

    Class[] clazz = {Select.class};
    Object[] obj = {select};
    try {
      Method method = queryVis.getClass().getDeclaredMethod("setSelect", clazz);
      method.setAccessible(true);
      method.invoke(queryVis, obj);
    } catch (Exception e) {
      fail("Something went wrong using reflection to execute private method.");
      e.printStackTrace();
    }

    ((SybaseQueryVisitor) queryVis).handleJoinConstruction(table);

    assertEquals("\"tab1\" HOLDLOCK", queryVis.toString());

    table.addFullJoin(table2, col1.equal(col2));
    queryVis = getVisitor();

    try {
      Method method = queryVis.getClass().getDeclaredMethod("setSelect", clazz);
      method.setAccessible(true);
      method.invoke(queryVis, obj);
    } catch (Exception e) {
      fail("Something went wrong using reflection to execute private method.");
      e.printStackTrace();
    }

    ((SybaseQueryVisitor) queryVis).handleJoinConstruction(table);

    assertEquals(
        "(\"tab1\" HOLDLOCK FULL JOIN \"tab2\" HOLDLOCK ON "
            + "\"tab1\".\"col1\"=\"tab2\".\"col2\")",
        queryVis.toString());
  }
  @SuppressWarnings("unchecked")
  public void testAddTableNamesWithoutLock() {
    Select select = new Select("test");
    select.setLocked(false);
    Visitor queryVis = getVisitor();
    Class[] clazz = {Select.class};
    Object[] obj = {select};
    try {
      Method method = queryVis.getClass().getDeclaredMethod("setSelect", clazz);
      method.setAccessible(true);
      method.invoke(queryVis, obj);
    } catch (Exception e) {
      fail("Something went wrong using reflection to execute private method.");
      e.printStackTrace();
    }

    Table table = new Table("TestTable");
    ((SybaseQueryVisitor) queryVis).addTableNames(table);

    assertEquals("\"TestTable\"", queryVis.toString());

    queryVis = getVisitor();

    try {
      Method method = queryVis.getClass().getDeclaredMethod("setSelect", clazz);
      method.setAccessible(true);
      method.invoke(queryVis, obj);
    } catch (Exception e) {
      fail("Something went wrong using reflection to execute private method.");
      e.printStackTrace();
    }

    TableAlias tblAls = new TableAlias(table, "TestTableAlias");
    ((SybaseQueryVisitor) queryVis).addTableNames(tblAls);

    assertEquals("\"TestTable\" \"TestTableAlias\"", queryVis.toString());
  }
  @Test
  public void testSelectWithJoinsBreadthWithLock() {
    Visitor queryVis = getVisitor();

    Table tab1 = new Table("FOO");
    Column col1 = new Column(tab1, "ID");

    Table tab2 = new Table("BAR");
    Column col21 = new Column(tab2, "ID");

    TableAlias tab3 = new TableAlias(new Table("ABC"), "xyz");
    Column col3 = new Column(tab3, "ID");

    Table tab4 = new Table("tab4");

    Table tab5 = new Table("tab5");

    Select sel = new Select(tab1);
    sel.setLocked(true);
    queryVis.visit(sel);
    assertEquals("SELECT * FROM \"FOO\" HOLDLOCK", queryVis.toString());

    queryVis = getVisitor();
    tab1.addInnerJoin(tab2, col1.equal(col21));
    tab1.addInnerJoin(tab3, col1.equal(col3));
    tab1.addInnerJoin(tab4);
    tab1.addInnerJoin(tab5);
    queryVis.visit(sel);
    assertEquals(
        "SELECT * FROM "
            + "((((\"FOO\" HOLDLOCK INNER JOIN \"BAR\" HOLDLOCK ON "
            + "\"FOO\".\"ID\"=\"BAR\".\"ID\") "
            + "INNER JOIN \"ABC\" \"xyz\" HOLDLOCK ON \"FOO\".\"ID\"=\"xyz\".\"ID\") "
            + "INNER JOIN \"tab4\" HOLDLOCK) "
            + "INNER JOIN \"tab5\" HOLDLOCK)",
        queryVis.toString());
  }
  @Test
  public void testSelectNoConditionNoExpressionWithLock() throws Exception {
    Select select = new Select("TestTable");
    select.setLocked(true);

    Visitor queryVis = getVisitor();
    queryVis.visit(select);

    String expected = "SELECT * FROM \"TestTable\" HOLDLOCK";

    assertEquals(expected, queryVis.toString());

    Object result = null;
    try {
      Method method = queryVis.getClass().getDeclaredMethod("getSelect", (Class[]) null);
      method.setAccessible(true);
      result = method.invoke(queryVis, (Object[]) null);
    } catch (Exception e) {
      fail("Something went wrong using reflection to execute private method.");
      e.printStackTrace();
    }

    assertEquals(select, result);
  }
  @Test
  public void testSelectWithJoinsDepthWithLock() {
    Visitor queryVis = getVisitor();

    Table tab1 = new Table("FOO");
    Column col1 = new Column(tab1, "ID");

    Table tab2 = new Table("BAR");
    Column col21 = new Column(tab2, "ID");
    Column col22 = new Column(tab2, "ABC_ID");

    TableAlias tab3 = new TableAlias(new Table("ABC"), "xyz");
    Column col3 = new Column(tab3, "ID");

    Select sel = new Select(tab1);
    sel.setLocked(true);
    queryVis.visit(sel);
    assertEquals("SELECT * FROM \"FOO\" HOLDLOCK", queryVis.toString());

    queryVis = getVisitor();
    tab1.addInnerJoin(tab2, col1.equal(col21));
    queryVis.visit(sel);
    assertEquals(
        "SELECT * "
            + "FROM (\"FOO\" HOLDLOCK "
            + "INNER JOIN \"BAR\" HOLDLOCK "
            + "ON \"FOO\".\"ID\"=\"BAR\".\"ID\")",
        queryVis.toString());

    queryVis = getVisitor();
    tab2.addLeftJoin(tab3, col22.equal(col3));
    queryVis.visit(sel);
    assertEquals(
        "SELECT * "
            + "FROM (\"FOO\" HOLDLOCK "
            + "INNER JOIN (\"BAR\" HOLDLOCK "
            + "LEFT JOIN \"ABC\" \"xyz\" HOLDLOCK "
            + "ON \"BAR\".\"ABC_ID\"=\"xyz\".\"ID\") "
            + "ON \"FOO\".\"ID\"=\"BAR\".\"ID\")",
        queryVis.toString());
  }