Move importers to uhabits-core

uhabits-core/src/test/java/org/isoron/uhabits/BaseUnitTest.java

@@ -66,7 +66,7 @@ public class BaseUnitTest
     }
 
     @After
-    public void tearDown()
+    public void tearDown() throws Exception
     {
         validateMockitoUsage();
         DateUtils.setFixedLocalTime(null);

uhabits-core/src/test/java/org/isoron/uhabits/core/io/HabitsCSVExporterTest.java

@@ -0,0 +1,116 @@
+/*
+ * Copyright (C) 2017 Á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.core.io;
+
+import org.apache.commons.io.*;
+import org.isoron.uhabits.*;
+import org.isoron.uhabits.core.models.*;
+import org.junit.*;
+
+import java.io.*;
+import java.nio.file.*;
+import java.util.*;
+import java.util.zip.*;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+public class HabitsCSVExporterTest extends BaseUnitTest
+{
+    private File baseDir;
+
+    @Before
+    public void setUp() throws Exception
+    {
+        super.setUp();
+        habitList.add(fixtures.createShortHabit());
+        habitList.add(fixtures.createEmptyHabit());
+        baseDir = Files.createTempDirectory("csv").toFile();
+        assertNotNull(baseDir);
+    }
+
+    @Override
+    public void tearDown() throws Exception
+    {
+        FileUtils.deleteDirectory(baseDir);
+        super.tearDown();
+    }
+
+    @Test
+    public void testExportCSV() throws IOException
+    {
+        List<Habit> selected = new LinkedList<>();
+        for (Habit h : habitList) selected.add(h);
+
+        HabitsCSVExporter exporter =
+            new HabitsCSVExporter(habitList, selected, baseDir);
+        String filename = exporter.writeArchive();
+        assertAbsolutePathExists(filename);
+
+        File archive = new File(filename);
+        unzip(archive);
+
+        assertPathExists("Habits.csv");
+        assertPathExists("001 Wake up early");
+        assertPathExists("001 Wake up early/Checkmarks.csv");
+        assertPathExists("001 Wake up early/Scores.csv");
+        assertPathExists("002 Meditate/Checkmarks.csv");
+        assertPathExists("002 Meditate/Scores.csv");
+        assertPathExists("Checkmarks.csv");
+        assertPathExists("Scores.csv");
+    }
+
+    private void unzip(File file) throws IOException
+    {
+        ZipFile zip = new ZipFile(file);
+        Enumeration<? extends ZipEntry> e = zip.entries();
+
+        while (e.hasMoreElements())
+        {
+            ZipEntry entry = e.nextElement();
+            InputStream stream = zip.getInputStream(entry);
+
+            String outputFilename =
+                String.format("%s/%s", baseDir.getAbsolutePath(),
+                    entry.getName());
+            File out = new File(outputFilename);
+            File parent = out.getParentFile();
+            if (parent != null) parent.mkdirs();
+
+            IOUtils.copy(stream, new FileOutputStream(out));
+        }
+
+        zip.close();
+    }
+
+    private void assertPathExists(String s)
+    {
+        assertAbsolutePathExists(
+            String.format("%s/%s", baseDir.getAbsolutePath(), s));
+    }
+
+    private void assertAbsolutePathExists(String s)
+    {
+        File file = new File(s);
+        assertTrue(
+            String.format("File %s should exist", file.getAbsolutePath()),
+            file.exists());
+    }
+}

uhabits-core/src/test/java/org/isoron/uhabits/core/io/ImportTest.java

@@ -0,0 +1,184 @@
+/*
+ * Copyright (C) 2017 Á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.core.io;
+
+import android.support.annotation.*;
+
+import org.apache.commons.io.*;
+import org.isoron.uhabits.*;
+import org.isoron.uhabits.core.database.*;
+import org.isoron.uhabits.core.models.*;
+import org.isoron.uhabits.core.utils.*;
+import org.junit.*;
+
+import java.io.*;
+import java.sql.*;
+import java.util.*;
+
+import sun.reflect.generics.reflectiveObjects.*;
+
+import static junit.framework.TestCase.assertFalse;
+import static org.hamcrest.MatcherAssert.*;
+import static org.hamcrest.core.IsEqual.*;
+import static org.isoron.uhabits.core.models.Frequency.THREE_TIMES_PER_WEEK;
+import static org.junit.Assert.assertTrue;
+
+public class ImportTest extends BaseUnitTest
+{
+    @Override
+    @Before
+    public void setUp() throws Exception
+    {
+        super.setUp();
+        DateUtils.setFixedLocalTime(null);
+    }
+
+    @Test
+    public void testHabitBullCSV() throws IOException
+    {
+        importFromFile("habitbull.csv");
+
+        assertThat(habitList.size(), equalTo(4));
+
+        Habit habit = habitList.getByPosition(0);
+        assertThat(habit.getName(), equalTo("Breed dragons"));
+        assertThat(habit.getDescription(), equalTo("with love and fire"));
+        assertThat(habit.getFrequency(), equalTo(Frequency.DAILY));
+        assertTrue(containsRepetition(habit, 2016, 3, 18));
+        assertTrue(containsRepetition(habit, 2016, 3, 19));
+        assertFalse(containsRepetition(habit, 2016, 3, 20));
+    }
+
+    @Test
+    public void testLoopDB() throws IOException
+    {
+        importFromFile("loop.db");
+
+        assertThat(habitList.size(), equalTo(9));
+
+        Habit habit = habitList.getByPosition(0);
+        assertThat(habit.getName(), equalTo("Wake up early"));
+        assertThat(habit.getFrequency(), equalTo(THREE_TIMES_PER_WEEK));
+        assertTrue(containsRepetition(habit, 2016, 3, 14));
+        assertTrue(containsRepetition(habit, 2016, 3, 16));
+        assertFalse(containsRepetition(habit, 2016, 3, 17));
+    }
+
+    @Test
+    public void testRewireDB() throws IOException
+    {
+        importFromFile("rewire.db");
+
+        assertThat(habitList.size(), equalTo(3));
+
+        Habit habit = habitList.getByPosition(0);
+        assertThat(habit.getName(), equalTo("Wake up early"));
+        assertThat(habit.getFrequency(), equalTo(THREE_TIMES_PER_WEEK));
+        assertFalse(habit.hasReminder());
+        assertFalse(containsRepetition(habit, 2015, 12, 31));
+        assertTrue(containsRepetition(habit, 2016, 1, 18));
+        assertTrue(containsRepetition(habit, 2016, 1, 28));
+        assertFalse(containsRepetition(habit, 2016, 3, 10));
+
+        habit = habitList.getByPosition(1);
+        assertThat(habit.getName(), equalTo("brush teeth"));
+        assertThat(habit.getFrequency(), equalTo(THREE_TIMES_PER_WEEK));
+        assertThat(habit.hasReminder(), equalTo(true));
+
+        Reminder reminder = habit.getReminder();
+        assertThat(reminder.getHour(), equalTo(8));
+        assertThat(reminder.getMinute(), equalTo(0));
+        boolean[] reminderDays = { false, true, true, true, true, true, false };
+        assertThat(reminder.getDays().toArray(), equalTo(reminderDays));
+    }
+
+    @Test
+    public void testTickmateDB() throws IOException
+    {
+        importFromFile("tickmate.db");
+
+        assertThat(habitList.size(), equalTo(3));
+
+        Habit h = habitList.getByPosition(0);
+        assertThat(h.getName(), equalTo("Vegan"));
+        assertTrue(containsRepetition(h, 2016, 1, 24));
+        assertTrue(containsRepetition(h, 2016, 2, 5));
+        assertTrue(containsRepetition(h, 2016, 3, 18));
+        assertFalse(containsRepetition(h, 2016, 3, 14));
+    }
+
+    private boolean containsRepetition(Habit h, int year, int month, int day)
+    {
+        GregorianCalendar date = DateUtils.getStartOfTodayCalendar();
+        date.set(year, month - 1, day);
+        return h.getRepetitions().containsTimestamp(date.getTimeInMillis());
+    }
+
+    private void copyAssetToFile(String assetPath, File dst) throws IOException
+    {
+        InputStream in = getClass().getResourceAsStream(assetPath);
+        if(in == null) {
+            File file = new File("uhabits-core/src/test/resources/" + assetPath);
+            if(file.exists()) in = new FileInputStream(file);
+        }
+
+        IOUtils.copy(in, new FileOutputStream(dst));
+    }
+
+    private void importFromFile(String assetFilename) throws IOException
+    {
+        File file = File.createTempFile("asset", "");
+        copyAssetToFile(assetFilename, file);
+        assertTrue(file.exists());
+        assertTrue(file.canRead());
+
+        DatabaseOpener opener = new DatabaseOpener() {
+            @Override
+            public Database open(@NonNull File file)
+            {
+                try
+                {
+                    return new JdbcDatabase(DriverManager.getConnection(
+                        String.format("jdbc:sqlite:%s", file.getAbsolutePath())));
+                }
+                catch (SQLException e)
+                {
+                    throw new RuntimeException(e);
+                }
+            }
+
+            @Override
+            public File getProductionDatabaseFile()
+            {
+                throw new NotImplementedException();
+            }
+        };
+        GenericImporter importer = new GenericImporter(habitList,
+            new LoopDBImporter(habitList, opener),
+            new RewireDBImporter(habitList, modelFactory, opener),
+            new TickmateDBImporter(habitList, modelFactory, opener),
+            new HabitBullCSVImporter(habitList, modelFactory));
+
+        assertTrue(importer.canHandle(file));
+        importer.importHabitsFromFile(file);
+
+        file.delete();
+    }
+}

uhabits-core/src/test/java/org/isoron/uhabits/core/models/RepetitionListTest.java

@@ -74,7 +74,7 @@ public class RepetitionListTest extends BaseUnitTest
 
     @Override
     @After
-    public void tearDown()
+    public void tearDown() throws Exception
     {
         super.tearDown();
     }

uhabits-core/src/test/java/org/isoron/uhabits/core/models/sqlite/SQLiteHabitListTest.java

@@ -74,7 +74,7 @@ public class SQLiteHabitListTest extends BaseUnitTest
     }
 
     @Override
-    public void tearDown()
+    public void tearDown() throws Exception
     {
         habitList.getObservable().removeListener(listener);
         super.tearDown();

uhabits-core/src/test/java/org/isoron/uhabits/core/models/sqlite/records/RepetitionRecordTest.java

@@ -23,7 +23,6 @@ package org.isoron.uhabits.core.models.sqlite.records;
 
 import org.isoron.uhabits.*;
 import org.isoron.uhabits.core.models.*;
-import org.isoron.uhabits.core.models.sqlite.records.*;
 import org.junit.*;
 
 import static org.hamcrest.MatcherAssert.assertThat;

uhabits-core/src/test/resources/habitbull.csv

@@ -0,0 +1,19 @@
+HabitName,HabitDescription,HabitCategory,CalendarDate,Value,CommentText
+Breed dragons,with love and fire,Diet & Food,2016-03-18,1,
+Breed dragons,with love and fire,Diet & Food,2016-03-19,1,
+Breed dragons,with love and fire,Diet & Food,2016-03-21,1,
+Reduce sleep,only 2 hours per day,Time Management,2016-03-15,1,
+Reduce sleep,only 2 hours per day,Time Management,2016-03-16,1,
+Reduce sleep,only 2 hours per day,Time Management,2016-03-17,1,
+Reduce sleep,only 2 hours per day,Time Management,2016-03-21,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-15,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-16,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-18,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-21,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-15,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-16,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-18,1,
+No-arms pushup,Become like water my friend!,Fitness,2016-03-21,1,
+Grow spiritually,"transcend ego, practice compassion, smile and breath",Meditation,2016-03-15,1,
+Grow spiritually,"transcend ego, practice compassion, smile and breath",Meditation,2016-03-17,1,
+Grow spiritually,"transcend ego, practice compassion, smile and breath",Meditation,2016-03-21,1,