[hibernate#1979] Test for EmbeddedId

Author: DavideD

hibernate-reactive-core/src/test/java/org/hibernate/reactive/EmbeddedIdTest.java

@@ -0,0 +1,169 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive;
+
+import java.util.Collection;
+import java.util.Objects;
+import java.util.Set;
+
+import org.junit.jupiter.api.Test;
+
+import io.vertx.junit5.Timeout;
+import io.vertx.junit5.VertxTestContext;
+import jakarta.persistence.Column;
+import jakarta.persistence.Embeddable;
+import jakarta.persistence.EmbeddedId;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Table;
+
+import static java.util.concurrent.TimeUnit.MINUTES;
+import static org.assertj.core.api.Assertions.assertThat;
+
+@Timeout(value = 10, timeUnit = MINUTES)
+public class EmbeddedIdTest extends BaseReactiveTest {
+
+	@Override
+	protected Collection<Class<?>> annotatedEntities() {
+		return Set.of( Shipment.class );
+	}
+
+	@Test
+	public void test(VertxTestContext context) {
+		ShipmentId id1 = new ShipmentId( "a", "b" );
+		Shipment shipment1 = new Shipment( id1, "test" );
+
+		ShipmentId id2 = new ShipmentId( "c", "d" );
+		Shipment shipment2 = new Shipment( id2, "test2" );
+
+		test( context, getMutinySessionFactory()
+				.withTransaction( s -> s.persistAll( shipment1, shipment2 ) )
+				.chain( () -> getMutinySessionFactory().withTransaction( s -> s.find( Shipment.class, id1 ) ) )
+				.invoke( result -> assertThat( result ).isEqualTo( shipment1 ) )
+		);
+	}
+
+	@Entity(name = "Shipment")
+	@Table(name = "Shipment")
+	public static class Shipment {
+
+		@EmbeddedId
+		private ShipmentId shipmentId;
+
+		@Column(name = "field")
+		private String field;
+
+		public Shipment() {
+		}
+
+		public Shipment(ShipmentId shipmentId, String field) {
+			this.shipmentId = shipmentId;
+			this.field = field;
+		}
+
+		public ShipmentId getShipmentId() {
+			return shipmentId;
+		}
+
+		public void setShipmentId(ShipmentId shipmentId) {
+			this.shipmentId = shipmentId;
+		}
+
+		public String getField() {
+			return field;
+		}
+
+		public void setField(String field) {
+			this.field = field;
+		}
+
+		@Override
+		public String toString() {
+			return "Shipment{" +
+					"shipmentId=" + shipmentId +
+					", field='" + field + '\'' +
+					'}';
+		}
+
+		@Override
+		public boolean equals(Object o) {
+			if ( this == o ) {
+				return true;
+			}
+			if ( o == null || getClass() != o.getClass() ) {
+				return false;
+			}
+			Shipment table = (Shipment) o;
+			return Objects.equals( shipmentId, table.shipmentId ) &&
+					Objects.equals( field, table.field );
+		}
+
+		@Override
+		public int hashCode() {
+			return Objects.hash( shipmentId, field );
+		}
+	}
+
+
+	@Embeddable
+	public static class ShipmentId {
+
+		@Column(name = "sp_country")
+		private String country;
+
+		@Column(name = "sp_city")
+		private String city;
+
+		public ShipmentId(String country, String city) {
+			this.country = country;
+			this.city = city;
+		}
+
+		public ShipmentId() {
+		}
+
+		public String getCountry() {
+			return country;
+		}
+
+		public String getCity() {
+			return city;
+		}
+
+		public void setCountry(String country) {
+			this.country = country;
+		}
+
+		public void setCity(String city) {
+			this.city = city;
+		}
+
+		@Override
+		public String toString() {
+			return "TableId{" +
+					"p='" + country + '\'' +
+					", t='" + city + '\'' +
+					'}';
+		}
+
+		@Override
+		public boolean equals(Object o) {
+			if ( this == o ) {
+				return true;
+			}
+			if ( o == null || getClass() != o.getClass() ) {
+				return false;
+			}
+			ShipmentId tableId = (ShipmentId) o;
+			return Objects.equals( country, tableId.country ) &&
+					Objects.equals( city, tableId.city );
+		}
+
+		@Override
+		public int hashCode() {
+			return Objects.hash( country, city );
+		}
+	}
+}