Fix other equality checks

disneystreaming · May 30, 2024 · a9a8206 · a9a8206
1 parent 43faffb
commit a9a8206
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/modules/core/src/alloy/proto/validation/ProtoIndexTraitValidator.java b/modules/core/src/alloy/proto/validation/ProtoIndexTraitValidator.java
@@ -67,7 +67,7 @@ private Optional<UnionShape> asProtoInlinedOneOf(Shape shape) {
 
 	private Stream<StructureShape> shapeUsingUnionMembers(Model model, UnionShape union) {
 		return model.getMemberShapes().stream().flatMap(m -> {
-			if (m.getTarget() == union.getId()) {
+			if (m.getTarget().equals(union.getId())) {
 				return OptionHelper.toStream(model.getShape(m.getContainer()).flatMap(Shape::asStructureShape));
 			} else {
 				return Stream.empty();

diff --git a/modules/core/src/alloy/validation/DataExamplesTraitValidator.java b/modules/core/src/alloy/validation/DataExamplesTraitValidator.java
@@ -33,7 +33,7 @@ public List<ValidationEvent> validate(Model model) {
 		for (Shape shape : model.getShapesWithTrait(DataExamplesTrait.class)) {
 			DataExamplesTrait trt = shape.getTrait(DataExamplesTrait.class).get();
 			for (DataExamplesTrait.DataExample example : trt.getExamples()) {
-				if (example.getExampleType() == DataExamplesTrait.DataExampleType.SMITHY) {
+				if (example.getExampleType().equals(DataExamplesTrait.DataExampleType.SMITHY)) {
 					NodeValidationVisitor visitor = createVisitor(example.getContent(), model, shape);
 					events.addAll(shape.accept(visitor));
 				}