정적 메서드를 통해 액세스 할 수 있음에도 불구하고 DataSource Dependency Injection을 모의 사용하는 방법

Question

내 데이터베이스 코드를 단위 테스트하는 데 Mockito, DBUnit 및 을 사용하고 있습니다. 물론 통합 테스트도 쓰고 있습니다.

테스트중인 시스템 (테스트중인 클래스)에 조롱 된 DataSource을 삽입하는 방법을 이해하는 데 문제가 있습니다. DataSource은 연결 풀링에 사용되므로 다른 클래스는이 클래스 DataSource의 인스턴스를 검색하기 위해 같은 클래스에서 static 메서드를 호출 할 수 있습니다. 즉, DataSource은 어디서나 생성자에 주입되지 않으므로 내 테스트에는 조롱 된 DataSource을 삽입 할 생성자가 없습니다.

내 실제 코드의 논리를 변경하여 개인 변수가 null인지 확인하고, 그렇다면 주입 된 데이터 소스 (테스트에만 필요하기 때문에 잘못된 디자인)를 사용합니다. 정적 메서드는 연결 풀의 소스를 검색 (더 나은 디자인).

DataSource을 생성자가 받아 들일 수있는 클래스가 아닌 클래스에 삽입하려면 대신 정적 메서드를 호출하여 종속성을 검색하면되므로 어떻게해야합니까?

클래스

public DBConnection(DBSource dbSource) { // <--- Constructor only required for test purposes :(
     this.dbSource = dbSource; 
    } 

    public final void createCompsDB() { 
     Connection conn = null; 
     Statement statement = null; 
     try { 
      if(dbSource==null){ 
       conn = DBSource.getInstance().getConnection(); 
      }else{ 
       conn = dbSource.getConnection(); /** Likely bad design, since dbSource is only NOT null for tests, so that you can inject the mocked datasource :( */ 
      } 
      statement = conn.createStatement(); 
      statement.executeUpdate("CREATE DATABASE placesdb"); 
      System.out.println("Database created..."); 
     } catch (SQLException e) { 
       // ... 
      } 
     } finally { 
      // Close Resources... 
     } 
    } 
} 

테스트 클래스를 테스트하는 - 테스트 DBSource.java

protected DBSource() throws IOException, SQLException, PropertyVetoException { 
     ds = new BasicDataSource(); 
     ds.setDriverClassName("org.postgresql.Driver"); 
     ds.setUsername("user"); 
     ds.setPassword("pass"); 
     ds.setUrl("jdbc:postgresql://localhost:5432/placesdb"); 
    } 

    public static DBSource getInstance() { // <--- Static method means dependent classes don't need to accept injections 
     if (datasource == null) { 
      datasource = new DBSource(); 
      return datasource; 
     } else { 
      return datasource; 
     } 
    } 

    public Connection getConnection() throws SQLException { 
     return this.ds.getConnection(); 
    } 
} 

Answer 1

M

public class DBConnectionTest { 
     final Statement statement = mock(Statement.class); 
     final Connection connection = mock(Connection.class); 
     final DBSource dataSource = mock(DBSource.class); 

    @Before 
    public void setUp() throws SQLException, IOException, PropertyVetoException { 
     when(dataSource.getConnection()).thenReturn(connection); 
     when(connection.createStatement()).thenReturn(statement); 
    } 

    @Test 
    public void testCreateCompDBIfNotAlready() throws Exception { 
     DBConnection dbConnection = new DBConnection(localDB, dataSource); /** This constructor is only needed for testing :(. How do I avoid it since all the classes I need to test don't require the dependency to be injected? */ 
     dbConnection.createCompsDB();  
     verify(statement).executeUpdate("CREATE DATABASE PLACES"); 
    } 
} 

전달 PowerMockito를 사용하여 정적 클래스 메서드를 수행 할 수 있습니다. 테스트 클래스는 같은 것을해야한다 :

@RunWith(PowerMockRunner.class) 
@PrepareForTest(DBSource.class) 
public class DBConnectionTest { 
    @Mock 
    final Statement statement; 
    @Mock 
    final Connection connection; 
    @Mock 
    final DBSource dbsource; 

    @Before 
    public void setUp() throws SQLException, IOException, PropertyVetoException { 
     PowerMockito.mockStatic(DBSource.class); 
     when(DbSource.getInstance()).thenReturn(dbsource); 
     when(dbsource.getConnection()).thenReturn(connection); 
     when(connection.createStatement()).thenReturn(statement); 
    } 

    @Test 
    public void testCreateCompDBIfNotAlready() throws Exception { 
     DBConnection dbConnection = new DBConnection(localDB); // No test-only constructor anymore 
     dbConnection.createCompsDB();  
     verify(statement).executeUpdate("CREATE DATABASE PLACES"); 
    } 
} 

당신은 PowerMock과 조롱에 대한 here 자세한 내용을보실 수 있습니다.