#17[30m]. Add tests for CategoryRepo.

Author: Evgenii Kanivets

app/src/androidTest/java/com/blogspot/e_kanivets/moneytracker/repo/AccountRepoTest.java

@@ -23,11 +23,13 @@ public class AccountRepoTest extends TestCase {
     private AccountRepo repo;
 
     public void setUp() throws Exception {
+        super.setUp();
         DbHelper mock = Mockito.mock(DbHelper.class);
         repo = new AccountRepo(mock);
     }
 
     public void tearDown() throws Exception {
+        super.tearDown();
         repo = null;
     }
 

app/src/androidTest/java/com/blogspot/e_kanivets/moneytracker/repo/CategoryRepoTest.java

@@ -0,0 +1,72 @@
+package com.blogspot.e_kanivets.moneytracker.repo;
+
+import android.content.ContentValues;
+import android.database.Cursor;
+
+import com.blogspot.e_kanivets.moneytracker.DbHelper;
+import com.blogspot.e_kanivets.moneytracker.model.Category;
+
+import junit.framework.TestCase;
+
+import org.mockito.Mockito;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Android Test case.
+ * Created on 3/1/16.
+ *
+ * @author Evgenii Kanivets
+ */
+public class CategoryRepoTest extends TestCase {
+    private CategoryRepo repo;
+
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+        DbHelper mock = Mockito.mock(DbHelper.class);
+        repo = new CategoryRepo(mock);
+    }
+
+    @Override
+    public void tearDown() throws Exception {
+        super.tearDown();
+        repo = null;
+    }
+
+    public void testGetTable() throws Exception {
+        assertEquals(DbHelper.TABLE_CATEGORIES, repo.getTable());
+    }
+
+    public void testContentValues() throws Exception {
+        Category category = new Category(1, "category");
+
+        ContentValues expected = new ContentValues();
+        expected.put(DbHelper.NAME_COLUMN, "category");
+
+        ContentValues actual = repo.contentValues(category);
+
+        assertEquals(expected, actual);
+
+        assertNull(repo.contentValues(null));
+    }
+
+    public void testGetListFromCursor() throws Exception {
+        assertEquals(new ArrayList<Category>(), repo.getListFromCursor(Mockito.mock(Cursor.class)));
+
+        Cursor mockCursor = Mockito.mock(Cursor.class);
+        Mockito.when(mockCursor.moveToFirst()).thenReturn(true);
+        Mockito.when(mockCursor.getColumnIndex(DbHelper.ID_COLUMN)).thenReturn(1);
+        Mockito.when(mockCursor.getColumnIndex(DbHelper.NAME_COLUMN)).thenReturn(2);
+        Mockito.when(mockCursor.getLong(1)).thenReturn(1L);
+        Mockito.when(mockCursor.getString(2)).thenReturn("category");
+
+        List<Category> expected = new ArrayList<>();
+        expected.add(new Category(1, "category"));
+
+        assertEquals(expected, repo.getListFromCursor(mockCursor));
+
+        assertEquals(new ArrayList<>(), repo.getListFromCursor(null));
+    }
+}

app/src/main/java/com/blogspot/e_kanivets/moneytracker/model/Account.java

@@ -59,6 +59,18 @@ public void take(int amount) {
         curSum -= amount;
     }
 
+    @SuppressWarnings("SimplifiableIfStatement")
+    @Override
+    public boolean equals(Object o) {
+        if (o instanceof Account) {
+            Account account = (Account) o;
+            return this.id == account.getId()
+                    && this.title.equals(account.getTitle())
+                    && this.curSum == account.getCurSum()
+                    && this.currency.equals(account.getCurrency());
+        } else return false;
+    }
+
     @SuppressWarnings("StringBufferReplaceableByString")
     @Override
     public String toString() {
@@ -72,14 +84,4 @@ public String toString() {
 
         return sb.toString();
     }
-
-    @SuppressWarnings("SimplifiableIfStatement")
-    @Override
-    public boolean equals(Object o) {
-        if (o instanceof Account) {
-            return ((Account) o).getId() == getId();
-        } else {
-            return false;
-        }
-    }
 }

app/src/main/java/com/blogspot/e_kanivets/moneytracker/model/Category.java

@@ -30,6 +30,15 @@ public String getName() {
         return name;
     }
 
+    @Override
+    public boolean equals(Object o) {
+        if (o instanceof Category) {
+            Category category = (Category) o;
+            return this.id == category.getId()
+                    && this.name.equals(category.getName());
+        } else return false;
+    }
+
     @SuppressWarnings("StringBufferReplaceableByString")
     @Override
     public String toString() {