WarningTest.java

/*
 * Microsoft JDBC Driver for SQL Server
 * 
 * Copyright(c) 2016 Microsoft Corporation All rights reserved.
 * 
 * This program is made available under the terms of the MIT License. See the LICENSE file in the project root for more information.
 */
package com.microsoft.sqlserver.jdbc.connection;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.util.Properties;

import org.junit.jupiter.api.Test;
import org.junit.platform.runner.JUnitPlatform;
import org.junit.runner.RunWith;

import com.microsoft.sqlserver.jdbc.SQLServerConnection;
import com.microsoft.sqlserver.testframework.AbstractTest;

@RunWith(JUnitPlatform.class)
public class WarningTest extends AbstractTest {
	@Test
	public void testWarnings() throws SQLException {
		SQLServerConnection conn = (SQLServerConnection) DriverManager.getConnection(connectionString);

		Properties info = conn.getClientInfo();
		conn.setClientInfo(info);
		SQLWarning warn = conn.getWarnings();
		assertEquals(null, warn, "Warnings found.");

		Properties info2 = new Properties();
		String[] infoArray = { "prp1", "prp2", "prp3", "prp4", "prp5" };
		for (int i = 0; i < 5; i++) {
			info2.put(infoArray[i], "");
		}
		conn.setClientInfo(info2);
		warn = conn.getWarnings();
		for (int i = 4; i >= 0; i--) {
			assertTrue(warn.toString().contains(infoArray[i]), "Warnings not found!");
			warn = warn.getNextWarning();
		}
		conn.clearWarnings();

		conn.setClientInfo("prop7", "");
		warn = conn.getWarnings();
		assertTrue(warn.toString().contains("prop7"), "Warnings not found!");
		warn = warn.getNextWarning();
		assertEquals(null, warn, "Warnings found!");
	}
}