Example #1
0
 @BeforeClass
 public static void setUpBeforeClass() throws Exception {
   tgds = new TGroupDataSource();
   tgds.setDbGroupKey("dbKey0");
   List<DataSourceWrapper> dataSourceWrappers = new ArrayList<DataSourceWrapper>();
   DataSourceWrapper dsw1 = new DataSourceWrapper("db1", "rw", db1, DBType.MYSQL);
   DataSourceWrapper dsw2 = new DataSourceWrapper("db2", "r", db2, DBType.MYSQL);
   dataSourceWrappers.add(dsw1);
   dataSourceWrappers.add(dsw2);
   tgds.init(dataSourceWrappers);
 }
Example #2
0
 @Test
 public void testExecuteUpdate() throws SQLException {
   TGroupConnection conn = null;
   PreparedStatement stat = null;
   try {
     conn = tgds.getConnection();
     stat = conn.prepareStatement("update xxxx set name = 'newname'");
     int affect = stat.executeUpdate();
     Assert.assertEquals(affect, 1);
     MockDataSource.showTrace();
   } finally {
     if (conn != null) {
       try {
         conn.close();
       } catch (SQLException e) {
       }
       if (stat != null) {
         try {
           stat.close();
         } catch (SQLException e) {
         }
       }
     }
   }
 }
Example #3
0
 @Test
 public void testExecuteQuery() throws SQLException {
   TGroupConnection conn = null;
   PreparedStatement stat = null;
   try {
     conn = tgds.getConnection();
     stat = conn.prepareStatement("select * from xxx where id=?");
     stat.setByte(1, (byte) 5);
     MockDataSource.addPreData("id:1,name:2");
     ResultSet result = stat.executeQuery();
     Assert.assertEquals(result.next(), true);
     Assert.assertEquals(result.getLong(1), 1L);
     Assert.assertEquals(result.getLong(2), 2L);
     MockDataSource.showTrace();
   } finally {
     if (conn != null) {
       try {
         conn.close();
       } catch (SQLException e) {
       }
       if (stat != null) {
         try {
           stat.close();
         } catch (SQLException e) {
         }
       }
     }
   }
 }
Example #4
0
  /** 同一个PreparedStatement先查询后更新 */
  @Test
  public void testExecute2() throws SQLException {
    TGroupConnection conn = null;
    PreparedStatement stat = null;
    try {
      conn = tgds.getConnection();
      stat = conn.prepareStatement("select * from xxx where id=?");
      stat.setByte(1, (byte) 5);
      boolean res = stat.execute();
      Assert.assertEquals(res, true);

      res = stat.execute("update t set name = 'newName'");
      Assert.assertEquals(res, false);
      MockDataSource.showTrace();
    } finally {
      if (conn != null) {
        try {
          conn.close();
        } catch (SQLException e) {
        }
        if (stat != null) {
          try {
            stat.close();
          } catch (SQLException e) {
          }
        }
      }
    }
  }
Example #5
0
 @Test
 public void testAddBatchSql() throws SQLException {
   TGroupConnection conn = null;
   Statement stat = null;
   try {
     conn = tgds.getConnection();
     stat = conn.createStatement();
     stat.addBatch("update t set name = 'newName' ");
     stat.addBatch("update t set type = 2 ");
     int[] affectedRow = stat.executeBatch();
     System.out.println(Arrays.toString(affectedRow));
     MockDataSource.showTrace();
     Assert.assertTrue(MockDataSource.hasMethod("db", "db1", "executeBatch"));
   } finally {
     if (conn != null) {
       try {
         conn.close();
       } catch (SQLException e) {
       }
       if (stat != null) {
         try {
           stat.close();
         } catch (SQLException e) {
         }
       }
     }
   }
 }
Example #6
0
  @Test
  public void testAddBatch() throws SQLException {
    TGroupConnection conn = null;
    PreparedStatement stat = null;
    try {
      conn = tgds.getConnection();
      stat = conn.prepareStatement("update test set type=? where id = ?");

      stat.setInt(1, 1);
      stat.setString(2, "2askjfoue33");
      stat.addBatch();

      stat.setInt(1, 2);
      stat.setString(2, "retrtorut48");
      stat.addBatch();

      int[] affectedRow = stat.executeBatch();
      System.out.println(Arrays.toString(affectedRow));
      MockDataSource.showTrace();
      Assert.assertTrue(MockDataSource.hasMethod("db", "db1", "executeBatch"));
    } finally {
      if (conn != null) {
        try {
          conn.close();
        } catch (SQLException e) {
        }
        if (stat != null) {
          try {
            stat.close();
          } catch (SQLException e) {
          }
        }
      }
    }
  }
  protected static void setMatrixMockInfo(String groupsPath, String key, boolean isOracle)
      throws Exception {
    // -----------------获取groups信息
    String groupsStr = LoadPropsUtil.loadProps2OneLine(groupsPath, key);
    if (groupsStr == null || StringUtils.isBlank(groupsStr)) {
      throw new Exception("指定path = " + groupsPath + ",key = " + key + "的groups信息为null或者为空字符。");
    }

    // -----------------oracle or mysql
    String groupPath = BaseAtomGroupTestCase.GROUP_PATH;
    String atomPath = BaseAtomGroupTestCase.ATOM_PATH;
    if (isOracle) {
      groupPath = BaseAtomGroupTestCase.GROUP_ORA_PATH;
      atomPath = BaseAtomGroupTestCase.ATOM_ORA_PATH;
    }

    // -----------------获取group信息
    BaseAtomGroupTestCase.dataMap = new HashMap<String, String>();
    String[] groupArr = groupsStr.split(",");

    for (String group : groupArr) {
      group = group.trim();
      String groupStr = "";
      groupStr =
          LoadPropsUtil.loadProps2OneLine(
              groupPath + group + BaseAtomGroupTestCase.PROPERTIES_FILE, group);
      if (groupsStr != null && StringUtils.isNotBlank(groupsStr)) {
        // 获取atom信息
        String[] atomArr = groupStr.split(",");
        for (String atom : atomArr) {
          atom = atom.trim();
          atom = atom.substring(0, atom.indexOf(":"));
          BaseAtomGroupTestCase.initAtomConfig(
              atomPath + atom, BaseAtomGroupTestCase.APPNAME, atom);
        }
        // 获取groupkey
        BaseAtomGroupTestCase.dataMap.put(TGroupDataSource.getFullDbGroupKey(group), groupStr);
      }
    }

    // -----------------dbgroups
    BaseAtomGroupTestCase.dataMap.put(
        new MessageFormat("com.taobao.tddl.v1_{0}_dbgroups")
            .format(new Object[] {BaseAtomGroupTestCase.APPNAME}),
        groupsStr);

    // 建立MockServer
    MockServer.setConfigInfos(BaseAtomGroupTestCase.dataMap);
  }
Example #8
0
  @Test
  public void java_sql_Statement_api_support() throws Exception {
    TGroupConnection conn = null;
    Statement stmt = null;
    try {
      conn = tgds.getConnection();
      stmt = conn.createStatement();

      String insertSQL = "insert into crud(f1,f2) values(10,'str')";
      String updateSQL = "update crud set f2='str2'";
      String selectSQL = "select * from crud";
      String showSQL = "show create table crud";

      assertFalse(stmt.execute(insertSQL));
      assertTrue(stmt.execute(selectSQL));
      assertTrue(stmt.execute(showSQL));
      assertFalse(stmt.execute(insertSQL, Statement.RETURN_GENERATED_KEYS));
      assertFalse(stmt.execute(insertSQL, new int[] {1}));
      assertFalse(stmt.execute(insertSQL, new String[] {"col"}));

      stmt.addBatch(insertSQL);
      stmt.addBatch(updateSQL);

      int[] updateCounts = stmt.executeBatch();
      assertEquals(updateCounts.length, 2);
      MockDataSource.addPreData("id:1,name:2");
      assertTrue(stmt.executeQuery(selectSQL).next());
      assertEquals(stmt.executeUpdate(insertSQL), 1);
      assertEquals(stmt.executeUpdate(insertSQL, Statement.RETURN_GENERATED_KEYS), 1);
      assertEquals(stmt.executeUpdate(insertSQL, new int[] {1}), 1);
      assertEquals(stmt.executeUpdate(insertSQL, new String[] {"col"}), 1);
    } finally {
      if (conn != null) {
        try {
          conn.close();
        } catch (SQLException e) {
        }

        if (stmt != null) {
          try {
            stmt.close();
          } catch (SQLException e) {
          }
        }
      }
    }
  }