Merge pull request #333 from gfx/delete_all

add OrmaDatabase#deleteAll()
maskarade · Nov 11, 2016 · 29beea3 · 29beea3
2 parents 741809e + 4936cfe
commit 29beea3
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 7 deletions.
diff --git a/...ild/generated/source/apt/debug/com/github/gfx/android/orma/example/orma/OrmaDatabase.java b/...ild/generated/source/apt/debug/com/github/gfx/android/orma/example/orma/OrmaDatabase.java
@@ -90,6 +90,10 @@ public Completable transactionNonExclusiveAsync(@NonNull Runnable task) {
     return connection.transactionNonExclusiveAsync(task);
   }
 
+  public void deleteAll() {
+    connection.deleteAll();
+  }
+
   /**
    * Retrieves a model from a cursor. */
   @NonNull

diff --git a/library/src/test/java/com/github/gfx/android/orma/test/OrmaDatabaseTest.java b/library/src/test/java/com/github/gfx/android/orma/test/OrmaDatabaseTest.java
@@ -72,7 +72,7 @@ public void testNumberOfSchemas() throws Exception {
     }
 
     @Test
-    public void testResetDatabase() throws Exception {
+    public void testDeleteAll() throws Exception {
         OrmaDatabase db = OrmaDatabase.builder(getContext())
                 .name(NAME)
                 .trace(true)
@@ -91,12 +91,7 @@ public Author call() {
 
         assertThat(db.selectFromAuthor().count(), is(1));
 
-        getContext().deleteDatabase(NAME);
-        db = OrmaDatabase.builder(getContext())
-                .name(NAME)
-                .trace(true)
-                .tryParsingSql(false)
-                .build();
+        db.deleteAll();
 
         assertThat(db.selectFromAuthor().count(), is(0));
     }

diff --git a/processor/src/main/java/com/github/gfx/android/orma/processor/generator/DatabaseWriter.java b/processor/src/main/java/com/github/gfx/android/orma/processor/generator/DatabaseWriter.java
@@ -298,6 +298,13 @@ public List<MethodSpec> buildMethodSpecs(ClassName builderClass) {
                         .build()
         );
 
+        methodSpecs.add(
+                MethodSpec.methodBuilder("deleteAll")
+                        .addModifiers(Modifier.PUBLIC)
+                        .addStatement("$L.deleteAll()", connection)
+                        .build()
+        );
+
         database.getSchemas().forEach(schema -> {
             String simpleModelName = schema.getModelClassName().simpleName();
             CodeBlock schemaInstance = schema.createSchemaInstanceExpr();