Question

我有这个代码连接数据库，这是由Strutsapplication定义的。

public void initConnection()
   {
     if (this.con == null) {
       try
       {
         Context ctx = new InitialContext();
         Context envContext = (Context)ctx.lookup("java:comp/env");
         this.ds = ((DataSource)envContext.lookup("jdbc/FooBar"));
         if (this.ds != null)
         {
           this.con = this.ds.getConnection();
           if (this.con != null) {
             LOGGER.info("datasbase connection established");
           } else {
             LOGGER.error("there was an error during connectiong to the database");
           }
         }
       }
       catch (Exception e)
       {
         LOGGER.error(e);
       }
     }
   }

现在，使用context.xml定义Context！

<Context>
    <Resource name="jdbc/FooBar" auth="Container" type="javax.sql.DataSource"
        maxActive="20" maxIdle="30" maxWait="10000" username="root" password=""
        driverClassName="com.mysql.jdbc.Driver"
        url="jdbc:mysql://localhost:3306/ProductiveDB?autoReconnect=true" />
</Context>

有没有办法使用Java模拟这个上下文？我正在编写单元测试，我需要连接到名为TestDB的数据库。

编辑：我正在使用mockito创建模拟我的类。我只想使用Java创建类似于XML的东西，例如：

Resource resource = new Resource();
resource.setAuth("Container");
bla bla bla
resource.setURL("jdbc:mysql://localhost:3306/TestDB?aztoReconnect=true");

然后使用

将其添加到我的上下文中

new InitialContext().addToContext("jdbc/FooBar",resource);

编辑II： 我在我的setup函数中编辑了我的代码，如下所示：

ds.setDriverClassName("com.mysql.jdbc.Driver");
ds.setUrl("jdbc:mysql://localhost:3306/FooBar");
ds.setUsername("root");
ds.setPassword("");
Context ctx = new InitialContext();
System.out.println(ctx);
System.out.println(ctx.getEnvironment());
ctx.addToEnvironment("java:comp/env/jdbc/FooBar", ds);

我认为，这样，我更接近我的解决方案，但现在我收到以下错误：

javax.naming.NoInitialContextException: Need to specify class name in environment or system property, or as an applet parameter, or in an application resource file:  java.naming.factory.initial
at javax.naming.spi.NamingManager.getInitialContext(Unknown Source)
at javax.naming.InitialContext.getDefaultInitCtx(Unknown Source)
at javax.naming.InitialContext.getEnvironment(Unknown Source)
at com.foo.bar.su.tests.StepDefs.prepareTests(StepDefs.java:55)
at com.foo.bar.su.tests.HomeTest.setUp(HomeTest.java:22)

我能弄清楚，这个错误显然意味着，我的环境。无法找到。但我怎样才能创造＆＃34;它，还是什么？

Answer 1

您应该避免以编程方式查找，因为它会使您的代码更难以测试。使用框架来控制依赖注入会更好。然后你可以使用像DBUnit这样的东西。

如果无法使用依赖注入，您仍然可以使代码更易于测试。例如，您可以将数据源查找隔离如下：

public abstract class MyAbstractDao {
    private DataSource ds = null;

    protected DataSource getDataSource() {
        if (ds == null) {
            Context ctx = new InitialContext();
            Context envContext = (Context)ctx.lookup("java:comp/env");
            return (DataSource)envContext.lookup("jdbc/FooBar");    
        }
        return ds;
    }

    public void setDataSource(DataSource ds) {
        this.ds = ds;
    }
}

编辑：例如，您可以使用apache DBCP并以编程方式创建数据源：

BasicDataSource ds = new BasicDataSource();
ds.setDriverClassName("com.mysql.jdbc.Driver");
ds.setUrl("jdbc:mysql://localhost:3306/TestDB");
ds.setUsername("root");
ds.setPassword("");

现在您可以注入测试中使用的所需数据源

Answer 2

我使用JMockit（我最喜欢的模拟工具包）来做到这一点。

@Mocked(stubOutClassInitialization=true) final InitialContext unused = null;

编辑1：在另一个案例中，我有这样的事情：

new Expectations() {
    InitialContext ctx;
    {
        ...
        ctx = new InitialContext(); // expect constructor
        ctx.lookup(anyString); result = myMockThingie;
        ...
    }
}

在我的情况下，我并不特别感兴趣的是连接到备用数据库，所以这并不是你想要的情况，尽管看起来很容易满足你的需求。

编辑2：再想一想，也许你可以在这里完全躲避嘲笑。为什么不仅仅为了测试的目的，创建测试主题的子类并覆盖initConnection（）方法以使用您想要的连接？我经常使用这种技术，也许这对你来说最简单。请注意，您可以在测试类中创建一个静态内部类，该类是测试主题的子类，因此不必担心使用专门用于测试的工件来混淆您的生产代码。希望它有所帮助！

如何模拟jdbc的上下文数据源

2 个答案: