package jnpf.base.util;
import jnpf.base.model.uploadedit.UploadEditDraft;
import jnpf.exception.DataException;
import jnpf.util.JsonUtil;
import jnpf.util.TicketUtil;
import org.junit.Test;
import java.util.*;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
public class UploadTemplateSourceTest {
    Map<String,Object> file(String id, String template) {
        Map<String,Object> file = new LinkedHashMap<>(); file.put("fileId", id); file.put("name", "same.docx");
        if (template != null) file.put(UploadTemplateSource.FIELD, template); return file;
    }
    @Test public void editingKeepsSourceDeletingReuploadingClearsAndDiscardRestores() {
        UploadEditDraft draft = new UploadEditDraft(); draft.setRevision(0L);
        draft.setBaselineJson(JsonUtil.getObjectToString(List.of(file("A", "templateA")))); draft.setCurrentJson(draft.getBaselineJson());
        UploadEditStateMachine.begin(draft, 0L, "A"); String lease = draft.getLeaseId();
        assertTrue(UploadEditStateMachine.save(draft, lease, "save1", file("edited", "forged")));
        assertEquals("templateA", UploadEditStateMachine.files(draft.getCurrentJson()).get(0).get(UploadTemplateSource.FIELD));
        UploadEditStateMachine.replace(draft, draft.getRevision(), List.of());
        assertFalse(UploadEditStateMachine.save(draft, lease, "late", file("late", "templateA")));
        var upload = file("new", "forged");
        UploadTemplateSource.normalize(List.of(upload), List.of(file("A", "templateA")), "tenant", "owner");
        assertFalse(upload.containsKey(UploadTemplateSource.FIELD));
        UploadEditStateMachine.replace(draft, draft.getRevision(), List.of(upload));
        UploadEditStateMachine.discard(draft, draft.getRevision());
        assertEquals("templateA", UploadEditStateMachine.files(draft.getCurrentJson()).get(0).get(UploadTemplateSource.FIELD));
    }
    @Test public void newVerifiedReceiptOverridesSameFileIdAndRejectsForgedReceipt() {
        var submitted = file("A", "forged"); submitted.put("uploadReceipt", "new");
        Map<String,Object> proof = file("A", "templateB"); proof.put("purpose", "upload-edit-source");
        proof.put("tenantId", "tenant"); proof.put("ownerId", "owner");
        try (var tickets = mockStatic(TicketUtil.class)) {
            tickets.when(() -> TicketUtil.parseTicket("new")).thenReturn(JsonUtil.getObjectToString(proof));
            UploadTemplateSource.normalize(List.of(submitted), List.of(file("A", "templateA")), "tenant", "owner");
            assertEquals("templateB", submitted.get(UploadTemplateSource.FIELD));
            try { UploadTemplateSource.normalize(List.of(submitted), List.of(), "other", "owner"); fail("tenant mismatch must fail"); } catch (DataException expected) { }
        }
    }
    @Test public void serverTimeIsPreservedButClientTimeIsNotTrusted() {
        var trusted = file("A", "templateA"); trusted.put("contentSavedAt", 100L);
        var submitted = file("A", "forged"); submitted.put("contentSavedAt", 999L);
        UploadTemplateSource.normalize(List.of(submitted), List.of(trusted), "tenant", "owner");
        assertEquals(100L, submitted.get("contentSavedAt"));
        var ordinary = file("B", "forged"); ordinary.put("contentSavedAt", 999L);
        UploadTemplateSource.normalize(List.of(ordinary), List.of(trusted), "tenant", "owner");
        assertFalse(ordinary.containsKey("contentSavedAt"));
        assertFalse(ordinary.containsKey(UploadTemplateSource.FIELD));
    }
    @Test public void savedCopyKeepsTemplateButUsesNewServerTimeAndMetadataToken() {
        var original = file("A", "templateA"); original.put("contentSavedAt", 100L);
        UploadEditDraft draft = new UploadEditDraft(); draft.setRevision(0L);
        draft.setBaselineJson(JsonUtil.getObjectToString(List.of(original))); draft.setCurrentJson(draft.getBaselineJson());
        UploadEditStateMachine.begin(draft, 0L, "A");
        var saved = file("B", "forged"); saved.put("contentSavedAt", 200L);
        assertTrue(UploadEditStateMachine.save(draft, draft.getLeaseId(), "saved", saved));
        var actual = UploadEditStateMachine.files(draft.getCurrentJson()).get(0);
        assertEquals("templateA", actual.get(UploadTemplateSource.FIELD));
        assertEquals(200L, ((Number) actual.get("contentSavedAt")).longValue());
        var otherTemplate = file("A", "templateB"); otherTemplate.put("contentSavedAt", 100L);
        assertEquals(UploadEditStateMachine.contentToken(List.of(original)), UploadEditStateMachine.contentToken(List.of(otherTemplate)));
        assertNotEquals(UploadEditStateMachine.metadataToken(List.of(original)), UploadEditStateMachine.metadataToken(List.of(otherTemplate)));
    }
}