@Test(groups = "slow") public void testWithArgument() throws Exception { dbi.registerArgumentFactory(new EnumArgumentFactory()); final SomethingSqlDao somethingSqlDao = dbi.onDemand(SomethingSqlDao.class); final Bier bierPierre = Bier.ipa; somethingSqlDao.create(1, "pierre", bierPierre); final Bier bierStephane = Bier.hefeweizen; somethingSqlDao.create(2, "stephane", bierStephane); final String bierStringPierre = somethingSqlDao.getBier(1); Assert.assertEquals(bierStringPierre, bierPierre.toString()); final String bierStringStephane = somethingSqlDao.getBier(2); Assert.assertEquals(bierStringStephane, bierStephane.toString()); }
@BeforeClass(groups = "slow") public void beforeClass() throws Exception { loadSystemPropertiesFromClasspath("/queue.properties"); clock = new ClockMock(); embeddedDB = new MySQLEmbeddedDB("killbillq", "killbillq", "killbillq"); embeddedDB.initialize(); embeddedDB.start(); final String ddl = toString(Resources.getResource("com/ning/billing/queue/ddl.sql").openStream()); embeddedDB.executeScript(ddl); dbi = new DBI(embeddedDB.getDataSource()); dbi.registerArgumentFactory(new UUIDArgumentFactory()); dbi.registerArgumentFactory(new DateTimeZoneArgumentFactory()); dbi.registerArgumentFactory(new DateTimeArgumentFactory()); dbi.registerArgumentFactory(new LocalDateArgumentFactory()); dbi.registerArgumentFactory(new EnumArgumentFactory()); dbi.registerMapper(new UUIDMapper()); final ConfigSource configSource = new SimplePropertyConfigSource(System.getProperties()); persistentBusConfig = new ConfigurationObjectFactory(configSource).buildWithReplacements(PersistentBusConfig.class, ImmutableMap.<String, String>of("instanceName", "main")); notificationQueueConfig = new ConfigurationObjectFactory(configSource).buildWithReplacements(NotificationQueueConfig.class, ImmutableMap.<String, String>of("instanceName", "main")); }