4
4
5
5
package mozilla.components.service.fretboard
6
6
7
+ import org.junit.Assert.assertEquals
8
+ import org.junit.Assert.assertTrue
7
9
import org.junit.Test
8
10
import org.junit.runner.RunWith
9
11
import org.mockito.Mockito.`when`
@@ -58,4 +60,34 @@ class FretboardTest {
58
60
verify(experimentSource).getExperiments(ExperimentsSnapshot (listOf (Experiment (" id0" )), null ))
59
61
verify(experimentStorage).save(ExperimentsSnapshot (listOf (Experiment (" id" )), null ))
60
62
}
63
+
64
+ @Test
65
+ fun testExperiments () {
66
+ val experimentSource = mock(ExperimentSource ::class .java)
67
+ val experimentStorage = mock(ExperimentStorage ::class .java)
68
+ val experiments = listOf (
69
+ Experiment (" first-id" ),
70
+ Experiment (" second-id" )
71
+ )
72
+ `when `(experimentStorage.retrieve()).thenReturn(ExperimentsSnapshot (experiments, null ))
73
+ val fretboard = Fretboard (experimentSource, experimentStorage)
74
+ var returnedExperiments = fretboard.experiments
75
+ assertEquals(0 , returnedExperiments.size)
76
+ fretboard.loadExperiments()
77
+ returnedExperiments = fretboard.experiments
78
+ assertEquals(2 , returnedExperiments.size)
79
+ assertTrue(returnedExperiments.contains(experiments[0 ]))
80
+ assertTrue(returnedExperiments.contains(experiments[1 ]))
81
+ }
82
+
83
+ @Test
84
+ fun testExperimentsNoExperiments () {
85
+ val experimentSource = mock(ExperimentSource ::class .java)
86
+ val experimentStorage = mock(ExperimentStorage ::class .java)
87
+ val experiments = listOf<Experiment >()
88
+ `when `(experimentStorage.retrieve()).thenReturn(ExperimentsSnapshot (experiments, null ))
89
+ val fretboard = Fretboard (experimentSource, experimentStorage)
90
+ val returnedExperiments = fretboard.experiments
91
+ assertEquals(0 , returnedExperiments.size)
92
+ }
61
93
}
0 commit comments