Android/app/src/test/java/protect/card_locker/TestHelpers.java

package protect.card_locker;

import android.app.Activity;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;

import java.io.FileNotFoundException;

public class TestHelpers {
    static public DBHelper getEmptyDb(Activity activity) {
        DBHelper db = new DBHelper(activity);

        // Make sure no files remain
        Cursor cursor = db.getLoyaltyCardCursor();
        cursor.moveToFirst();
        while (!cursor.isAfterLast()) {
            int cardID = cursor.getColumnIndex(DBHelper.LoyaltyCardDbIds.ID);

            try {
                Utils.saveCardImage(activity.getApplicationContext(), null, cardID, true);
            } catch (FileNotFoundException ignored) {}
            try {
                Utils.saveCardImage(activity.getApplicationContext(), null, cardID, false);
            } catch (FileNotFoundException ignored) {}

            cursor.moveToNext();
        }

        // Make sure DB is empty
        SQLiteDatabase database = db.getWritableDatabase();
        database.execSQL("delete from " + DBHelper.LoyaltyCardDbIds.TABLE);
        database.execSQL("delete from " + DBHelper.LoyaltyCardDbGroups.TABLE);
        database.execSQL("delete from " + DBHelper.LoyaltyCardDbIdsGroups.TABLE);
        database.close();

        return db;
    }
}