Mockito: วิธีการตรวจสอบวิธีการถูกเรียกบนวัตถุที่สร้างขึ้นภายในวิธี?

ฉันใหม่กับ Mockito

จากคลาสด้านล่างฉันจะใช้ Mockito เพื่อตรวจสอบว่าsomeMethodถูกเรียกใช้หลังจากfooถูกเรียกได้อย่างไร

public class Foo
{
    public void foo(){
        Bar bar = new Bar();
        bar.someMethod();
    }
}

ฉันต้องการโทรยืนยันดังต่อไปนี้

verify(bar, times(1)).someMethod();

ที่เป็นเช่นล้อเลียนของbarBar

ฉีดพึ่งพา

หากคุณฉีดอินสแตนซ์บาร์หรือโรงงานที่ใช้สำหรับสร้างอินสแตนซ์บาร์ (หรือหนึ่งใน 483 วิธีการอื่นในการทำเช่นนี้) คุณจะต้องมีการเข้าถึงที่จำเป็นเพื่อทำการทดสอบ

ตัวอย่างโรงงาน:

ให้คลาส Foo เขียนดังนี้:

public class Foo {
  private BarFactory barFactory;

  public Foo(BarFactory factory) {
    this.barFactory = factory;
  }

  public void foo() {
    Bar bar = this.barFactory.createBar();
    bar.someMethod();
  }
}

ในวิธีการทดสอบของคุณคุณสามารถฉีด BarFactory เช่นนี้:

@Test
public void testDoFoo() {
  Bar bar = mock(Bar.class);
  BarFactory myFactory = new BarFactory() {
    public Bar createBar() { return bar;}
  };

  Foo foo = new Foo(myFactory);
  foo.foo();

  verify(bar, times(1)).someMethod();
}

โบนัส: นี่เป็นตัวอย่างของวิธีที่ TDD สามารถผลักดันการออกแบบรหัสของคุณ

คำตอบแบบคลาสสิกคือ "คุณทำไม่ได้" คุณทดสอบ API สาธารณะFooไม่ใช่ภายใน

มีพฤติกรรมของFooวัตถุ (หรือน้อยกว่าที่ดีบางอย่างวัตถุอื่น ๆ ในสภาพแวดล้อม) ที่ได้รับผลกระทบจากfoo()? ถ้าเป็นเช่นนั้นทดสอบว่า และถ้าไม่วิธีการทำอะไร?

หากคุณไม่ต้องการใช้ DI หรือโรงงาน คุณสามารถปรับโครงสร้างห้องเรียนของคุณในแบบที่ยุ่งยากเล็กน้อย:

public class Foo {
    private Bar bar;

    public void foo(Bar bar){
        this.bar = (bar != null) ? bar : new Bar();
        bar.someMethod();
        this.bar = null;  // for simulating local scope
    }
}

และชั้นทดสอบของคุณ:

@RunWith(MockitoJUnitRunner.class)
public class FooTest {
    @Mock Bar barMock;
    Foo foo;

    @Test
    public void testFoo() {
       foo = new Foo();
       foo.foo(barMock);
       verify(barMock, times(1)).someMethod();
    }
}

คลาสที่เรียกเมธอด foo ของคุณจะทำดังนี้:

public class thirdClass {

   public void someOtherMethod() {
      Foo myFoo = new Foo();
      myFoo.foo(null);
   }
}

อย่างที่คุณเห็นเมื่อเรียกใช้เมธอดด้วยวิธีนี้คุณไม่จำเป็นต้องนำเข้าคลาสบาร์ในคลาสอื่นที่เรียกเมธอด foo ซึ่งอาจเป็นสิ่งที่คุณต้องการ

แน่นอนว่าข้อเสียคือคุณอนุญาตให้ผู้โทรตั้งค่า Bar Object

หวังว่ามันจะช่วย

โซลูชันสำหรับโค้ดตัวอย่างของคุณโดยใช้ PowerMockito.whenNew

• mockito-all 1.10.8
• powermock-core 1.6.1
• powermock-module-junit4 1.6.1
• powermock-api-mockito 1.6.1
• มิถุนายน 4.12

FooTest.java

package foo;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

//Both @PrepareForTest and @RunWith are needed for `whenNew` to work 
@RunWith(PowerMockRunner.class)
@PrepareForTest({ Foo.class })
public class FooTest {

    // Class Under Test
    Foo cut;

    @Mock
    Bar barMock;

    @Before
    public void setUp() throws Exception {
        cut = new Foo();

    }

    @After
    public void tearDown() {
        cut = null;

    }

    @Test
    public void testFoo() throws Exception {

        // Setup
        PowerMockito.whenNew(Bar.class).withNoArguments()
                .thenReturn(this.barMock);

        // Test
        cut.foo();

        // Validations
        Mockito.verify(this.barMock, Mockito.times(1)).someMethod();

    }

}

เอาต์พุต JUnit

ฉันคิดว่า Mockito @InjectMocksเป็นวิธีที่จะไป

ขึ้นอยู่กับความตั้งใจของคุณคุณสามารถใช้:

1. ตัวสร้างการฉีด
2. หัวฉีดคุณสมบัติการฉีด
3. การฉีดภาคสนาม

ข้อมูลเพิ่มเติมในเอกสาร

ด้านล่างเป็นตัวอย่างของการฉีดภาคสนาม:

ชั้นเรียน:

public class Foo
{
    private Bar bar = new Bar();

    public void foo() 
    {
        bar.someMethod();
    }
}

public class Bar
{
    public void someMethod()
    {
         //something
    }
}

ทดสอบ:

@RunWith(MockitoJUnitRunner.class)
public class FooTest
{
    @Mock
    Bar bar;

    @InjectMocks
    Foo foo;

    @Test
    public void FooTest()
    {
        doNothing().when( bar ).someMethod();
        foo.foo();
        verify(bar, times(1)).someMethod();
    }
}

ใช่ถ้าคุณต้องการ / จำเป็นต้องทำจริง ๆ คุณสามารถใช้ PowerMock นี่ควรจะเป็นทางเลือกสุดท้าย ด้วย PowerMock คุณสามารถทำให้การคืนค่าจำลองจากการเรียกไปยังตัวสร้าง จากนั้นทำการตรวจสอบบนจำลอง ที่กล่าวว่า csturtz คือคำตอบที่ "ถูกต้อง"

นี่คือลิงค์ไปสู่การสร้างจำลองของวัตถุใหม่

อีกวิธีง่ายๆที่จะเพิ่มคำสั่งบันทึกบางอย่างเพื่อ bar.someMethod () แล้วตรวจสอบให้แน่ใจคุณสามารถเห็นข้อความดังกล่าวเมื่อการทดสอบของคุณดำเนินการดูตัวอย่างที่นี่: วิธีการทำ JUnit ยืนยันข้อความในบันทึก

ที่เป็นประโยชน์โดยเฉพาะอย่างยิ่งเมื่อ Bar.someMethod () privateของคุณคือ