Examples with Database com.vaticle.typedb.core.common.parameters.Options.Database used on opensource projects

Example 1 with Database

use of com.vaticle.typedb.core.common.parameters.Options.Database in project grakn by graknlabs.

the class QueryTest method test_query_undefine.

@Test
public void test_query_undefine() throws IOException {
    Util.resetDirectory(dataDir);
    try (TypeDB.DatabaseManager typedb = CoreDatabaseManager.open(options)) {
        typedb.create(database);
        try (TypeDB.Session session = typedb.session(database, Arguments.Session.Type.SCHEMA)) {
            try (TypeDB.Transaction transaction = session.transaction(Arguments.Transaction.Type.WRITE)) {
                TypeQLDefine query = TypeQL.parseQuery(new String(Files.readAllBytes(Paths.get("test/integration/schema.tql")), UTF_8));
                transaction.query().define(query);
                transaction.commit();
            }
            try (TypeDB.Transaction transaction = session.transaction(Arguments.Transaction.Type.WRITE)) {
                String queryString = "undefine analysis abstract, owns created, plays commit-analysis:analysis;";
                TypeQLUndefine query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine rule performance-tracker-rule;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine performance-tracker relates tracker;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine email regex '.+\\@.+\\..+';";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine index sub attribute, value long;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                // undefine first 4 rules
                queryString = "undefine rule repo-fork-rule;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine rule repo-dependency-transitive-rule;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine rule repo-dependency-transitive-type-rule;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                queryString = "undefine rule repo-collaborator-org-rule;";
                query = TypeQL.parseQuery(queryString);
                transaction.query().undefine(query);
                transaction.commit();
            }
            try (TypeDB.Transaction tx = session.transaction(Arguments.Transaction.Type.READ)) {
                EntityType analysis = tx.concepts().getEntityType("analysis");
                RelationType performanceTracker = tx.concepts().getRelationType("performance-tracker");
                RoleType commitAnalysisAnalysis = tx.concepts().getRelationType("commit-analysis").getRelates("analysis");
                AttributeType.DateTime created = tx.concepts().getAttributeType("created").asDateTime();
                AttributeType.String email = tx.concepts().getAttributeType("email").asString();
                assertNotNulls(analysis, performanceTracker, commitAnalysisAnalysis, created, email);
                assertFalse(analysis.isAbstract());
                assertTrue(analysis.getOwns().noneMatch(att -> att.equals(created)));
                assertTrue(analysis.getPlays().noneMatch(rol -> rol.equals(commitAnalysisAnalysis)));
                assertTrue(performanceTracker.getRelates().noneMatch(rol -> rol.getLabel().name().equals("tracker")));
                assertNull(email.getRegex());
                AttributeType index = tx.concepts().getAttributeType("index");
                assertNull(index);
                assertNull(tx.logic().getRule("repo-fork-rule"));
                assertNull(tx.logic().getRule("repo-dependency-transitive-rule"));
                assertNull(tx.logic().getRule("repo-dependency-transitive-type-rule"));
                assertNull(tx.logic().getRule("repo-collaborator-org-rule"));
                // check total count
                assertEquals(15 - 5, tx.logic().rules().toList().size());
                // a query that used to trigger a rule should not cause an error
                List<ConceptMap> answers = tx.query().match(TypeQL.parseQuery("match $x isa repo-fork;").asMatch()).toList();
            }
        }
    }
}

Example 2 with Database

use of com.vaticle.typedb.core.common.parameters.Options.Database in project grakn by graknlabs.

the class QueryTest method test_query_define.

@Test
public void test_query_define() throws IOException {
    Util.resetDirectory(dataDir);
    try (TypeDB.DatabaseManager typedb = CoreDatabaseManager.open(options)) {
        typedb.create(database);
        try (TypeDB.Session session = typedb.session(database, Arguments.Session.Type.SCHEMA)) {
            try (TypeDB.Transaction transaction = session.transaction(Arguments.Transaction.Type.WRITE)) {
                TypeQLDefine query = TypeQL.parseQuery(new String(Files.readAllBytes(Paths.get("test/integration/schema.tql")), UTF_8));
                transaction.query().define(query);
                transaction.commit();
            }
            try (TypeDB.Transaction tx = session.transaction(Arguments.Transaction.Type.READ)) {
                AttributeType.String name = tx.concepts().getAttributeType("name").asString();
                AttributeType.String symbol = tx.concepts().getAttributeType("symbol").asString();
                AttributeType.Boolean active = tx.concepts().getAttributeType("active").asBoolean();
                AttributeType.Long priority = tx.concepts().getAttributeType("priority").asLong();
                assertNotNulls(name, symbol, active, priority);
                EntityType organisation = tx.concepts().getEntityType("organisation");
                EntityType team = tx.concepts().getEntityType("team");
                EntityType user = tx.concepts().getEntityType("user");
                EntityType repository = tx.concepts().getEntityType("repository");
                EntityType branchRule = tx.concepts().getEntityType("branch-rule");
                EntityType commit = tx.concepts().getEntityType("commit");
                assertNotNulls(organisation, team, user, repository, branchRule, commit);
                assertTrue(organisation.getOwns().anyMatch(a -> a.equals(name)));
                assertTrue(team.getOwns().anyMatch(a -> a.equals(symbol)));
                assertTrue(user.getOwns().anyMatch(a -> a.equals(name)));
                assertTrue(repository.getOwns().anyMatch(a -> a.equals(active)));
                assertTrue(branchRule.getOwns().anyMatch(a -> a.equals(priority)));
                assertTrue(commit.getOwns().anyMatch(a -> a.equals(symbol)));
                RelationType orgTeam = tx.concepts().getRelationType("org-team");
                RelationType teamMember = tx.concepts().getRelationType("team-member");
                RelationType repoDependency = tx.concepts().getRelationType("repo-dependency");
                assertNotNulls(orgTeam, teamMember, repoDependency);
                RoleType orgTeam_org = orgTeam.getRelates("org");
                RoleType orgTeam_team = orgTeam.getRelates("team");
                RoleType teamMember_team = teamMember.getRelates("team");
                RoleType teamMember_member = teamMember.getRelates("member");
                assertNotNulls(orgTeam_org, orgTeam_team, teamMember_team, teamMember_member);
                assertTrue(organisation.getPlays().anyMatch(r -> r.equals(orgTeam_org)));
                assertTrue(team.getPlays().anyMatch(r -> r.equals(orgTeam_team)));
                assertTrue(team.getPlays().anyMatch(r -> r.equals(teamMember_team)));
                assertTrue(user.getPlays().anyMatch(r -> r.equals(teamMember_member)));
                // check first 4 rules
                assertNotNull(tx.logic().getRule("repo-fork-rule"));
                assertNotNull(tx.logic().getRule("repo-dependency-transitive-rule"));
                assertNotNull(tx.logic().getRule("repo-dependency-transitive-type-rule"));
                assertNotNull(tx.logic().getRule("repo-collaborator-org-rule"));
                // check total count
                assertEquals(15, tx.logic().rules().toList().size());
            }
        }
    }
}