from datetime import date, timedelta
from room_cal.reservations import Reservation


async def test_add_reservation_round_trip():
  res = Reservation()
  checkin = date.today() - timedelta(days=30)
  checkout = checkin + timedelta(days=1)

  data = {
    'guest_name':      'Bob Test',
    'service':        'Test Service',
    'reservation_id': 'A1-B2-C3',
    'checkin_date':   checkin,
    'checkout_date':  checkout,
  }

  try:
    await res.set(data)

    stored = await res.get('A1-B2-C3', checkin, checkout)
    assert stored is not None, "Reservation not found after set"
    assert stored.guest_name == 'Bob Test'
    assert stored.reservation_id == 'A1-B2-C3'
  finally:
    event = await res._find('A1-B2-C3', checkin, checkout)
    if event:
      await res.delete(event)