@Test
  public void givenEmptyJSONManyAssociationStateWhenAddingTwoRefsAtZeroIndexExpectCorrectOrder()
      throws JSONException {
    // Fake JSONManyAssociationState
    JSONObject state = new JSONObject();
    state.put(JSONKeys.PROPERTIES, new JSONObject());
    state.put(JSONKeys.ASSOCIATIONS, new JSONObject());
    state.put(JSONKeys.MANY_ASSOCIATIONS, new JSONObject());
    state.put(JSONKeys.NAMED_ASSOCIATIONS, new JSONObject());
    JSONEntityState entityState =
        new JSONEntityState(
            null,
            null,
            "0",
            System.currentTimeMillis(),
            EntityReference.parseEntityReference("123"),
            EntityStatus.NEW,
            null,
            state);
    JSONManyAssociationState jsonState = new JSONManyAssociationState(entityState, new JSONArray());

    jsonState.add(0, EntityReference.parseEntityReference("first"));
    jsonState.add(0, EntityReference.parseEntityReference("second"));

    assertThat(jsonState.count(), equalTo(2));
  }
  @Test
  public void givenEntityReferenceValueWhenSerializingAndDeserializingExpectEquals() {
    String serialized =
        valueSerialization.serialize(EntityReference.parseEntityReference("ABCD-1234"));
    assertThat(serialized, equalTo("ABCD-1234"));

    EntityReference deserialized =
        valueSerialization.deserialize(EntityReference.class, serialized);
    assertThat(deserialized, equalTo(EntityReference.parseEntityReference("ABCD-1234")));
  }
Ejemplo n.º 3
0
  @Override
  public EntityReference associationValueOf(QualifiedName stateName) {
    try {
      Object jsonValue = state.getJSONObject(JSONKeys.ASSOCIATIONS).opt(stateName.name());
      if (jsonValue == null) {
        return null;
      }

      EntityReference value =
          jsonValue == JSONObject.NULL
              ? null
              : EntityReference.parseEntityReference((String) jsonValue);
      return value;
    } catch (JSONException e) {
      throw new EntityStoreException(e);
    }
  }
  @Test
  public void givenJSONManyAssociationStateWhenChangingReferencesExpectCorrectBehavior()
      throws JSONException {
    // Fake JSONManyAssociationState
    JSONObject state = new JSONObject();
    state.put(JSONKeys.PROPERTIES, new JSONObject());
    state.put(JSONKeys.ASSOCIATIONS, new JSONObject());
    state.put(JSONKeys.MANY_ASSOCIATIONS, new JSONObject());
    state.put(JSONKeys.NAMED_ASSOCIATIONS, new JSONObject());
    JSONEntityState entityState =
        new JSONEntityState(
            null,
            null,
            "0",
            System.currentTimeMillis(),
            EntityReference.parseEntityReference("123"),
            EntityStatus.NEW,
            null,
            state);
    JSONManyAssociationState jsonState = new JSONManyAssociationState(entityState, new JSONArray());

    assertThat(jsonState.contains(EntityReference.parseEntityReference("NOT_PRESENT")), is(false));

    jsonState.add(0, EntityReference.parseEntityReference("0"));
    jsonState.add(1, EntityReference.parseEntityReference("1"));
    jsonState.add(2, EntityReference.parseEntityReference("2"));

    assertThat(jsonState.contains(EntityReference.parseEntityReference("1")), is(true));

    assertThat(jsonState.get(0).identity(), equalTo("0"));
    assertThat(jsonState.get(1).identity(), equalTo("1"));
    assertThat(jsonState.get(2).identity(), equalTo("2"));

    assertThat(jsonState.count(), equalTo(3));

    jsonState.remove(EntityReference.parseEntityReference("1"));

    assertThat(jsonState.count(), equalTo(2));
    assertThat(jsonState.contains(EntityReference.parseEntityReference("1")), is(false));
    assertThat(jsonState.get(0).identity(), equalTo("0"));
    assertThat(jsonState.get(1).identity(), equalTo("2"));

    jsonState.add(2, EntityReference.parseEntityReference("1"));

    assertThat(jsonState.count(), equalTo(3));

    jsonState.add(0, EntityReference.parseEntityReference("A"));
    jsonState.add(0, EntityReference.parseEntityReference("B"));
    jsonState.add(0, EntityReference.parseEntityReference("C"));

    assertThat(jsonState.count(), equalTo(6));

    assertThat(jsonState.get(0).identity(), equalTo("C"));
    assertThat(jsonState.get(1).identity(), equalTo("B"));
    assertThat(jsonState.get(2).identity(), equalTo("A"));

    assertThat(jsonState.contains(EntityReference.parseEntityReference("C")), is(true));
    assertThat(jsonState.contains(EntityReference.parseEntityReference("B")), is(true));
    assertThat(jsonState.contains(EntityReference.parseEntityReference("A")), is(true));
    assertThat(jsonState.contains(EntityReference.parseEntityReference("0")), is(true));
    assertThat(jsonState.contains(EntityReference.parseEntityReference("2")), is(true));
    assertThat(jsonState.contains(EntityReference.parseEntityReference("1")), is(true));

    List<String> refList =
        toList(
            map(
                new Function<EntityReference, String>() {
                  @Override
                  public String map(EntityReference from) {
                    return from.identity();
                  }
                },
                jsonState));
    assertThat(refList.isEmpty(), is(false));
    assertArrayEquals(new String[] {"C", "B", "A", "0", "2", "1"}, refList.toArray());
  }