private static List<String> createRandomDocuments(int howMany) {
Random rand = new Random();
Json.ObjectMapperExt mapper = Json.mapper();
return IntStream.rangeClosed(0, howMany - 1)
.mapToObj(
(x) -> {
ObjectNode node = mapper.createObjectNode();
node.put("generated", "generated-" + rand.nextInt());
node.put("idx", x);
node.put("rand", rand.nextInt());
return node;
})
.map(
(doc) -> {
try {
ODocument d = DocumentService.create(TEST_COLLECTION, doc);
return d.<String>field("id");
} catch (Throwable throwable) {
return null;
}
})
.filter(Objects::nonNull)
.collect(Collectors.toList());
}
public class ScriptDocumentCommandTest {
private static final String TEST_USER = "******" + UUID.randomUUID();
private static final String TEST_ALT_USER = "******" + UUID.randomUUID();
private static final String TEST_COLLECTION = "script_command_test_coll_" + UUID.randomUUID();
private static volatile List<String> sGenIds;
private static final Json.ObjectMapperExt MAPPER = Json.mapper();
private static List<String> createRandomDocuments(int howMany) {
Random rand = new Random();
Json.ObjectMapperExt mapper = Json.mapper();
return IntStream.rangeClosed(0, howMany - 1)
.mapToObj(
(x) -> {
ObjectNode node = mapper.createObjectNode();
node.put("generated", "generated-" + rand.nextInt());
node.put("idx", x);
node.put("rand", rand.nextInt());
return node;
})
.map(
(doc) -> {
try {
ODocument d = DocumentService.create(TEST_COLLECTION, doc);
return d.<String>field("id");
} catch (Throwable throwable) {
return null;
}
})
.filter(Objects::nonNull)
.collect(Collectors.toList());
}
@BeforeClass
public static void initTestser() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", "admin", "admin");
ODocument user =
UserService.signUp(TEST_USER, TEST_USER, new Date(), null, null, null, null, false);
assertNotNull(user);
ODocument alt =
UserService.signUp(
TEST_ALT_USER, TEST_ALT_USER, new Date(), null, null, null, null, false);
assertNotNull(alt);
CollectionService.create(TEST_COLLECTION);
DbHelper.close(DbHelper.getConnection());
DbHelper.open("1234567890", TEST_USER, TEST_USER);
sGenIds = createRandomDocuments(10);
DbHelper.close(DbHelper.getConnection());
} catch (Throwable e) {
fail(ExceptionUtils.getFullStackTrace(e));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testGrantAndRevokeUpdate() {
running(
fakeApplication(),
() -> {
try {
// initial check. user TEST_ALT_USER cannot update the doc
try {
DbHelper.open("1234567890", TEST_ALT_USER, TEST_ALT_USER);
ObjectNode paramsUpdate = MAPPER.createObjectNode();
paramsUpdate.put("collection", TEST_COLLECTION);
paramsUpdate.put("id", sGenIds.get(0));
paramsUpdate.put("data", MAPPER.readTree("{\"upd\":\"updValue\"}"));
ObjectNode cmdUpdate = ScriptCommands.createCommand("documents", "put", paramsUpdate);
JsonNode nodeUpdate = CommandRegistry.execute(cmdUpdate, null);
DbHelper.close(DbHelper.getConnection());
fail("The user should not update the doc, but it dit it!");
} catch (CommandExecutionException e) {
} catch (Exception e) {
Logger.debug("OOOPS! something went wrong! ", e);
fail(ExceptionUtils.getFullStackTrace(e));
throw e;
} finally {
DbHelper.close(DbHelper.getConnection());
}
// use TEST_USER grant permission to update the doc to the user TEST_ALT_USER
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode params = MAPPER.createObjectNode();
ObjectNode users = MAPPER.createObjectNode();
ArrayNode update = MAPPER.createArrayNode();
update.add(TEST_ALT_USER);
users.put("update", update);
users.put("read", update);
params.put("collection", TEST_COLLECTION);
params.put("id", sGenIds.get(0));
params.put("users", users);
ObjectNode grant = ScriptCommands.createCommand("documents", "grant", params);
JsonNode node = CommandRegistry.execute(grant, null);
DbHelper.close(DbHelper.getConnection());
// now user TEST_ALT_USER can update the doc
DbHelper.open("1234567890", TEST_ALT_USER, TEST_ALT_USER);
ObjectNode paramsUpdate = MAPPER.createObjectNode();
paramsUpdate.put("collection", TEST_COLLECTION);
paramsUpdate.put("id", sGenIds.get(0));
paramsUpdate.put(
"data",
MAPPER.readTree(
"{\"generated\":\"generated-123\",\"rand\":123,\"idx\":0,\"upd\":\"updValue\"}"));
ObjectNode cmdUpdate = ScriptCommands.createCommand("documents", "put", paramsUpdate);
JsonNode nodeUpdate = CommandRegistry.execute(cmdUpdate, null);
DbHelper.close(DbHelper.getConnection());
// now the grant is revoked
DbHelper.open("1234567890", TEST_USER, TEST_USER);
params = MAPPER.createObjectNode();
users = MAPPER.createObjectNode();
update = MAPPER.createArrayNode();
update.add(TEST_ALT_USER);
users.put("update", update);
users.put("read", update);
params.put("collection", TEST_COLLECTION);
params.put("id", sGenIds.get(0));
params.put("users", users);
grant = ScriptCommands.createCommand("documents", "revoke", params);
node = CommandRegistry.execute(grant, null);
DbHelper.close(DbHelper.getConnection());
} catch (Throwable tr) {
Logger.debug(ExceptionUtils.getFullStackTrace(tr));
fail(ExceptionUtils.getFullStackTrace(tr));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testGrantAndRevokeRead() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", TEST_ALT_USER, TEST_ALT_USER);
ObjectNode coll = MAPPER.createObjectNode();
coll.put("collection", TEST_COLLECTION);
ObjectNode cmd = ScriptCommands.createCommand("documents", "list", coll);
JsonNode exec = CommandRegistry.execute(cmd, null);
assertNotNull(exec);
assertTrue(exec.isArray());
assertEquals(0, exec.size());
DbHelper.close(DbHelper.getConnection());
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode params = MAPPER.createObjectNode();
ObjectNode users = MAPPER.createObjectNode();
ArrayNode read = MAPPER.createArrayNode();
read.add(TEST_ALT_USER);
users.put("read", read);
params.put("collection", TEST_COLLECTION);
params.put("id", sGenIds.get(0));
params.put("users", users);
ObjectNode grant = ScriptCommands.createCommand("documents", "grant", params);
JsonNode node = CommandRegistry.execute(grant, null);
assertNotNull(node);
assertTrue(node.isBoolean());
assertTrue(node.asBoolean());
DbHelper.close(DbHelper.getConnection());
DbHelper.open("1234567890", TEST_ALT_USER, TEST_ALT_USER);
JsonNode execWithGrants = CommandRegistry.execute(cmd, null);
assertNotNull(execWithGrants);
assertTrue(execWithGrants.isArray());
assertEquals(1, execWithGrants.size());
DbHelper.close(DbHelper.getConnection());
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode revoke = ScriptCommands.createCommand("documents", "revoke", params);
JsonNode revoked = CommandRegistry.execute(revoke, null);
assertNotNull(revoked);
assertTrue(revoked.isBoolean());
assertTrue(revoked.asBoolean());
DbHelper.close(DbHelper.getConnection());
DbHelper.open("1234567890", TEST_ALT_USER, TEST_ALT_USER);
JsonNode execWithoutGrants = CommandRegistry.execute(cmd, null);
assertNotNull(execWithoutGrants);
assertTrue(execWithoutGrants.isArray());
assertEquals(0, execWithoutGrants.size());
DbHelper.close(DbHelper.getConnection());
} catch (Throwable tr) {
fail(ExceptionUtils.getFullStackTrace(tr));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testCommandListDocuments() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode cmd = MAPPER.createObjectNode();
ObjectNode p = MAPPER.createObjectNode();
p.put("collection", TEST_COLLECTION);
cmd.put(ScriptCommand.RESOURCE, "documents");
cmd.put(ScriptCommand.NAME, "list");
cmd.put(ScriptCommand.PARAMS, p);
JsonNode node = CommandRegistry.execute(cmd, null);
assertNotNull(node);
assertTrue(node.isArray());
assertEquals(10, node.size());
} catch (Throwable t) {
fail(ExceptionUtils.getFullStackTrace(t));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testCommandGetSingleDocument() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode cmd = MAPPER.createObjectNode();
ObjectNode p = MAPPER.createObjectNode();
p.put("collection", TEST_COLLECTION);
p.put("id", sGenIds.get(0));
cmd.put(ScriptCommand.RESOURCE, "documents");
cmd.put(ScriptCommand.NAME, "get");
cmd.put(ScriptCommand.PARAMS, p);
JsonNode node = CommandRegistry.execute(cmd, null);
assertNotNull(node);
assertTrue(node.isObject());
assertNotNull(node.get("generated"));
assertNotNull(node.get("id"));
assertEquals(node.get("id").asText(), sGenIds.get(0));
assertEquals(node.get("@class").asText(), TEST_COLLECTION);
} catch (Throwable t) {
fail(ExceptionUtils.getFullStackTrace(t));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testCommandGetFilteredCollection() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode cmd = MAPPER.createObjectNode();
ObjectNode p = MAPPER.createObjectNode();
ObjectNode q = MAPPER.createObjectNode();
q.put("where", "idx < ?");
ArrayNode params = MAPPER.createArrayNode();
params.add("5");
q.put("params", params);
p.put("collection", TEST_COLLECTION);
p.put("query", q);
cmd.put(ScriptCommand.RESOURCE, "documents");
cmd.put(ScriptCommand.NAME, "list");
cmd.put(ScriptCommand.PARAMS, p);
JsonNode node = CommandRegistry.execute(cmd, null);
assertNotNull(node);
assertTrue(node.isArray());
assertEquals(5, node.size());
} catch (Throwable t) {
fail(ExceptionUtils.getFullStackTrace(t));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testCreateDocument() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode params = MAPPER.createObjectNode();
ObjectNode doc = MAPPER.createObjectNode();
doc.put("fresh", "fresh");
params.put("collection", TEST_COLLECTION);
params.put("data", doc);
ObjectNode cmd = ScriptCommands.createCommand("documents", "post", params);
JsonNode exec = CommandRegistry.execute(cmd, null);
assertNotNull(exec);
assertTrue(exec.isObject());
assertNotNull(exec.get("id"));
assertEquals(TEST_COLLECTION, exec.get("@class").asText());
} catch (Throwable t) {
fail(ExceptionUtils.getFullStackTrace(t));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
@Test
public void testCommandAlterDocument() {
running(
fakeApplication(),
() -> {
try {
DbHelper.open("1234567890", TEST_USER, TEST_USER);
ObjectNode cmd = MAPPER.createObjectNode();
ObjectNode p = MAPPER.createObjectNode();
p.put("id", sGenIds.get(0));
p.put("collection", TEST_COLLECTION);
cmd.put(ScriptCommand.RESOURCE, "documents");
cmd.put(ScriptCommand.NAME, "get");
cmd.put(ScriptCommand.PARAMS, p);
JsonNode node = CommandRegistry.execute(cmd, null);
assertNotNull(node);
assertTrue(node.isObject());
ObjectNode doc = node.deepCopy();
doc.put("extra", "extra");
ObjectNode upd = MAPPER.createObjectNode();
upd.put(ScriptCommand.RESOURCE, "documents");
upd.put(ScriptCommand.NAME, "put");
ObjectNode params = MAPPER.createObjectNode();
params.put("collection", TEST_COLLECTION);
params.put("id", doc.get("id").asText());
params.put("data", doc);
upd.put(ScriptCommand.PARAMS, params);
JsonNode res = CommandRegistry.execute(upd, null);
assertNotNull(res);
assertTrue(res.isObject());
assertNotNull(res.get("extra"));
assertEquals(res.get("id"), doc.get("id"));
assertEquals("extra", res.get("extra").asText());
} catch (Throwable t) {
fail(ExceptionUtils.getFullStackTrace(t));
} finally {
DbHelper.close(DbHelper.getConnection());
}
});
}
}