Add tests for BaseViewModel: failure handling.

Author: android10

app/src/test/kotlin/com/fernandocejas/sample/framework/platform/BaseViewModelTest.kt

@@ -0,0 +1,27 @@
+package com.fernandocejas.sample.framework.platform
+
+import android.arch.lifecycle.MutableLiveData
+import com.fernandocejas.sample.AndroidTest
+import com.fernandocejas.sample.framework.exception.Failure
+import com.fernandocejas.sample.framework.exception.Failure.NetworkConnection
+import org.amshove.kluent.shouldBeInstanceOf
+import org.junit.Test
+
+class BaseViewModelTest : AndroidTest() {
+
+ @Test fun `should handle failure by updating live data`() {
+ val viewModel = MyViewModel()
+
+ viewModel.handleError(NetworkConnection())
+
+ val failure = viewModel.failure
+ val error = viewModel.failure.value
+
+ failure shouldBeInstanceOf MutableLiveData::class.java
+ error shouldBeInstanceOf NetworkConnection::class.java
+ }
+
+ private class MyViewModel : BaseViewModel() {
+ fun handleError(failure: Failure) = handleFailure(failure)
+ }
+}