-
Notifications
You must be signed in to change notification settings - Fork 428
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
This reverts commit 67f7fab.
- Loading branch information
Showing
3 changed files
with
138 additions
and
117 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
123 changes: 123 additions & 0 deletions
123
src/test/java/com/microsoft/sqlserver/jdbc/bulkCopy/BulkCopyMoneyTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,123 @@ | ||
package com.microsoft.sqlserver.jdbc.bulkCopy; | ||
|
||
import static org.junit.Assert.fail; | ||
import static org.junit.jupiter.api.Assertions.assertTrue; | ||
|
||
import java.io.ByteArrayInputStream; | ||
import java.io.InputStream; | ||
import java.nio.charset.StandardCharsets; | ||
import java.sql.Connection; | ||
import java.sql.ResultSet; | ||
import java.sql.SQLException; | ||
import java.sql.Statement; | ||
import java.util.HashMap; | ||
import java.util.Map; | ||
|
||
import org.junit.jupiter.api.AfterAll; | ||
import org.junit.jupiter.api.BeforeAll; | ||
import org.junit.jupiter.api.Test; | ||
import org.junit.platform.runner.JUnitPlatform; | ||
import org.junit.runner.RunWith; | ||
|
||
import com.microsoft.sqlserver.jdbc.RandomUtil; | ||
import com.microsoft.sqlserver.jdbc.SQLServerBulkCSVFileRecord; | ||
import com.microsoft.sqlserver.jdbc.SQLServerBulkCopy; | ||
import com.microsoft.sqlserver.jdbc.TestResource; | ||
import com.microsoft.sqlserver.jdbc.TestUtils; | ||
import com.microsoft.sqlserver.testframework.AbstractSQLGenerator; | ||
import com.microsoft.sqlserver.testframework.AbstractTest; | ||
import com.microsoft.sqlserver.testframework.Constants; | ||
import com.microsoft.sqlserver.testframework.PrepUtil; | ||
|
||
|
||
/** | ||
* Tests money/smallmoney limits with BulkCopy | ||
*/ | ||
@RunWith(JUnitPlatform.class) | ||
public class BulkCopyMoneyTest extends AbstractTest { | ||
static String encoding = Constants.UTF8; | ||
static String delimiter = Constants.COMMA; | ||
static String destTableName = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("moneyBulkCopyDest")); | ||
static String destTableName2 = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("moneyBulkCopyDest")); | ||
|
||
@Test | ||
/** | ||
* Tests money and smallmoney with bulkcopy using minimum and maximum values of each | ||
*/ | ||
public void testMoneyWithBulkCopy() throws Exception { | ||
try (Connection conn = PrepUtil.getConnection(connectionString)) { | ||
testMoneyLimits(-214799.3648, 922337203685387.5887, conn); // SMALLMONEY MIN | ||
testMoneyLimits(214799.3698, 922337203685387.5887, conn); // SMALLMONEY MAX | ||
testMoneyLimits(214719.3698, -922337203685497.5808, conn); // MONEY MIN | ||
testMoneyLimits(214719.3698, 922337203685478.5807, conn); // MONEY MAX | ||
} | ||
} | ||
|
||
private void testMoneyLimits(double smallMoneyVal, double moneyVal, Connection conn) throws Exception { | ||
SQLServerBulkCSVFileRecord fileRecord = constructFileRecord(smallMoneyVal, moneyVal); | ||
|
||
try { | ||
testMoneyWithBulkCopy(conn, fileRecord); | ||
fail(TestResource.getResource("R_expectedExceptionNotThrown")); | ||
} catch (SQLException e) { | ||
assertTrue(e.getMessage().matches(TestUtils.formatErrorMsg("R_valueOutOfRange")), e.getMessage()); | ||
} | ||
} | ||
|
||
private SQLServerBulkCSVFileRecord constructFileRecord(double smallMoneyVal, double moneyVal) throws Exception { | ||
Map<Object, Object> data = new HashMap(); | ||
data.put(smallMoneyVal, moneyVal); | ||
|
||
StringBuilder stringBuilder = new StringBuilder(); | ||
stringBuilder.append("smallmoneycol, moneycol\n"); | ||
|
||
for (Map.Entry entry : data.entrySet()) { | ||
stringBuilder.append(String.format("%s,%s\n", entry.getKey(), entry.getValue())); | ||
} | ||
|
||
byte[] bytes = stringBuilder.toString().getBytes(StandardCharsets.UTF_8); | ||
SQLServerBulkCSVFileRecord fileRecord = null; | ||
try (InputStream inputStream = new ByteArrayInputStream(bytes)) { | ||
fileRecord = new SQLServerBulkCSVFileRecord(inputStream, encoding, delimiter, true); | ||
} | ||
return fileRecord; | ||
} | ||
|
||
private void testMoneyWithBulkCopy(Connection conn, SQLServerBulkCSVFileRecord fileRecord) throws SQLException { | ||
try (SQLServerBulkCopy bulkCopy = new SQLServerBulkCopy(conn); Statement stmt = conn.createStatement()) { | ||
|
||
fileRecord.addColumnMetadata(1, "c1", java.sql.Types.DECIMAL, 10, 4); // with smallmoney | ||
fileRecord.addColumnMetadata(2, "c2", java.sql.Types.DECIMAL, 19, 4); // with money | ||
|
||
bulkCopy.setDestinationTableName(destTableName); | ||
bulkCopy.writeToServer(fileRecord); | ||
|
||
try (ResultSet rs = stmt.executeQuery("select * FROM " + destTableName + " order by c1"); | ||
SQLServerBulkCopy bcOperation = new SQLServerBulkCopy(conn);) { | ||
bcOperation.setDestinationTableName(destTableName2); | ||
bcOperation.writeToServer(rs); | ||
} | ||
} | ||
} | ||
|
||
@BeforeAll | ||
public static void setupTests() throws SQLException { | ||
try (Connection con = getConnection(); Statement stmt = con.createStatement()) { | ||
TestUtils.dropTableIfExists(destTableName, stmt); | ||
TestUtils.dropTableIfExists(destTableName2, stmt); | ||
|
||
String table = "create table " + destTableName + " (c1 smallmoney, c2 money)"; | ||
stmt.execute(table); | ||
table = "create table " + destTableName2 + " (c1 smallmoney, c2 money)"; | ||
stmt.execute(table); | ||
} | ||
} | ||
|
||
@AfterAll | ||
public static void cleanUp() throws Exception { | ||
try (Connection con = getConnection(); Statement stmt = con.createStatement()) { | ||
TestUtils.dropTableIfExists(destTableName, stmt); | ||
TestUtils.dropTableIfExists(destTableName2, stmt); | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters