mirror of https://github.com/iSoron/uhabits.git
parent
b674d14b49
commit
dc0b8deccf
@ -0,0 +1,81 @@
|
||||
/*
|
||||
* Copyright (C) 2016-2020 Álinson Santos Xavier <isoron@gmail.com>
|
||||
*
|
||||
* This file is part of Loop Habit Tracker.
|
||||
*
|
||||
* Loop Habit Tracker is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by the
|
||||
* Free Software Foundation, either version 3 of the License, or (at your
|
||||
* option) any later version.
|
||||
*
|
||||
* Loop Habit Tracker is distributed in the hope that it will be useful, but
|
||||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||||
* more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License along
|
||||
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package org.isoron.uhabits.database
|
||||
|
||||
import org.isoron.androidbase.*
|
||||
import org.isoron.uhabits.*
|
||||
import org.isoron.uhabits.core.utils.*
|
||||
import org.junit.*
|
||||
import java.io.*
|
||||
|
||||
class AutoBackupTest : BaseAndroidTest() {
|
||||
@Test
|
||||
fun testRun() {
|
||||
DateUtils.setFixedLocalTime(40 * DateUtils.DAY_LENGTH)
|
||||
val basedir = AndroidDirFinder(targetContext).getFilesDir("Backups")!!
|
||||
createTestFiles(basedir, 30)
|
||||
|
||||
val autoBackup = AutoBackup(targetContext)
|
||||
autoBackup.run(keep=5)
|
||||
|
||||
for (k in 1..25) assertDoesNotExist("${basedir.path}/test-$k.txt")
|
||||
for (k in 26..30) assertExists("${basedir.path}/test-$k.txt")
|
||||
assertExists("${basedir.path}/Loop Habits Backup 1970-02-10 000000.db")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun testRunWithEmptyDir() {
|
||||
val basedir = AndroidDirFinder(targetContext).getFilesDir("Backups")!!
|
||||
removeAllFiles(basedir)
|
||||
basedir.delete()
|
||||
|
||||
// Should not crash
|
||||
val autoBackup = AutoBackup(targetContext)
|
||||
autoBackup.run()
|
||||
}
|
||||
|
||||
private fun assertExists(path: String) {
|
||||
assertTrue("File $path should exist", File(path).exists())
|
||||
}
|
||||
|
||||
private fun assertDoesNotExist(path: String) {
|
||||
assertFalse("File $path should not exist", File(path).exists())
|
||||
}
|
||||
|
||||
private fun createTestFiles(basedir: File, nfiles: Int) {
|
||||
removeAllFiles(basedir)
|
||||
for (k in 1..nfiles) {
|
||||
touch("${basedir.path}/test-$k.txt", DateUtils.DAY_LENGTH * k)
|
||||
}
|
||||
}
|
||||
|
||||
private fun touch(path: String, time: Long) {
|
||||
val file = File(path)
|
||||
FileOutputStream(file).close()
|
||||
file.setLastModified(time)
|
||||
}
|
||||
|
||||
private fun removeAllFiles(dir: File) {
|
||||
dir.list().forEach { path ->
|
||||
val file = File("${dir.path}/$path")
|
||||
assertTrue(file.delete())
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,61 @@
|
||||
/*
|
||||
* Copyright (C) 2016-2020 Álinson Santos Xavier <isoron@gmail.com>
|
||||
*
|
||||
* This file is part of Loop Habit Tracker.
|
||||
*
|
||||
* Loop Habit Tracker is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by the
|
||||
* Free Software Foundation, either version 3 of the License, or (at your
|
||||
* option) any later version.
|
||||
*
|
||||
* Loop Habit Tracker is distributed in the hope that it will be useful, but
|
||||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||||
* more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License along
|
||||
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package org.isoron.uhabits.database
|
||||
|
||||
import android.content.*
|
||||
import android.util.*
|
||||
import org.isoron.androidbase.*
|
||||
import org.isoron.uhabits.core.utils.*
|
||||
import org.isoron.uhabits.utils.*
|
||||
import java.io.*
|
||||
|
||||
class AutoBackup(private val context: Context) {
|
||||
|
||||
private val basedir = AndroidDirFinder(context).getFilesDir("Backups")!!
|
||||
|
||||
fun run(keep: Int = 5) {
|
||||
val files = listBackupFiles()
|
||||
var newestTimestamp = 0L;
|
||||
if (files.isNotEmpty()) {
|
||||
newestTimestamp = files.last().lastModified()
|
||||
}
|
||||
val todayTimestamp = DateUtils.getStartOfToday()
|
||||
removeOldest(files, keep)
|
||||
if (todayTimestamp - newestTimestamp > DateUtils.DAY_LENGTH) {
|
||||
DatabaseUtils.saveDatabaseCopy(context, basedir)
|
||||
}
|
||||
}
|
||||
|
||||
private fun removeOldest(files: ArrayList<File>, keep: Int) {
|
||||
files.sortBy { -it.lastModified() }
|
||||
for (k in keep until files.size) {
|
||||
Log.i("AutoBackup", "Removing ${files[k]}")
|
||||
files[k].delete()
|
||||
}
|
||||
}
|
||||
|
||||
private fun listBackupFiles(): ArrayList<File> {
|
||||
val files = ArrayList<File>()
|
||||
for (path in basedir.list()!!) {
|
||||
files.add(File("${basedir.path}/$path"))
|
||||
}
|
||||
return files
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue