trigger SharinPixCase on Case (after insert) {
	SharinPixCaseHandler handler = new SharinPixCaseHandler();
	if (Trigger.isAfter && Trigger.isInsert) {
		handler.uploadImageOnCase(Trigger.new);
	}
}

public with sharing class SharinPixCaseHandler  {  
    public void uploadImageOnCase(List<Case> lstNewCase) {
        // Collect SharinPixTemporaryID values from new Case records
        Set<String> tempIds = new Set<String>();
        List<Case> lstFilteredCase = new List<Case>();
        for (Case cas : lstNewCase) {
            if (cas.sharinPixTemporaryID__c != null) {
                tempIds.add(cas.sharinPixTemporaryID__c);
                lstFilteredCase.add(cas);
            }
        }

        if (tempIds.isEmpty()) { return; }
        
        // Query temporary records (Pending Album records)
        Map<String, List<String>> tempIdToImageIdsMap = new Map<String, List<String>>();  
        List<SharinPixPendingAlbum__c> pendingAlbums = [SELECT Id, ImageIds__c, TemporaryId__c FROM SharinPixPendingAlbum__c WHERE TemporaryId__c IN :tempIds];
        for (SharinPixPendingAlbum__c pendingAlbum : pendingAlbums) {
            // Deserialize JSON string from ImageIds__c into a List<String> to handle image IDs
            List<String> lstImageIds = (List<String>) JSON.deserialize(pendingAlbum.ImageIds__c, List<String>.class);
            if (tempIdToImageIdsMap.containsKey(pendingAlbum.TemporaryId__c)) {
                tempIdToImageIdsMap.get(pendingAlbum.TemporaryId__c).addAll(lstImageIds);
            } else {
                tempIdToImageIdsMap.put(pendingAlbum.TemporaryId__c, lstImageIds);
            }
        }
        
        for (Case cas : lstFilteredCase) {   
            if (tempIdToImageIdsMap.get(cas.sharinPixTemporaryID__c) != null){
                System.enqueueJob(new ImageMoveQueueable(tempIdToImageIdsMap.get(cas.sharinPixTemporaryID__c), cas.Id, cas.sharinPixTemporaryID__c ));
            }
        }
    }
}

@IsTest
public class SharinPixCaseHandlerTest {

    // method to create mock Case records
    private static List<Case> createMockCases(Integer count, Boolean withTempId) {
        List<Case> cases = new List<Case>();
        for (Integer i = 0; i < count; i++) {
            Case cas = new Case(Subject = 'Test Case ' + i);
            if (withTempId) {
                cas.sharinPixTemporaryID__c = 'TEMPID' + i;
            }
            cases.add(cas);
        }
        return cases;
    }

    // method to create mock SharinPixPendingAlbum__c records
    private static void createMockPendingAlbums(List<String> tempIds) {
        List<SharinPixPendingAlbum__c> albums = new List<SharinPixPendingAlbum__c>();
        for (String id : tempIds) {
            SharinPixPendingAlbum__c album = new SharinPixPendingAlbum__c(
                TemporaryId__c = id,
                ImageIds__c = JSON.serialize(new List<String>{'IMG1' + id, 'IMG2' + id})
            );
            albums.add(album);
        }
        insert albums;
    }

    @IsTest
    static void testCasesWithNoTemporaryId() {
        List<Case> cases = createMockCases(3, false);
        insert cases;
        SharinPixCaseHandler handler = new SharinPixCaseHandler();
        Test.startTest();
        handler.uploadImageOnCase(cases);
        Test.stopTest();
    }

    @IsTest
    static void testCasesWithTemporaryId() {
        List<Case> cases = createMockCases(2, true);
        insert cases;
        createMockPendingAlbums(new List<String>{'TEMPID0', 'TEMPID1'});

        SharinPixCaseHandler handler = new SharinPixCaseHandler();
        Test.startTest();
        handler.uploadImageOnCase(cases);
        Test.stopTest();
    }
    
    @IsTest
    static void testCasesWithMultipleSameTemporaryId() {
        List<Case> cases = createMockCases(2, true);
        insert cases;
        createMockPendingAlbums(new List<String>{'TEMPID0', 'TEMPID0'});

        SharinPixCaseHandler handler = new SharinPixCaseHandler();
        Test.startTest();
        handler.uploadImageOnCase(cases);
        Test.stopTest();
    }
}

public with sharing class ImageMoveQueueable implements Queueable, Database.AllowsCallouts {
    private List<String> imageIds;
    private Id caseId;
    private String TemporaryID;

    public ImageMoveQueueable(List<String> imageIds, Id caseId, String TemporaryID) {
        this.imageIds = imageIds;
        this.caseId = caseId;
        this.TemporaryID = TemporaryID;
    }

    public void execute(QueueableContext context) {
       	List<SharinPixPendingAlbum__c> pendingAlbumsToDelete = [SELECT Id FROM SharinPixPendingAlbum__c WHERE TemporaryId__c =: TemporaryID];
        delete pendingAlbumsToDelete;
        if(Test.isRunningTest()) { return; }
        sharinpix.Image.moveImages(imageIds, caseId);
    }
}

@IsTest
public class ImageMoveQueueableTest {

    @IsTest
    static void testImageMoveQueueableExecution() {
        // Setup test data
        Case testCase = new Case(Subject = 'Test Case');
        insert testCase;
        List<String> imageIds = new List<String>{'Image1', 'Image2', 'Image3'};
        String tempID = 'TEMP123';
        ImageMoveQueueable queueable = new ImageMoveQueueable(imageIds, testCase.Id, tempID);
        
        Test.startTest();
        Id jobId = System.enqueueJob(queueable);
        Test.stopTest();

        AsyncApexJob job = [SELECT Status, JobType FROM AsyncApexJob WHERE Id = :jobId];
        System.assertEquals('Completed', job.Status, 'Job has completed after execution.');
    }
}