так правильнее

src/main/java/com/labwork1/app/student/controller/SessionDto.java

@@ -1,6 +1,7 @@
 package com.labwork1.app.student.controller;
 
 import com.labwork1.app.student.model.Session;
+import com.labwork1.app.student.model.SessionExtension;
 
 import java.sql.Timestamp;
 
@@ -16,14 +17,25 @@ public class SessionDto {
     public SessionDto() {
     }
 
-    public SessionDto(Session session) {
+    public SessionDto(SessionExtension session) {
         this.id = session.getId();
         this.price = session.getPrice();
         this.timestamp = session.getTimestamp();
         this.capacity = session.getCapacity();
         this.maxCount = session.getMaxCount();
-        if (session.getCinema() != null) this.cinema = new CinemaDto(session.getCinema());
-        else this.cinema = null;
+        if (session.getCinema() != null) {
+            this.cinema = new CinemaDto(session.getCinema());
+        } else this.cinema = null;
+    }
+
+    public SessionDto(Session session) {
+        this.id = session.getId();
+        this.price = session.getPrice();
+        this.timestamp = session.getTimestamp();
+        this.maxCount = session.getMaxCount();
+        if (session.getCinema() != null) {
+            this.cinema = new CinemaDto(session.getCinema());
+        } else this.cinema = null;
     }
 
     public int getMaxCount() {

src/main/java/com/labwork1/app/student/model/Session.java

@@ -12,23 +12,21 @@ import java.util.Objects;
 public class Session {
     @Id
     @GeneratedValue(strategy = GenerationType.AUTO)
-    private Long id;
+    protected Long id;
     @NotNull(message = "price can't be null or empty")
-    private Double price;
+    protected Double price;
     @NotNull(message = "timestamp can't be null or empty")
     @Column
     @Temporal(TemporalType.TIMESTAMP)
-    private Timestamp timestamp;
+    protected Timestamp timestamp;
     @OneToMany(mappedBy = "session", fetch = FetchType.EAGER)
     private List<OrderSession> orders;
     @ManyToOne(fetch = FetchType.EAGER)
     @JoinColumn(name = "cinema_fk")
-    private Cinema cinema;
+    protected Cinema cinema;
     @NotNull(message = "maxCount can't be null or empty")
     @Column
-    private Integer maxCount;
-    @Transient
-    private Long capacity;
+    protected Integer maxCount;
 
     public Session() {
     }
@@ -43,14 +41,13 @@ public class Session {
         this.maxCount = maxCount;
     }
 
-    public Session(Session session, Long capacity) {
+    public Session(Session session) {
         this.id = session.getId();
         this.price = session.getPrice();
         this.timestamp = session.getTimestamp();
         this.orders = session.getOrders();
         this.cinema = session.getCinema();
         this.maxCount = session.getMaxCount();
-        this.capacity = capacity;
     }
 
     public Cinema getCinema() {
@@ -95,7 +92,6 @@ public class Session {
                 "id=" + id +
                 ", price='" + price + '\'' +
                 ", timestamp='" + timestamp.toString() + '\'' +
-                ", capacity='" + capacity + '\'' +
                 ", maxCount='" + maxCount.toString() + '\'' +
                 ", cinema='" + cinema.toString() + '\'' +
                 '}';
@@ -136,12 +132,4 @@ public class Session {
     public List<OrderSession> getOrders() {
         return orders;
     }
-
-    public Long getCapacity() {
-        return capacity;
-    }
-
-    public void setCapacity(Long capacity) {
-        this.capacity = capacity;
-    }
 }

src/main/java/com/labwork1/app/student/model/SessionExtension.java

@@ -0,0 +1,17 @@
+package com.labwork1.app.student.model;
+
+public class SessionExtension extends Session {
+    private Long capacity;
+    public SessionExtension(Session session, Long capacity) {
+        super(session);
+        this.capacity = capacity;
+    }
+
+    public Long getCapacity() {
+        return capacity;
+    }
+
+    public void setCapacity(Long capacity) {
+        this.capacity = capacity;
+    }
+}

src/main/java/com/labwork1/app/student/repository/SessionRepository.java

@@ -1,6 +1,7 @@
 package com.labwork1.app.student.repository;
 
 import com.labwork1.app.student.model.Session;
+import com.labwork1.app.student.model.SessionExtension;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.Query;
 import org.springframework.data.repository.query.Param;
@@ -11,13 +12,13 @@ import java.util.Optional;
 public interface SessionRepository extends JpaRepository<Session, Long> {
     @Query("Select sum(os.count) from OrderSession os where os.session.id = :sessionId")
     Optional<Integer> getCapacity(@Param("sessionId") Long sessionId);
-    @Query("Select new com.labwork1.app.student.model.Session(s, sum(os.count)) from Session s " +
+    @Query("Select new com.labwork1.app.student.model.SessionExtension(s, sum(os.count)) from Session s " +
             "left join OrderSession os on s.id = os.session.id " +
             "group by s")
-    List<Session> getSessionsWithCapacity();
-    @Query("Select new com.labwork1.app.student.model.Session(s, sum(os.count)) from Session s " +
+    List<SessionExtension> getSessionsWithCapacity();
+    @Query("Select new com.labwork1.app.student.model.SessionExtension(s, sum(os.count)) from Session s " +
             "left join OrderSession os on s.id = os.session.id " +
             "where s.id = :sessionId " +
             "group by s")
-    Optional<Session> getSessionWithCapacity(@Param("sessionId") Long sessionId);
+    Optional<SessionExtension> getSessionWithCapacity(@Param("sessionId") Long sessionId);
 }

src/main/java/com/labwork1/app/student/service/SessionService.java

@@ -2,6 +2,7 @@ package com.labwork1.app.student.service;
 
 import com.labwork1.app.student.model.Cinema;
 import com.labwork1.app.student.model.Session;
+import com.labwork1.app.student.model.SessionExtension;
 import com.labwork1.app.student.repository.SessionRepository;
 import com.labwork1.app.util.validation.ValidatorUtil;
 import org.springframework.stereotype.Service;
@@ -32,13 +33,13 @@ public class SessionService {
     }
 
     @Transactional(readOnly = true)
-    public Session findSession(Long id) {
+    public SessionExtension findSession(Long id) {
         return sessionRepository.getSessionWithCapacity(id)
                 .orElseThrow(() -> new SessionNotFoundException(id));
     }
 
     @Transactional(readOnly = true)
-    public List<Session> findAllSessions() {
+    public List<SessionExtension> findAllSessions() {
         return sessionRepository.getSessionsWithCapacity();
     }
 

src/test/java/com/labwork1/app/JpaCustomerTests.java

@@ -67,7 +67,7 @@ public class JpaCustomerTests {
         // у заказа 2 сеанса
         orderService.addSession(order2.getId(), session1.getId(), 2);
 
-        List<Session> result = sessionService.findAllSessions();
+        List<SessionExtension> result = sessionService.findAllSessions();
 
         Assertions.assertEquals(sessionService.getCapacity(session1.getId()), 2);