@isTest
private class NewDiagnosisPartControllerTest {

    static testMethod void testMethod1() {
        List<RecordType> rectIE = [select Id from RecordType where IsActive = true and SobjectType = 'Account' and Name = 'Customer IE'];
        List<RecordType> rectOpp = [select Id from RecordType where IsActive = true and SobjectType = 'Opportunity' and Name = 'SSBD'];
        Id pricebookId = Test.getStandardPricebookId();
        
        Pricebook2 pricebook = new Pricebook2(
            Name = 'IE',
            ProductSegment__c = 'IE',
            TradeType__c = 'Taxation',
            SalesChannel__c = 'direct',
            MachineParts__c = 'Machine',
            CurrencyIsoCode = 'CNY'
        );
        insert pricebook;
        
        Product2 product1 = new Product2();
        product1.Name = 'product1';
        product1.IsActive = true;
        product1.ProductCode = 'product1';
        product1.ProductStatus__c = '1';
        product1.NMPAStatus_one__c = 'Z1';
        product1.MaterialStatus_one_Start__c = Date.today().addDays(-22);
        product1.MaterialStatus_one_End__c = Date.today().addDays(22);

        insert product1;

        SWO__c swo = new SWO__c();
        swo.Name = 'Test';
        insert swo;

        PageReference page = new PageReference('/apex/NewDiagnosisPart?swoId=' + swo.Id );
        System.Test.setCurrentPage(page);
        NewDiagnosisPartController controller = new NewDiagnosisPartController();
        controller.init();
        controller.diagnosisPart.Product__c = product1.Id;
        controller.diagnosisPart.QUANTITY__c = 1;
        controller.deleteAction();
        controller.save();

        Diagnosis_Part__c diagnosisPart = new Diagnosis_Part__c();
        diagnosisPart.SWO__c = swo.Id;
        insert diagnosisPart;

        PageReference page1 = new PageReference('/apex/NewDiagnosisPart?id='+diagnosisPart.Id + '&sfdc.override=1');
        System.Test.setCurrentPage(page1);
        NewDiagnosisPartController controller1 = new NewDiagnosisPartController(new ApexPages.StandardController(diagnosisPart));
        controller1.init();
        controller1.deleteAction();
        controller1.ReturnSWO();
    }
}