Examples with Dataset com.bakdata.conquery.models.datasets.Dataset used on opensource projects

Example 21 with Dataset

use of com.bakdata.conquery.models.datasets.Dataset in project conquery by bakdata.

the class SerializationTests method table.

@Test
public void table() throws JSONException, IOException {
    Dataset dataset = new Dataset();
    dataset.setName("datasetName");
    Table table = new Table();
    Column column = new Column();
    column.setLabel("colLabel");
    column.setName("colName");
    column.setType(MajorTypeId.STRING);
    column.setTable(table);
    table.setColumns(new Column[] { column });
    table.setDataset(dataset);
    table.setLabel("tableLabel");
    table.setName("tableName");
    CentralRegistry registry = new CentralRegistry();
    registry.register(dataset);
    registry.register(table);
    registry.register(column);
    final Validator validator = Validators.newValidator();
    SerializationTestUtil.forType(Table.class).registry(registry).injectables(new Injectable() {

        @Override
        public MutableInjectableValues inject(MutableInjectableValues values) {
            return values.add(Validator.class, validator);
        }
    }).test(table);
}

Example 22 with Dataset

use of com.bakdata.conquery.models.datasets.Dataset in project conquery by bakdata.

the class SerializationTests method cqConcept.

@Test
public void cqConcept() throws JSONException, IOException {
    final Dataset dataset = new Dataset();
    dataset.setName("dataset");
    final TreeConcept concept = new TreeConcept();
    concept.setName("concept");
    concept.setDataset(dataset);
    final ConceptTreeConnector connector = new ConceptTreeConnector();
    connector.setConcept(concept);
    concept.setConnectors(List.of(connector));
    final CQConcept cqConcept = new CQConcept();
    cqConcept.setElements(List.of(concept));
    cqConcept.setLabel("Label");
    final CQTable cqTable = new CQTable();
    cqTable.setConnector(connector);
    cqTable.setFilters(List.of());
    cqTable.setConcept(cqConcept);
    cqConcept.setTables(List.of(cqTable));
    final CentralRegistry registry = new CentralRegistry();
    registry.register(dataset);
    registry.register(concept);
    registry.register(connector);
    SerializationTestUtil.forType(CQConcept.class).registry(registry).test(cqConcept);
}

Example 23 with Dataset

use of com.bakdata.conquery.models.datasets.Dataset in project conquery by bakdata.

the class TestConquery method getSupport.

public synchronized StandaloneSupport getSupport(String name) {
    try {
        log.info("Setting up dataset");
        int count = NAME_COUNTS.merge(name, 0, (a, b) -> a + 1);
        if (count > 0) {
            name += "[" + count + "]";
        }
        Dataset dataset = new Dataset(name);
        standaloneCommand.getManager().getAdmin().getAdminDatasetProcessor().addDataset(dataset);
        return createSupport(dataset.getId(), name);
    } catch (Exception e) {
        return fail("Failed to create a support for " + name, e);
    }
}

Example 24 with Dataset

use of com.bakdata.conquery.models.datasets.Dataset in project conquery by bakdata.

the class QueryCleanupTaskTest method createManagedQuery.

private ManagedQuery createManagedQuery() {
    final CQAnd root = new CQAnd();
    root.setChildren(new ArrayList<>());
    ConceptQuery query = new ConceptQuery(root);
    final ManagedQuery managedQuery = new ManagedQuery(query, null, new Dataset("test"));
    managedQuery.setCreationTime(LocalDateTime.now().minus(queryExpiration).minusDays(1));
    STORAGE.addExecution(managedQuery);
    return managedQuery;
}

Example 25 with Dataset

use of com.bakdata.conquery.models.datasets.Dataset in project conquery by bakdata.

the class IdRefrenceTest method testListReferences.

@Test
public void testListReferences() throws IOException {
    CentralRegistry registry = new CentralRegistry();
    Dataset dataset = new Dataset();
    dataset.setName("dataset");
    Table table = new Table();
    table.setDataset(dataset);
    table.setName("table");
    registry.register(dataset);
    registry.register(table);
    final CentralRegistry metaRegistry = new CentralRegistry();
    User user = new User("usermail", "userlabel", mock(MetaStorage.class));
    metaRegistry.register(user);
    String json = Jackson.MAPPER.writeValueAsString(new ListHolder(Collections.singletonList(table), Collections.singletonList(user)));
    assertThat(json).contains("\"user.usermail\"").contains("\"dataset.table\"");
    ListHolder holder = new SingletonNamespaceCollection(registry, metaRegistry).injectIntoNew(Jackson.MAPPER.readerFor(ListHolder.class)).readValue(json);
    assertThat(holder.getUsers().get(0)).isSameAs(user);
    assertThat(holder.getTables().get(0)).isSameAs(table);
}