mirror of
https://github.com/CatimaLoyalty/Android.git
synced 2026-01-01 03:27:55 -05:00
441 lines
16 KiB
Java
441 lines
16 KiB
Java
package protect.card_locker;
|
|
|
|
import android.app.Activity;
|
|
import android.content.ContentValues;
|
|
import android.database.Cursor;
|
|
import android.database.sqlite.SQLiteDatabase;
|
|
import android.graphics.Color;
|
|
|
|
import com.google.zxing.BarcodeFormat;
|
|
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.robolectric.Robolectric;
|
|
import org.robolectric.RobolectricTestRunner;
|
|
import org.robolectric.annotation.Config;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertFalse;
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertNull;
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
@RunWith(RobolectricTestRunner.class)
|
|
@Config(sdk = 23)
|
|
public class DatabaseTest
|
|
{
|
|
private DBHelper db;
|
|
|
|
private static final Integer DEFAULT_HEADER_COLOR = Color.BLACK;
|
|
private static final Integer DEFAULT_HEADER_TEXT_COLOR = Color.WHITE;
|
|
|
|
@Before
|
|
public void setUp()
|
|
{
|
|
Activity activity = Robolectric.setupActivity(MainActivity.class);
|
|
db = new DBHelper(activity);
|
|
}
|
|
|
|
@Test
|
|
public void addRemoveOneGiftCard()
|
|
{
|
|
assertEquals(0, db.getLoyaltyCardCount());
|
|
long id = db.insertLoyaltyCard("store", "note", "cardId", BarcodeFormat.UPC_A.toString(), DEFAULT_HEADER_COLOR, 0);
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
LoyaltyCard loyaltyCard = db.getLoyaltyCard(1);
|
|
assertNotNull(loyaltyCard);
|
|
assertEquals("store", loyaltyCard.store);
|
|
assertEquals("note", loyaltyCard.note);
|
|
assertEquals("cardId", loyaltyCard.cardId);
|
|
assertEquals(0, loyaltyCard.starStatus);
|
|
assertEquals(BarcodeFormat.UPC_A.toString(), loyaltyCard.barcodeType);
|
|
|
|
result = db.deleteLoyaltyCard(1);
|
|
assertTrue(result);
|
|
assertEquals(0, db.getLoyaltyCardCount());
|
|
assertNull(db.getLoyaltyCard(1));
|
|
}
|
|
|
|
@Test
|
|
public void updateGiftCard()
|
|
{
|
|
long id = db.insertLoyaltyCard("store", "note", "cardId", BarcodeFormat.UPC_A.toString(), DEFAULT_HEADER_COLOR, 0);
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
result = db.updateLoyaltyCard(1, "store1", "note1", "cardId1", BarcodeFormat.AZTEC.toString(), DEFAULT_HEADER_COLOR);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
LoyaltyCard loyaltyCard = db.getLoyaltyCard(1);
|
|
assertNotNull(loyaltyCard);
|
|
assertEquals("store1", loyaltyCard.store);
|
|
assertEquals("note1", loyaltyCard.note);
|
|
assertEquals("cardId1", loyaltyCard.cardId);
|
|
assertEquals(0, loyaltyCard.starStatus);
|
|
assertEquals(BarcodeFormat.AZTEC.toString(), loyaltyCard.barcodeType);
|
|
}
|
|
|
|
@Test
|
|
public void updateGiftCardOnlyStar()
|
|
{
|
|
long id = db.insertLoyaltyCard("store", "note", "cardId", BarcodeFormat.UPC_A.toString(), DEFAULT_HEADER_COLOR, 0);
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
result = db.updateLoyaltyCardStarStatus(1, 1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
LoyaltyCard loyaltyCard = db.getLoyaltyCard(1);
|
|
assertNotNull(loyaltyCard);
|
|
assertEquals("store", loyaltyCard.store);
|
|
assertEquals("note", loyaltyCard.note);
|
|
assertEquals("cardId", loyaltyCard.cardId);
|
|
assertEquals(1, loyaltyCard.starStatus);
|
|
assertEquals(BarcodeFormat.UPC_A.toString(), loyaltyCard.barcodeType);
|
|
}
|
|
|
|
@Test
|
|
public void updateMissingGiftCard()
|
|
{
|
|
assertEquals(0, db.getLoyaltyCardCount());
|
|
|
|
boolean result = db.updateLoyaltyCard(1, "store1", "note1", "cardId1",
|
|
BarcodeFormat.UPC_A.toString(), DEFAULT_HEADER_COLOR);
|
|
assertEquals(false, result);
|
|
assertEquals(0, db.getLoyaltyCardCount());
|
|
}
|
|
|
|
@Test
|
|
public void emptyGiftCardValues()
|
|
{
|
|
long id = db.insertLoyaltyCard("", "", "", "", null, 0);
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
LoyaltyCard loyaltyCard = db.getLoyaltyCard(1);
|
|
assertNotNull(loyaltyCard);
|
|
assertEquals("", loyaltyCard.store);
|
|
assertEquals("", loyaltyCard.note);
|
|
assertEquals("", loyaltyCard.cardId);
|
|
assertEquals("", loyaltyCard.barcodeType);
|
|
}
|
|
|
|
@Test
|
|
public void giftCardsViaCursor()
|
|
{
|
|
final int CARDS_TO_ADD = 10;
|
|
|
|
// Add the gift cards in reverse order, to ensure
|
|
// that they are sorted
|
|
for(int index = CARDS_TO_ADD-1; index >= 0; index--)
|
|
{
|
|
long id = db.insertLoyaltyCard("store" + index, "note" + index, "cardId" + index,
|
|
BarcodeFormat.UPC_A.toString(), index, 0);
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
}
|
|
|
|
assertEquals(CARDS_TO_ADD, db.getLoyaltyCardCount());
|
|
|
|
Cursor cursor = db.getLoyaltyCardCursor();
|
|
assertNotNull(cursor);
|
|
|
|
assertEquals(CARDS_TO_ADD, cursor.getCount());
|
|
|
|
cursor.moveToFirst();
|
|
|
|
for(int index = 0; index < CARDS_TO_ADD; index++)
|
|
{
|
|
assertEquals("store"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.STORE)));
|
|
assertEquals("note"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.NOTE)));
|
|
assertEquals("cardId"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.CARD_ID)));
|
|
assertEquals("cardId"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.CARD_ID)));
|
|
assertEquals(BarcodeFormat.UPC_A.toString(), cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.BARCODE_TYPE)));
|
|
assertEquals(0, cursor.getInt(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.STAR_STATUS)));
|
|
assertEquals(index, cursor.getInt(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.HEADER_COLOR)));
|
|
|
|
cursor.moveToNext();
|
|
}
|
|
|
|
assertTrue(cursor.isAfterLast());
|
|
}
|
|
|
|
@Test
|
|
public void giftCardsViaCursorWithOneStarred() //sorting test; stared card should appear first
|
|
{
|
|
final int CARDS_TO_ADD = 10;
|
|
long id;
|
|
// Add the gift cards in reverse order and add one with STAR, to ensure
|
|
// that they are sorted
|
|
for(int index = CARDS_TO_ADD-1; index >= 0; index--)
|
|
{
|
|
if (index == CARDS_TO_ADD-1) {
|
|
id = db.insertLoyaltyCard("store" + index, "note" + index, "cardId" + index,
|
|
BarcodeFormat.UPC_A.toString(), index, 1);
|
|
}
|
|
|
|
else {
|
|
id = db.insertLoyaltyCard("store" + index, "note" + index, "cardId" + index,
|
|
BarcodeFormat.UPC_A.toString(), index, 0);
|
|
}
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
}
|
|
|
|
assertEquals(CARDS_TO_ADD, db.getLoyaltyCardCount());
|
|
|
|
Cursor cursor = db.getLoyaltyCardCursor();
|
|
assertNotNull(cursor);
|
|
|
|
assertEquals(CARDS_TO_ADD, cursor.getCount());
|
|
|
|
cursor.moveToFirst();
|
|
int index = CARDS_TO_ADD-1 ;
|
|
assertEquals("store"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.STORE)));
|
|
assertEquals("note"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.NOTE)));
|
|
assertEquals("cardId"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.CARD_ID)));
|
|
assertEquals("cardId"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.CARD_ID)));
|
|
assertEquals(BarcodeFormat.UPC_A.toString(), cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.BARCODE_TYPE)));
|
|
assertEquals(1, cursor.getInt(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.STAR_STATUS)));
|
|
assertEquals(index, cursor.getInt(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.HEADER_COLOR)));
|
|
|
|
cursor.moveToNext();
|
|
|
|
for(index = 0; index < CARDS_TO_ADD-1; index++)
|
|
{
|
|
assertEquals("store"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.STORE)));
|
|
assertEquals("note"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.NOTE)));
|
|
assertEquals("cardId"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.CARD_ID)));
|
|
assertEquals("cardId"+index, cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.CARD_ID)));
|
|
assertEquals(BarcodeFormat.UPC_A.toString(), cursor.getString(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.BARCODE_TYPE)));
|
|
assertEquals(0, cursor.getInt(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.STAR_STATUS)));
|
|
assertEquals(index, cursor.getInt(cursor.getColumnIndexOrThrow(DBHelper.LoyaltyCardDbIds.HEADER_COLOR)));
|
|
|
|
cursor.moveToNext();
|
|
}
|
|
|
|
assertTrue(cursor.isAfterLast());
|
|
}
|
|
|
|
private void setupDatabaseVersion1(SQLiteDatabase database)
|
|
{
|
|
// Delete the tables as they exist now
|
|
database.execSQL("drop table " + DBHelper.LoyaltyCardDbIds.TABLE);
|
|
database.execSQL("drop table " + DBHelper.LoyaltyCardDbGroups.TABLE);
|
|
database.execSQL("drop table " + DBHelper.LoyaltyCardDbIdsGroups.TABLE);
|
|
|
|
// Create the table as it existed in revision 1
|
|
database.execSQL("create table " + DBHelper.LoyaltyCardDbIds.TABLE + "(" +
|
|
DBHelper.LoyaltyCardDbIds.ID + " INTEGER primary key autoincrement," +
|
|
DBHelper.LoyaltyCardDbIds.STORE + " TEXT not null," +
|
|
DBHelper.LoyaltyCardDbIds.CARD_ID + " TEXT not null," +
|
|
DBHelper.LoyaltyCardDbIds.BARCODE_TYPE + " TEXT not null)");
|
|
}
|
|
|
|
private int insertCardVersion1(SQLiteDatabase database,
|
|
final String store, final String cardId,
|
|
final String barcodeType)
|
|
{
|
|
ContentValues contentValues = new ContentValues();
|
|
contentValues.put(DBHelper.LoyaltyCardDbIds.STORE, store);
|
|
contentValues.put(DBHelper.LoyaltyCardDbIds.CARD_ID, cardId);
|
|
contentValues.put(DBHelper.LoyaltyCardDbIds.BARCODE_TYPE, barcodeType);
|
|
final long newId = database.insert(DBHelper.LoyaltyCardDbIds.TABLE, null, contentValues);
|
|
assertTrue(newId != -1);
|
|
return (int)newId;
|
|
}
|
|
|
|
@Test
|
|
public void addRemoveOneGroup()
|
|
{
|
|
assertEquals(0, db.getGroupCount());
|
|
long id = db.insertGroup("group one");
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getGroupCount());
|
|
|
|
Group group = db.getGroup("group one");
|
|
assertNotNull(group);
|
|
assertEquals("group one", group._id);
|
|
|
|
result = db.deleteGroup("group one");
|
|
assertTrue(result);
|
|
assertEquals(0, db.getGroupCount());
|
|
assertNull(db.getGroup("group one"));
|
|
}
|
|
|
|
@Test
|
|
public void updateGroup()
|
|
{
|
|
long id = db.insertGroup("group one");
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getGroupCount());
|
|
|
|
result = db.updateGroup("group one", "group one renamed");
|
|
assertTrue(result);
|
|
assertEquals(1, db.getGroupCount());
|
|
|
|
// Group one no longer exists
|
|
Group group = db.getGroup("group one");
|
|
assertNull(group);
|
|
|
|
// But group one renamed does
|
|
Group group2 = db.getGroup("group one renamed");
|
|
assertNotNull(group2);
|
|
assertEquals("group one renamed", group2._id);
|
|
}
|
|
|
|
@Test
|
|
public void updateMissingGroup()
|
|
{
|
|
assertEquals(0, db.getGroupCount());
|
|
|
|
boolean result = db.updateGroup("group one", "new name");
|
|
assertEquals(false, result);
|
|
assertEquals(0, db.getGroupCount());
|
|
}
|
|
|
|
@Test
|
|
public void emptyGroupValues()
|
|
{
|
|
long id = db.insertGroup("");
|
|
boolean result = (id != -1);
|
|
assertFalse(result);
|
|
assertEquals(0, db.getLoyaltyCardCount());
|
|
}
|
|
|
|
@Test
|
|
public void duplicateGroupName()
|
|
{
|
|
assertEquals(0, db.getGroupCount());
|
|
long id = db.insertGroup("group one");
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getGroupCount());
|
|
|
|
Group group = db.getGroup("group one");
|
|
assertNotNull(group);
|
|
assertEquals("group one", group._id);
|
|
|
|
// Should fail on duplicate
|
|
long id2 = db.insertGroup("group one");
|
|
boolean result2 = (id2 != -1);
|
|
assertFalse(result2);
|
|
assertEquals(1, db.getGroupCount());
|
|
}
|
|
|
|
@Test
|
|
public void updateGroupDuplicate()
|
|
{
|
|
long id = db.insertGroup("group one");
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getGroupCount());
|
|
|
|
long id2 = db.insertGroup("group two");
|
|
boolean result2 = (id2 != -1);
|
|
assertTrue(result2);
|
|
assertEquals(2, db.getGroupCount());
|
|
|
|
// Should fail when trying to rename group two to one
|
|
boolean result3 = db.updateGroup("group two", "group one");
|
|
assertFalse(result3);
|
|
assertEquals(2, db.getGroupCount());
|
|
|
|
// Rename failed so both should still be the same
|
|
Group group = db.getGroup("group one");
|
|
assertNotNull(group);
|
|
assertEquals("group one", group._id);
|
|
|
|
Group group2 = db.getGroup("group two");
|
|
assertNotNull(group2);
|
|
assertEquals("group two", group2._id);
|
|
}
|
|
|
|
@Test
|
|
public void cardAddAndRemoveGroups()
|
|
{
|
|
// Create card
|
|
assertEquals(0, db.getLoyaltyCardCount());
|
|
long id = db.insertLoyaltyCard("store", "note", "cardId", BarcodeFormat.UPC_A.toString(), DEFAULT_HEADER_COLOR, 0);
|
|
boolean result = (id != -1);
|
|
assertTrue(result);
|
|
assertEquals(1, db.getLoyaltyCardCount());
|
|
|
|
// Create two groups to only one card
|
|
assertEquals(0, db.getGroupCount());
|
|
long gid = db.insertGroup("one");
|
|
boolean gresult = (gid != -1);
|
|
assertTrue(gresult);
|
|
|
|
long gid2 = db.insertGroup("two");
|
|
boolean gresult2 = (gid2 != -1);
|
|
assertTrue(gresult2);
|
|
|
|
assertEquals(2, db.getGroupCount());
|
|
|
|
Group group1 = db.getGroup("one");
|
|
|
|
// Card has no groups by default
|
|
List<Group> cardGroups = db.getLoyaltyCardGroups(1);
|
|
assertEquals(0, cardGroups.size());
|
|
|
|
// Add one groups to card
|
|
List<Group> groupList1 = new ArrayList<>();
|
|
groupList1.add(group1);
|
|
db.setLoyaltyCardGroups(1, groupList1);
|
|
|
|
List<Group> cardGroups1 = db.getLoyaltyCardGroups(1);
|
|
assertEquals(1, cardGroups1.size());
|
|
assertEquals(cardGroups1.get(0)._id, group1._id);
|
|
assertEquals(1, db.getGroupCardCount("one"));
|
|
assertEquals(0, db.getGroupCardCount("two"));
|
|
|
|
// Remove groups
|
|
db.setLoyaltyCardGroups(1, new ArrayList<Group>());
|
|
List<Group> cardGroups2 = db.getLoyaltyCardGroups(1);
|
|
assertEquals(0, cardGroups2.size());
|
|
assertEquals(0, db.getGroupCardCount("one"));
|
|
assertEquals(0, db.getGroupCardCount("two"));
|
|
}
|
|
|
|
@Test
|
|
public void databaseUpgradeFromVersion1()
|
|
{
|
|
SQLiteDatabase database = db.getWritableDatabase();
|
|
|
|
// Setup the database as it appeared in revision 1
|
|
setupDatabaseVersion1(database);
|
|
|
|
// Insert a budget and transaction
|
|
int newCardId = insertCardVersion1(database, "store", "cardId", BarcodeFormat.UPC_A.toString());
|
|
|
|
// Upgrade database
|
|
db.onUpgrade(database, DBHelper.ORIGINAL_DATABASE_VERSION, DBHelper.DATABASE_VERSION);
|
|
|
|
// Determine that the entries are queryable and the fields are correct
|
|
LoyaltyCard card = db.getLoyaltyCard(newCardId);
|
|
assertEquals("store", card.store);
|
|
assertEquals("cardId", card.cardId);
|
|
assertEquals(BarcodeFormat.UPC_A.toString(), card.barcodeType);
|
|
assertEquals("", card.note);
|
|
assertEquals(null, card.headerColor);
|
|
assertEquals(null, card.headerTextColor);
|
|
|
|
database.close();
|
|
}
|
|
}
|