Examples with ColumnSetBuilder org.activityinfo.store.query.server.ColumnSetBuilder used on opensource projects

Example 11 with ColumnSetBuilder

use of org.activityinfo.store.query.server.ColumnSetBuilder in project activityinfo by bedatadriven.

the class HrdCatalogTest method subFormTest.

@Test
public void subFormTest() {
    // Typical scenario with a household interview form
    // and a repeating househould member form
    FormClass hhForm = new FormClass(ResourceId.generateId());
    FormClass memberForm = new FormClass(ResourceId.generateId());
    memberForm.setParentFormId(hhForm.getId());
    hhForm.setParentFormId(ResourceId.valueOf("foo"));
    hhForm.setLabel("Household interview");
    FormField hhIdField = hhForm.addField().setLabel("Household ID").setType(TextType.SIMPLE);
    hhForm.addField().setLabel("Household Memmbers").setType(new SubFormReferenceType(memberForm.getId()));
    memberForm.setLabel("Household Members");
    FormField nameField = memberForm.addField().setLabel("Name").setType(TextType.SIMPLE);
    FormField ageField = memberForm.addField().setLabel("Age").setType(new QuantityType("years"));
    HrdStorageProvider catalog = new HrdStorageProvider();
    catalog.create(hhForm);
    catalog.create(memberForm);
    TypedRecordUpdate hh1 = new TypedRecordUpdate();
    hh1.setUserId(userId);
    hh1.setRecordId(ResourceId.generateSubmissionId(hhForm));
    hh1.set(hhIdField.getId(), TextValue.valueOf("HH1"));
    TypedRecordUpdate hh2 = new TypedRecordUpdate();
    hh2.setUserId(userId);
    hh2.setRecordId(ResourceId.generateSubmissionId(hhForm));
    hh2.set(hhIdField.getId(), TextValue.valueOf("HH2"));
    TypedRecordUpdate father1 = new TypedRecordUpdate();
    father1.setUserId(userId);
    father1.setRecordId(ResourceId.generateSubmissionId(memberForm));
    father1.setParentId(hh1.getRecordId());
    father1.set(nameField.getId(), TextValue.valueOf("Homer"));
    father1.set(ageField.getId(), new Quantity(40));
    TypedRecordUpdate father2 = new TypedRecordUpdate();
    father2.setUserId(userId);
    father2.setRecordId(ResourceId.generateSubmissionId(memberForm));
    father2.setParentId(hh2.getRecordId());
    father2.set(nameField.getId(), TextValue.valueOf("Ned"));
    father2.set(ageField.getId(), new Quantity(41));
    Optional<FormStorage> hhCollection = catalog.getForm(hhForm.getId());
    assertTrue(hhCollection.isPresent());
    hhCollection.get().add(hh1);
    hhCollection.get().add(hh2);
    Optional<FormStorage> memberCollection = catalog.getForm(memberForm.getId());
    assertTrue(memberCollection.isPresent());
    memberCollection.get().add(father1);
    memberCollection.get().add(father2);
    QueryModel queryModel = new QueryModel(memberForm.getId());
    queryModel.selectResourceId().as("id");
    queryModel.selectField("Household ID").as("hh_id");
    queryModel.selectField("Name").as("member_name");
    queryModel.selectField("Age").as("member_age");
    ColumnSetBuilder builder = new ColumnSetBuilder(catalog, new NullFormSupervisor());
    ColumnSet columnSet = builder.build(queryModel);
    System.out.println(columnSet);
    assertThat(columnSet.getNumRows(), equalTo(2));
}

Example 12 with ColumnSetBuilder

use of org.activityinfo.store.query.server.ColumnSetBuilder in project activityinfo by bedatadriven.

the class HrdCatalogTest method createResource.

@Test
public void createResource() {
    FormClass formClass = new FormClass(ResourceId.generateId());
    formClass.setParentFormId(ResourceId.valueOf("foo"));
    formClass.setLabel("NFI Distributions");
    FormField nameField = formClass.addField(ResourceId.generateId()).setLabel("Village name").setCode("VILLAGE").setType(TextType.SIMPLE);
    HrdStorageProvider catalog = new HrdStorageProvider();
    Updater updater = new Updater(catalog, userId, new BlobAuthorizerStub(), new HrdSerialNumberProvider());
    catalog.create(formClass);
    String[] villageNames = new String[] { "Rutshuru", "Beni", "Goma" };
    for (String villageName : villageNames) {
        TypedRecordUpdate update = new TypedRecordUpdate();
        update.setUserId(userId);
        update.setFormId(formClass.getId());
        update.setRecordId(ResourceId.generateSubmissionId(formClass));
        update.set(nameField.getId(), TextValue.valueOf(villageName));
        updater.execute(update);
    }
    QueryModel queryModel = new QueryModel(formClass.getId());
    queryModel.selectResourceId().as("id");
    queryModel.selectField("VILLAGE").as("village");
    ColumnSetBuilder builder = new ColumnSetBuilder(catalog, new NullFormSupervisor());
    ColumnSet columnSet = builder.build(queryModel);
    System.out.println(columnSet);
}

Example 13 with ColumnSetBuilder

use of org.activityinfo.store.query.server.ColumnSetBuilder in project activityinfo by bedatadriven.

the class SubFormAggregationTest method subFormAggregationTest.

@Test
public void subFormAggregationTest() {
    // Typical scenario with a household interview form
    // and a repeating househould member form
    FormClass siteForm = new FormClass(ResourceId.generateId());
    siteForm.setParentFormId(ResourceId.ROOT_ID);
    FormClass monthlyForm = new FormClass(ResourceId.generateId());
    monthlyForm.setParentFormId(siteForm.getId());
    monthlyForm.setSubFormKind(SubFormKind.MONTHLY);
    siteForm.setLabel("Household interview");
    FormField villageField = siteForm.addField().setLabel("Village Name").setType(TextType.SIMPLE);
    siteForm.addField().setLabel("Maximum Beneficiaries").setCode("BENE").setType(new CalculatedFieldType("MAX(HH)"));
    siteForm.addField().setLabel("Monthly Activities").setType(new SubFormReferenceType(monthlyForm.getId()));
    monthlyForm.setLabel("Monthly Activities");
    FormField countField = monthlyForm.addField().setLabel("Number of Beneficiaries").setCode("HH").setType(new QuantityType("households"));
    HrdStorageProvider catalog = new HrdStorageProvider();
    catalog.create(siteForm);
    catalog.create(monthlyForm);
    TypedRecordUpdate v1 = new TypedRecordUpdate();
    v1.setUserId(userId);
    v1.setRecordId(ResourceId.generateSubmissionId(siteForm));
    v1.set(villageField.getId(), TextValue.valueOf("Rutshuru"));
    TypedRecordUpdate v2 = new TypedRecordUpdate();
    v2.setUserId(userId);
    v2.setRecordId(ResourceId.generateSubmissionId(siteForm));
    v2.set(villageField.getId(), TextValue.valueOf("Beni"));
    TypedRecordUpdate month1 = new TypedRecordUpdate();
    month1.setUserId(userId);
    month1.setRecordId(ResourceId.generateSubmissionId(monthlyForm));
    month1.setParentId(v1.getRecordId());
    month1.set(countField.getId(), new Quantity(40));
    TypedRecordUpdate month2 = new TypedRecordUpdate();
    month2.setUserId(userId);
    month2.setRecordId(ResourceId.generateSubmissionId(monthlyForm));
    month2.setParentId(v1.getRecordId());
    month2.set(countField.getId(), new Quantity(30));
    TypedRecordUpdate month3 = new TypedRecordUpdate();
    month3.setUserId(userId);
    month3.setRecordId(ResourceId.generateSubmissionId(monthlyForm));
    month3.setParentId(v2.getRecordId());
    month3.set(countField.getId(), new Quantity(47));
    FormStorage siteCollection = catalog.getForm(siteForm.getId()).get();
    siteCollection.add(v1);
    siteCollection.add(v2);
    Optional<FormStorage> monthCollection = catalog.getForm(monthlyForm.getId());
    assertTrue(monthCollection.isPresent());
    monthCollection.get().add(month1);
    monthCollection.get().add(month2);
    monthCollection.get().add(month3);
    QueryModel queryModel = new QueryModel(siteForm.getId());
    queryModel.selectResourceId().as("id");
    queryModel.selectField("Village Name").as("village");
    queryModel.selectField("BENE").as("max_hh");
    ColumnSetBuilder builder = new ColumnSetBuilder(catalog, new NullFormSupervisor());
    ColumnSet columnSet = builder.build(queryModel);
    System.out.println(columnSet);
    assertThat(columnSet.getNumRows(), equalTo(2));
}

Example 14 with ColumnSetBuilder

use of org.activityinfo.store.query.server.ColumnSetBuilder in project activityinfo by bedatadriven.

the class DateFunctionQueryTest method testToday.

@Test
public void testToday() {
    TestingStorageProvider catalog = new TestingStorageProvider();
    IntakeForm intakeForm = catalog.getIntakeForm();
    ColumnSetBuilder builder = new ColumnSetBuilder(catalog, new NullFormScanCache(), new NullFormSupervisor());
    QueryModel queryModel = new QueryModel(intakeForm.getFormId());
    queryModel.selectExpr("TODAY()").as("today");
    queryModel.selectExpr("YEARFRAC(TODAY(), DOB)").as("age");
    queryModel.selectExpr("DOB").as("dob");
    ColumnSet columnSet = builder.build(queryModel);
    ColumnView today = columnSet.getColumnView("today");
    ColumnView age = columnSet.getColumnView("age");
    ColumnView dob = columnSet.getColumnView("dob");
    for (int i = 0; i < columnSet.getNumRows(); i++) {
        if (dob.getString(i) != null) {
            double ageInYears = age.getDouble(i);
            System.out.println(dob.get(i) + " " + ageInYears);
            if (Double.isNaN(ageInYears)) {
                throw new AssertionError();
            }
        }
    }
    System.out.println(age);
}

Example 15 with ColumnSetBuilder

use of org.activityinfo.store.query.server.ColumnSetBuilder in project activityinfo by bedatadriven.

the class EnumerationQueryTest method enumRefTests.

@Test
public void enumRefTests() {
    TestingStorageProvider catalog = new TestingStorageProvider();
    ColumnSetBuilder builder = new ColumnSetBuilder(catalog, new NullFormScanCache(), new NullFormSupervisor());
    IntakeForm intakeForm = catalog.getIntakeForm();
    QueryModel queryModel = new QueryModel(intakeForm.getFormId());
    queryModel.selectField(intakeForm.getNationalityFieldId()).as("nationality");
    queryModel.selectExpr(intakeForm.getNationalityFieldId() + "==" + "\"" + intakeForm.getPalestinianId() + "\"").as("palestinian");
    queryModel.selectExpr("\"" + intakeForm.getPalestinianId() + "\"" + "==" + intakeForm.getNationalityFieldId()).as("palestinianInverse");
    ColumnSet columnSet = builder.build(queryModel);
    ColumnView nationality = columnSet.getColumnView("nationality");
    ColumnView palestinian = columnSet.getColumnView("palestinian");
    ColumnView palestinianInverse = columnSet.getColumnView("palestinianInverse");
    // Correct Tests
    assertThat(nationality.get(1).toString(), equalTo(NAT_PAL));
    assertThat(Boolean.valueOf(palestinian.get(1).toString()), equalTo(true));
    assertThat(Boolean.valueOf(palestinianInverse.get(1).toString()), equalTo(true));
    // Incorrect Tests
    // Multiple selected values should return null
    assertThat(nationality.get(5), equalTo(null));
}