Skip to content

Commit

Permalink
Add bulk update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
vladmihalcea committed Dec 31, 2022
1 parent 92eddb1 commit ecf89dc
Show file tree
Hide file tree
Showing 7 changed files with 742 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
package io.hypersistence.utils.hibernate.type.json.generic;

import io.hypersistence.utils.hibernate.type.json.JsonType;
import io.hypersistence.utils.hibernate.type.model.BaseEntity;
import io.hypersistence.utils.hibernate.type.model.Location;
import io.hypersistence.utils.hibernate.type.model.Ticket;
import io.hypersistence.utils.hibernate.util.AbstractMySQLIntegrationTest;
import io.hypersistence.utils.jdbc.validator.SQLStatementCountValidator;
import org.hibernate.annotations.Type;
import org.hibernate.query.NativeQuery;
import org.junit.Test;

import javax.persistence.Column;
Expand Down Expand Up @@ -71,6 +73,27 @@ public void test() {
SQLStatementCountValidator.assertUpdateCount(0);
}

@Test
public void testBulkUpdate() {
doInJPA(entityManager -> {
Location location = new Location();
location.setCountry("Romania");
location.setCity("Sibiu");

entityManager.createNativeQuery(
"update Event " +
"set location = :location " +
"where id = :id")
.setParameter("id", _event.getId())
.unwrap(NativeQuery.class)
.setParameter("location", location, new JsonType(Location.class))
.executeUpdate();

Event event = entityManager.find(Event.class, _event.getId());
assertEquals("Sibiu", event.getLocation().getCity());
});
}

@Entity(name = "Event")
@Table(name = "event")
public static class Event extends BaseEntity {
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,157 @@
package io.hypersistence.utils.hibernate.type.json.generic;

import io.hypersistence.utils.hibernate.type.json.JsonType;
import io.hypersistence.utils.hibernate.type.model.BaseEntity;
import io.hypersistence.utils.hibernate.type.model.Location;
import io.hypersistence.utils.hibernate.type.model.Ticket;
import io.hypersistence.utils.hibernate.util.AbstractPostgreSQLIntegrationTest;
import io.hypersistence.utils.jdbc.validator.SQLStatementCountValidator;
import org.hibernate.annotations.Type;
import org.hibernate.query.NativeQuery;
import org.junit.Test;

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.ManyToOne;
import javax.persistence.Table;

import static org.junit.Assert.assertEquals;

/**
* @author Vlad Mihalcea
*/
public class GenericPostgreSQLJsonTypeTest extends AbstractPostgreSQLIntegrationTest {

@Override
protected Class<?>[] entities() {
return new Class<?>[]{
Event.class,
Participant.class
};
}

@Override
protected String[] packages() {
return new String[]{
Location.class.getPackage().getName()
};
}

private Event _event;

private Participant _participant;

@Override
protected void afterInit() {

doInJPA(entityManager -> {
Event nullEvent = new Event();
nullEvent.setId(0L);
entityManager.persist(nullEvent);

Location location = new Location();
location.setCountry("Romania");
location.setCity("Cluj-Napoca");

Event event = new Event();
event.setId(1L);
event.setLocation(location);
entityManager.persist(event);

Ticket ticket = new Ticket();
ticket.setPrice(12.34d);
ticket.setRegistrationCode("ABC123");

Participant participant = new Participant();
participant.setId(1L);
participant.setTicket(ticket);
participant.setEvent(event);

entityManager.persist(participant);

_event = event;
_participant = participant;
});
}

@Test
public void testLoad() {
SQLStatementCountValidator.reset();

doInJPA(entityManager -> {
Event event = entityManager.find(Event.class, _event.getId());
assertEquals("Romania", event.getLocation().getCountry());
assertEquals("Cluj-Napoca", event.getLocation().getCity());
});

SQLStatementCountValidator.assertTotalCount(1);
SQLStatementCountValidator.assertSelectCount(1);
SQLStatementCountValidator.assertUpdateCount(0);
}

@Test
public void testBulkUpdate() {
doInJPA(entityManager -> {
Location location = new Location();
location.setCountry("Romania");
location.setCity("Sibiu");

entityManager.createNativeQuery(
"update Event " +
"set location = :location " +
"where id = :id")
.setParameter("id", _event.getId())
.unwrap(NativeQuery.class)
.setParameter("location", location, new JsonType(Location.class))
.executeUpdate();

Event event = entityManager.find(Event.class, _event.getId());
assertEquals("Sibiu", event.getLocation().getCity());
});
}

@Entity(name = "Event")
@Table(name = "event")
public static class Event extends BaseEntity {

@Type(type = "io.hypersistence.utils.hibernate.type.json.JsonType")
@Column(columnDefinition = "jsonb")
private Location location;

public Location getLocation() {
return location;
}

public void setLocation(Location location) {
this.location = location;
}
}

@Entity(name = "Participant")
@Table(name = "participant")
public static class Participant extends BaseEntity {

@Type(type = "io.hypersistence.utils.hibernate.type.json.JsonType")
@Column(columnDefinition = "jsonb")
private Ticket ticket;

@ManyToOne
private Event event;

public Ticket getTicket() {
return ticket;
}

public void setTicket(Ticket ticket) {
this.ticket = ticket;
}

public Event getEvent() {
return event;
}

public void setEvent(Event event) {
this.event = event;
}
}
}
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
package io.hypersistence.utils.hibernate.type.json.generic;

import io.hypersistence.utils.hibernate.type.json.JsonType;
import io.hypersistence.utils.hibernate.type.model.BaseEntity;
import io.hypersistence.utils.hibernate.type.model.Location;
import io.hypersistence.utils.hibernate.type.model.Ticket;
import io.hypersistence.utils.hibernate.util.AbstractMySQLIntegrationTest;
import io.hypersistence.utils.jdbc.validator.SQLStatementCountValidator;
import org.hibernate.annotations.Type;
import org.hibernate.query.NativeQuery;
import org.junit.Test;

import javax.persistence.Column;
Expand Down Expand Up @@ -111,6 +113,27 @@ public void test() {
});
}

@Test
public void testBulkUpdate() {
doInJPA(entityManager -> {
Location location = new Location();
location.setCountry("Romania");
location.setCity("Sibiu");

entityManager.createNativeQuery(
"update Event " +
"set location = :location " +
"where id = :id")
.setParameter("id", _event.getId())
.unwrap(NativeQuery.class)
.setParameter("location", location, new JsonType(Location.class))
.executeUpdate();

Event event = entityManager.find(Event.class, _event.getId());
assertEquals("Sibiu", event.getLocation().getCity());
});
}

@Entity(name = "Event")
@Table(name = "event")
public static class Event extends BaseEntity {
Expand Down
Loading

0 comments on commit ecf89dc

Please sign in to comment.