DepthTest.java

package org.softlang.company.tests;

import static org.junit.Assert.assertEquals;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.softlang.company.features.Depth;

public class DepthTest {

	private static Connection conn;
	public static final String NAME = "Acme Corporation";

	@Before
	public void setUp() throws ClassNotFoundException, SQLException {
		Class.forName("org.h2.Driver");
		System.out.println(System.getProperty("user.dir"));
		conn = DriverManager
				.getConnection(
						"jdbc:h2:mem:company;INIT=runscript from 'src/test/resources/Company.sql'\\;runscript from 'src/test/resources/sampleCompany.sql'",
						"sa", "");
	}

	@After
	public void tearDown() throws SQLException {
		conn.close();
	}

	@Test
	public void testDepth() {
		assertEquals(3, Depth.depth(conn, NAME));
	}

}