package jnpf.base.service;

import com.baomidou.dynamic.datasource.tx.TransactionContext;
import jnpf.base.ActionResult;
import jnpf.base.UserInfo;
import jnpf.exception.DataException;
import jnpf.util.TicketUtil;
import jnpf.util.UserProvider;
import org.junit.Test;
import java.util.Map;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

public class UploadEditNotificationsTest {
    @Test public void authorizedWatchOnlyWakesAfterCommitAndRejectsOtherUser() {
        var notifications = new UploadEditNotifications();
        UserInfo user = new UserInfo(); user.setTenantId("tenant"); user.setUserId("owner");
        try (var users = mockStatic(UserProvider.class); var tickets = mockStatic(TicketUtil.class)) {
            users.when(UserProvider::getUser).thenReturn(user);
            users.when(UserProvider::getLoginUserId).thenReturn("owner");
            tickets.when(() -> TicketUtil.createTicket(anyString(), eq(60L))).thenAnswer(call -> call.getArgument(0));
            tickets.when(() -> TicketUtil.parseTicket(anyString())).thenAnswer(call -> call.getArgument(0));
            String ticket = notifications.ticket(Map.of("draft", 0L));
            var waiting = notifications.watch(ticket);
            assertFalse(waiting.hasResult());
            TransactionContext.bind("test-transaction");
            try {
                notifications.changedAfterCommit("tenant", "draft");
                assertFalse(waiting.hasResult());
                TransactionContext.getSynchronizations().forEach(sync -> sync.afterCommit());
                assertTrue(waiting.hasResult());
                assertEquals(true, ((ActionResult<?>) waiting.getResult()).getData());
            } finally {
                TransactionContext.removeSynchronizations(); TransactionContext.remove();
            }
            // A missed event between state read and subscription is detected immediately.
            assertTrue(notifications.watch(ticket).hasResult());
            users.when(UserProvider::getLoginUserId).thenReturn("other");
            try { notifications.watch(ticket); fail("other user must not subscribe"); }
            catch (DataException expected) { assertTrue(expected.getMessage().contains("失效")); }
        }
    }
}