Mengejek variabel anggota kelas menggunakan Mockito

Saya seorang pemula untuk pengembangan dan ke unit test pada khususnya. Saya kira kebutuhan saya cukup sederhana, tetapi saya ingin tahu pendapat orang lain tentang ini.

Misalkan saya memiliki dua kelas seperti itu -

public class First {

    Second second ;

    public First(){
        second = new Second();
    }

    public String doSecond(){
        return second.doSecond();
    }
}

class Second {

    public String doSecond(){
        return "Do Something";
    }
}

Katakanlah saya sedang menulis unit test untuk menguji First.doSecond()metode. Namun, misalkan, saya ingin Second.doSecond()kelas Mock seperti itu. Saya menggunakan Mockito untuk melakukan ini.

public void testFirst(){
    Second sec = mock(Second.class);
    when(sec.doSecond()).thenReturn("Stubbed Second");

    First first = new First();
    assertEquals("Stubbed Second", first.doSecond());
}

Saya melihat bahwa ejekan itu tidak berpengaruh dan pernyataan itu gagal. Apakah tidak ada cara untuk mengejek variabel anggota kelas yang ingin saya uji. ?

Anda perlu menyediakan cara untuk mengakses variabel anggota sehingga Anda dapat memberikan tiruan (cara yang paling umum adalah metode setter atau konstruktor yang mengambil parameter).

Jika kode Anda tidak memberikan cara untuk melakukan ini, itu salah faktor untuk TDD (Test Driven Development).

Ini tidak mungkin jika Anda tidak dapat mengubah kode Anda. Tapi saya suka injeksi ketergantungan dan Mockito mendukungnya:

public class First {    
    @Resource
    Second second;

    public First() {
        second = new Second();
    }

    public String doSecond() {
        return second.doSecond();
    }
}

Tes Anda:

@RunWith(MockitoJUnitRunner.class)
public class YourTest {
   @Mock
   Second second;

   @InjectMocks
   First first = new First();

   public void testFirst(){
      when(second.doSecond()).thenReturn("Stubbed Second");
      assertEquals("Stubbed Second", first.doSecond());
   }
}

Ini sangat bagus dan mudah.

Jika Anda melihat dengan cermat kode Anda, Anda akan melihat bahwa secondproperti dalam pengujian Anda masih merupakan contoh Second, bukan tiruan (Anda tidak meneruskan tiruan ke firstdalam kode Anda).

Cara paling sederhana adalah membuat setter untuk seconddi Firstkelas dan memberikannya mock secara eksplisit.

Seperti ini:

public class First {

Second second ;

public First(){
    second = new Second();
}

public String doSecond(){
    return second.doSecond();
}

    public void setSecond(Second second) {
    this.second = second;
    }


}

class Second {

public String doSecond(){
    return "Do Something";
}
}

....

public void testFirst(){
Second sec = mock(Second.class);
when(sec.doSecond()).thenReturn("Stubbed Second");


First first = new First();
first.setSecond(sec)
assertEquals("Stubbed Second", first.doSecond());
}

Lain akan lulus Secondcontoh sebagai Firstparameter konstruktor.

Jika Anda tidak dapat mengubah kode, saya pikir satu-satunya pilihan adalah menggunakan refleksi:

public void testFirst(){
    Second sec = mock(Second.class);
    when(sec.doSecond()).thenReturn("Stubbed Second");


    First first = new First();
    Field privateField = PrivateObject.class.
        getDeclaredField("second");

    privateField.setAccessible(true);

    privateField.set(first, sec);

    assertEquals("Stubbed Second", first.doSecond());
}

Tapi Anda mungkin bisa, karena jarang melakukan tes pada kode yang tidak Anda kontrol (walaupun orang dapat membayangkan skenario di mana Anda harus menguji perpustakaan eksternal karena penulisnya tidak :))

Jika Anda tidak dapat mengubah variabel anggota, maka cara lain untuk melakukannya adalah menggunakan powerMockit dan menelepon

Second second = mock(Second.class)
when(second.doSecond()).thenReturn("Stubbed Second");
whenNew(Second.class).withAnyArguments.thenReturn(second);

Sekarang masalahnya adalah bahwa panggilan APAPUN ke Second baru akan mengembalikan contoh yang sama mengejek. Tetapi dalam kasus sederhana Anda ini akan berhasil.

Saya memiliki masalah yang sama di mana nilai pribadi tidak ditetapkan karena Mockito tidak memanggil konstruktor super. Inilah cara saya menambah ejekan dengan refleksi.

Pertama, saya membuat kelas TestUtils yang berisi banyak utilitas bermanfaat termasuk metode refleksi ini. Akses refleksi agak sulit untuk diterapkan setiap kali. Saya membuat metode ini untuk menguji kode pada proyek yang, karena satu dan lain alasan, tidak memiliki paket mengejek dan saya tidak diundang untuk memasukkannya.

public class TestUtils {
    // get a static class value
    public static Object reflectValue(Class<?> classToReflect, String fieldNameValueToFetch) {
        try {
            Field reflectField  = reflectField(classToReflect, fieldNameValueToFetch);
            reflectField.setAccessible(true);
            Object reflectValue = reflectField.get(classToReflect);
            return reflectValue;
        } catch (Exception e) {
            fail("Failed to reflect "+fieldNameValueToFetch);
        }
        return null;
    }
    // get an instance value
    public static Object reflectValue(Object objToReflect, String fieldNameValueToFetch) {
        try {
            Field reflectField  = reflectField(objToReflect.getClass(), fieldNameValueToFetch);
            Object reflectValue = reflectField.get(objToReflect);
            return reflectValue;
        } catch (Exception e) {
            fail("Failed to reflect "+fieldNameValueToFetch);
        }
        return null;
    }
    // find a field in the class tree
    public static Field reflectField(Class<?> classToReflect, String fieldNameValueToFetch) {
        try {
            Field reflectField = null;
            Class<?> classForReflect = classToReflect;
            do {
                try {
                    reflectField = classForReflect.getDeclaredField(fieldNameValueToFetch);
                } catch (NoSuchFieldException e) {
                    classForReflect = classForReflect.getSuperclass();
                }
            } while (reflectField==null || classForReflect==null);
            reflectField.setAccessible(true);
            return reflectField;
        } catch (Exception e) {
            fail("Failed to reflect "+fieldNameValueToFetch +" from "+ classToReflect);
        }
        return null;
    }
    // set a value with no setter
    public static void refectSetValue(Object objToReflect, String fieldNameToSet, Object valueToSet) {
        try {
            Field reflectField  = reflectField(objToReflect.getClass(), fieldNameToSet);
            reflectField.set(objToReflect, valueToSet);
        } catch (Exception e) {
            fail("Failed to reflectively set "+ fieldNameToSet +"="+ valueToSet);
        }
    }

}

Maka saya bisa menguji kelas dengan variabel pribadi seperti ini. Ini berguna untuk mengejek jauh di dalam pohon kelas yang tidak dapat Anda kendalikan.

@Test
public void testWithRectiveMock() throws Exception {
    // mock the base class using Mockito
    ClassToMock mock = Mockito.mock(ClassToMock.class);
    TestUtils.refectSetValue(mock, "privateVariable", "newValue");
    // and this does not prevent normal mocking
    Mockito.when(mock.somthingElse()).thenReturn("anotherThing");
    // ... then do your asserts
}

Saya memodifikasi kode saya dari proyek saya yang sebenarnya di sini, di halaman. Mungkin ada satu atau dua masalah kompilasi. Saya pikir Anda mendapatkan ide umum. Jangan ragu untuk mengambil kode dan menggunakannya jika Anda merasa berguna.

Banyak orang lain telah menyarankan Anda untuk memikirkan kembali kode Anda untuk membuatnya lebih dapat diuji - saran yang bagus dan biasanya lebih sederhana daripada yang saya sarankan.

Jika Anda tidak dapat mengubah kode untuk membuatnya lebih dapat diuji, PowerMock: https://code.google.com/p/powermock/

PowerMock memperluas Mockito (jadi Anda tidak perlu mempelajari kerangka kerja mock baru), menyediakan fungsionalitas tambahan. Ini termasuk kemampuan untuk memiliki konstruktor mengembalikan tiruan. Kuat, tetapi sedikit rumit - jadi gunakan dengan bijak.

Anda menggunakan pelari Mock yang berbeda. Dan Anda perlu menyiapkan kelas yang akan memanggil konstruktor. (Perhatikan bahwa ini adalah gotcha yang umum - persiapkan kelas yang memanggil konstruktor, bukan kelas yang dibangun)

@RunWith(PowerMockRunner.class)
@PrepareForTest({First.class})

Kemudian dalam set-up pengujian Anda, Anda bisa menggunakan metode whenNew untuk membuat konstruktor mengembalikan tiruan

whenNew(Second.class).withAnyArguments().thenReturn(mock(Second.class));

Ya, ini bisa dilakukan, seperti yang ditunjukkan tes berikut (ditulis dengan JMockit mocking API, yang saya kembangkan):

@Test
public void testFirst(@Mocked final Second sec) {
    new NonStrictExpectations() {{ sec.doSecond(); result = "Stubbed Second"; }};

    First first = new First();
    assertEquals("Stubbed Second", first.doSecond());
}

Namun, dengan Mockito, ujian semacam itu tidak dapat ditulis. Ini karena cara mengejek diimplementasikan di Mockito, di mana subclass dari kelas yang akan diejek dibuat; hanya instance dari subclass "mock" ini yang dapat memiliki perilaku mengejek, jadi Anda perlu meminta kode yang diuji menggunakannya alih-alih instance lainnya.

Jika Anda menginginkan alternatif untuk ReflectionTestUtils dari Spring di mockito, gunakan

Whitebox.setInternalState(first, "second", sec);