diff options
1 files changed, 6 insertions, 1 deletions
| -rw-r--r-- | src/EventSystem/Event.cpp | 7 |
diff --git a/src/EventSystem/Event.cpp b/src/EventSystem/Event.cpp index 9744e2e..607f225 100644 --- a/src/EventSystem/Event.cpp +++ b/src/EventSystem/Event.cpp @@ -11,12 +11,17 @@ Event::Event(const Hall* hall, String name, DateTime time) { StatusCode Event::reserveTicket(const Ticket& ticket, const char* password, const char* note) { if (tickets.contain(ticket)) return E_TicketAlreadyBought; - if (reservations.contain(Reservation(ticket, "", ""))) return E_TicketAlreadyReserved; + if (reservations.contain(ticket)) return E_TicketAlreadyReserved; reservations.insert(Reservation(ticket, password, note)); return Success; } StatusCode Event::cancelReservation(const Ticket& ticket) { + unsigned ind = reservations.findIndex(ticket); + if (ind == reservations.get_count()) + return W_TicketWasNotReserved; + reservations.removeAt(ind); + return Success; } |