SimpleToDo/app/src/androidTest/java/apps/jizzu/simpletodo/TaskDAOTest.kt at 24fcf38bc3e1966df5d8c09ffa25640c3c96c51e · Jizzu/SimpleToDo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package apps.jizzu.simpletodo

import android.content.Context
import androidx.room.Room
import androidx.test.core.app.ApplicationProvider
import androidx.test.ext.junit.runners.AndroidJUnit4
import apps.jizzu.simpletodo.data.database.TaskDao
import apps.jizzu.simpletodo.data.database.TasksDatabase
import apps.jizzu.simpletodo.data.models.Task
import org.junit.After
import org.junit.Assert.*
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import java.io.IOException
import java.util.*

@RunWith(AndroidJUnit4::class)
class TaskDAOTest {

    private lateinit var taskDao: TaskDao
    private lateinit var tasksDatabase: TasksDatabase

    @Before
    fun createDb() {
        val context = ApplicationProvider.getApplicationContext<Context>()
        tasksDatabase = Room.inMemoryDatabaseBuilder(
                context, TasksDatabase::class.java
        ).build()
        taskDao = tasksDatabase.taskDAO()
    }

    @Test
    fun testSaveTask() {
        val task = Task(1, "", "", Date().time, 1, Date().time)
        taskDao.saveTask(task)

        val tasks = taskDao.getAllTasks()
        assertEquals(1, tasks.size)
    }

    @Test
    fun testGetAllTasks() {
        var task = Task(1, "", "", Date().time, 1, Date().time)
        taskDao.saveTask(task)

        var tasks = taskDao.getAllTasks()
        assertEquals(1, tasks.size)

        task = Task(2, "", "", Date().time, 2, Date().time, false)
        taskDao.saveTask(task)

        tasks = taskDao.getAllTasks()
        assertEquals(2, tasks.size)

        task = Task(3, "", "", Date().time, 2, Date().time, true)
        taskDao.saveTask(task)

        tasks = taskDao.getAllTasks()
        assertEquals(3, tasks.size)
    }

    @Test
    fun testGetAllOpenTasks() {
        var task = Task(1, "", "", Date().time, 1, Date().time)
        taskDao.saveTask(task)

        var tasks = taskDao.getAllOpenTasks()
        assertEquals(1, tasks.size)

        task = Task(2, "", "", Date().time, 2, Date().time, false)
        taskDao.saveTask(task)

        tasks = taskDao.getAllOpenTasks()
        assertEquals(2, tasks.size)

        task = Task(3, "", "", Date().time, 2, Date().time, true)
        taskDao.saveTask(task)

        tasks = taskDao.getAllOpenTasks()
        assertEquals(2, tasks.size)
    }

    @Test
    fun testUpdateTask() {
        var tasks = taskDao.getAllTasks()
        assertEquals(0, tasks.size)

        val task = Task(1, "", "", Date().time, 1, Date().time)
        taskDao.saveTask(task)

        tasks = taskDao.getAllOpenTasks()
        assertEquals("", tasks[0].title)

        tasks[0].title = "Test"
        taskDao.updateTask(tasks[0])

        tasks = taskDao.getAllTasks()
        assertEquals(1, tasks.size)
        assertEquals("Test", tasks[0].title)
    }

    @Test
    fun testUpdateStatusTask() {
        var tasks = taskDao.getAllTasks()
        assertEquals(0, tasks.size)

        val task = Task(1, "Test Me", "Task to Complete", Date().time, 1, Date().time)
        taskDao.saveTask(task)

        tasks = taskDao.getAllOpenTasks()
        assertFalse(tasks[0].taskStatus)

        tasks[0].taskStatus = tasks[0].taskStatus.not()
        taskDao.updateTask(tasks[0])

        tasks = taskDao.getAllOpenTasks()
        assertEquals(0, tasks.size)
    }

    @After
    @Throws(IOException::class)
    fun closeDb() {
        tasksDatabase.close()
    }
}