use of com.xrtb.db.User in project XRTB by benmfaul.
the class Config method setup.
/**
* JUNIT Test configuration for shards.
*
*/
public static void setup(String shard, int port) throws Exception {
try {
if (server == null) {
DbTools tools = new DbTools(redisHost);
tools.clear();
tools.loadDatabase("database.json");
server = new RTBServer("./Campaigns/payday.json", shard, port, port + 1);
int wait = 0;
while (!server.isReady() && wait < 10) {
Thread.sleep(1000);
wait++;
}
if (wait == 10) {
fail("Server never started");
}
} else {
Configuration c = Configuration.getInstance();
c.campaignsList.clear();
User u = DataBaseObject.getInstance().get("ben");
for (Campaign camp : u.campaigns) {
c.addCampaign("ben", camp.adId);
}
}
} catch (Exception e) {
e.printStackTrace();
fail(e.toString());
}
}
use of com.xrtb.db.User in project XRTB by benmfaul.
the class Config method setup.
public static void setup() throws Exception {
try {
DbTools tools = new DbTools("localhost:3000");
tools.clear();
tools.loadDatabase("database.json");
if (server == null) {
server = new RTBServer("./Campaigns/payday.json");
int wait = 0;
while (!server.isReady() && wait < 10) {
Thread.sleep(1000);
wait++;
}
if (wait == 10) {
fail("Server never started");
}
if (server.getCampaigns().size() == 0) {
fail("NO CAMPAIGNS LOADED");
}
Thread.sleep(1000);
} else {
Configuration c = Configuration.getInstance();
c.campaignsList.clear();
User u = DataBaseObject.getInstance().get("ben");
for (Campaign camp : u.campaigns) {
c.addCampaign("ben", camp.adId);
}
}
} catch (Exception e) {
e.printStackTrace();
fail(e.toString());
}
}
use of com.xrtb.db.User in project XRTB by benmfaul.
the class TestAdx method testSetup.
@BeforeClass
public static void testSetup() {
try {
DbTools tools = new DbTools("localhost:3000");
tools.clear();
tools.loadDatabase("database.json");
if (server == null) {
server = new RTBServer("./Campaigns/payday.json");
int wait = 0;
while (!server.isReady() && wait < 10) {
Thread.sleep(1000);
wait++;
}
if (wait == 10) {
fail("Server never started");
}
Thread.sleep(1000);
} else {
Configuration c = Configuration.getInstance();
c.campaignsList.clear();
User u = DataBaseObject.getInstance().get("ben");
for (Campaign camp : u.campaigns) {
c.addCampaign("ben", camp.adId);
}
}
} catch (Exception e) {
e.printStackTrace();
fail(e.toString());
}
}
use of com.xrtb.db.User in project XRTB by benmfaul.
the class TestDatabase method makeFile.
/**
* Test making the user database from scratch/
* @throws Exception on JSON parsing of the file.
*/
@Test
public void makeFile() throws Exception {
List<User> list = new ArrayList();
User u = new User("ben");
list.add(u);
String content = new String(Files.readAllBytes(Paths.get("stub.json")));
Campaign c = new Campaign(content);
c.adId = "ben:new-campaign";
u.campaigns.add(c);
assertTrue(c.date.size() == 2);
content = DbTools.mapper.writer().withDefaultPrettyPrinter().writeValueAsString(list);
System.out.println(content);
;
System.out.println("-------------------------");
List<User> x = DbTools.mapper.readValue(content, DbTools.mapper.getTypeFactory().constructCollectionType(List.class, User.class));
User z = x.get(0);
System.out.println(z);
}
use of com.xrtb.db.User in project XRTB by benmfaul.
the class JunkDomainUser method testTwoThreadAccessDatabase.
/*@Test
public void testSingleThreadAccess() {
User u = new User();
u.name = "Ben";
try {
db.getInstance().put(u);
u = null;
u = db.getInstance().get("Ben");
assertTrue(u.name.equals("Ben"));
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
}
}*/
@Test
public void testTwoThreadAccessDatabase() {
User u = new User();
u.name = "Ben";
CountDownLatch latch = new CountDownLatch(2);
CountDownLatch flag = new CountDownLatch(1);
JunkUser ben = new JunkUser(flag, latch, "Ben");
JunkUser peter = new JunkUser(flag, latch, "Peter");
flag.countDown();
try {
latch.await(5, TimeUnit.SECONDS);
System.out.println("Check ben");
u = db.get("Ben");
System.out.println("BEN: " + DbTools.mapper.writer().withDefaultPrettyPrinter().writeValueAsString(u));
assertTrue(u.name.equals("Ben"));
u = null;
System.out.println("Check peter");
u = db.get("Peter");
System.out.println("Peter" + DbTools.mapper.writer().withDefaultPrettyPrinter().writeValueAsString(u));
assertTrue(u.name.equals("Peter"));
db.clear();
Thread.sleep(200);
u = db.get("Ben");
assertNull(u);
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
fail(e.getMessage());
}
}
Aggregations