Mockito, 이대로 괜찮은가?

@Service
public class TableGroupService {
    @Autowired
    private OrderTableDao orderTableDao;
    @Autowired
    private TableGroupDao tableGroupDao;

    @Transactional
    public TableGroup create(final TableGroup request) {
        List<Long> orderTables = request.getOrderTables()
            .stream()
            .map(OrderTable::getId)
            .collect(Collectors.toList());
    
        if (CollectionUtils.isEmpty(orderTableIds) || orderTableIds.size() < 2) {
            throw new IllegalArgumentException();
        }
    
        List<OrderTable> savedOrderTables = orderTableDao.findAllByIdIn(orderTableIds);
    
        if (orderTableIds.size() != savedOrderTables.size()) {
            throw new IllegalArgumentException();
        }
    
        TableGroup savedTableGroup = tableGroupDao.save(request);
    
        final Long tableGroupId = savedTableGroup.getId();
        for (OrderTable savedOrderTable : savedOrderTables) {
            savedOrderTable.setTableGroupId(tableGroupId);
            savedOrderTable.setEmpty(false);
            orderTableDao.save(savedOrderTable);
        }
    
        return savedTableGroup;
    }
}

@ExtendWith(MockitoExtension.class)
class TableGroupServiceTest {
    @InjectMocks
    private TableGroupService tableGroupService;
    @Mock
    private OrderTableDao tableDao;
    @Mock
    private TableGroupDao tableGroupDao;
    
    @DisplayName("테이블 그룹 추가")
    @Test
    void create() {
        List<OrderTable> orderTables = Arrays.asList(TABLE1, TABLE2);
        TableGroup request = TableGroup.builder()
            .orderTables(orderTables)
            .build();

        List<Long> tableIds = orderTables.stream()
            .map(OrderTable::getId)
            .collect(Collectors.toList());
        given(tableDao.findAllByIdIn(tableIds)).willReturn(orderTables);
        given(tableGroupDao.save(request)).willReturn(TABLE_GROUP);
    
        TableGroup createdTableGroup = tableGroupService.create(request);
    
        assertAll(
            () -> assertThat(createdTableGroup.getOrderTables().get(0).getTableGroupId()).isEqualTo(TABLE_GROUP.getId()),
            () -> assertThat(createdTableGroup.getOrderTables().get(1).getTableGroupId()).isEqualTo(TABLE_GROUP.getId()),
            () -> assertThat(createdTableGroup.getOrderTables().get(0).isEmpty()).isFalse(),
            () -> assertThat(createdTableGroup.getOrderTables().get(1).isEmpty()).isFalse()
        );
    }
}