Added getter for DataBaseMigrations in IDataBasesApi

timsixth · Jun 16, 2023 · bff3919 · bff3919
1 parent c48a3d0
commit bff3919
Show file tree

Hide file tree

Showing 6 changed files with 25 additions and 5 deletions.
diff --git a/src/main/java/pl/timsixth/databasesapi/DatabasesAPI.java b/src/main/java/pl/timsixth/databasesapi/DatabasesAPI.java
@@ -4,6 +4,7 @@
 import pl.timsixth.databasesapi.api.IDataBasesApi;
 import pl.timsixth.databasesapi.config.IConfigFile;
 import pl.timsixth.databasesapi.database.ISQLDataBase;
+import pl.timsixth.databasesapi.database.migration.DataBaseMigrations;
 import pl.timsixth.databasesapi.database.migration.Migrations;
 
 @RequiredArgsConstructor
@@ -26,4 +27,9 @@ public Migrations getMigrations() {
         return databasesApiPlugin.migrations;
     }
 
+    @Override
+    public DataBaseMigrations getDataBaseMigrations() {
+        return databasesApiPlugin.dataBaseMigrations;
+    }
+
 }
diff --git a/src/main/java/pl/timsixth/databasesapi/DatabasesApiPlugin.java b/src/main/java/pl/timsixth/databasesapi/DatabasesApiPlugin.java
@@ -21,7 +21,7 @@ public final class DatabasesApiPlugin extends JavaPlugin {
     ISQLDataBase currentSQLDataBase;
     IConfigFile configFile;
     Migrations migrations;
-    private DataBaseMigrations dataBaseMigrations;
+    DataBaseMigrations dataBaseMigrations;
     private static IDataBasesApi dataBasesApi;
 
     @SneakyThrows

diff --git a/src/main/java/pl/timsixth/databasesapi/api/IDataBasesApi.java b/src/main/java/pl/timsixth/databasesapi/api/IDataBasesApi.java
@@ -2,16 +2,30 @@
 
 import pl.timsixth.databasesapi.config.IConfigFile;
 import pl.timsixth.databasesapi.database.ISQLDataBase;
+import pl.timsixth.databasesapi.database.migration.DataBaseMigrations;
 import pl.timsixth.databasesapi.database.migration.Migrations;
 
 /**
  * The interface to manage API
  */
 public interface IDataBasesApi {
-
+    /**
+     * @return interface which represents config file
+     */
     IConfigFile getConfig();
+
+    /**
+     * @return current sql database which is connected
+     */
     ISQLDataBase getCurrentSqlDataBase();
 
+    /**
+     * @return the class which represents every migration to migrate
+     */
     Migrations getMigrations();
 
+    /**
+     * @return the class which represents every migration in database
+     */
+    DataBaseMigrations getDataBaseMigrations();
 }
diff --git a/src/main/java/pl/timsixth/databasesapi/database/migration/DataBaseMigrations.java b/src/main/java/pl/timsixth/databasesapi/database/migration/DataBaseMigrations.java
@@ -111,7 +111,7 @@ public void updateMigration(DataBaseMigration dataBaseMigration, IMigration migr
     /**
      * Rollbacks all migrations from database
      */
-    public void migrationsRollback() throws ExecutionException, InterruptedException {
+    public void rollbackMigrations() throws ExecutionException, InterruptedException {
         ISQLDataBase currentSqlDataBase = DatabasesApiPlugin.getApi().getCurrentSqlDataBase();
 
         String truncateMigrationTable = "TRUNCATE " + MIGRATION_TABLE_NAME;

diff --git a/src/test/java/pl/timsixth/databasesapi/tests/MigrationTest.java b/src/test/java/pl/timsixth/databasesapi/tests/MigrationTest.java
@@ -56,7 +56,7 @@ public void shouldMigrateSingleMigration() {
 
     @Test
     public void shouldRollbackAllMigrations() throws ExecutionException, InterruptedException {
-        dataBaseMigrationsStub.migrationsRollback();
+        dataBaseMigrationsStub.rollbackMigrations();
 
         assertEquals(0, dataBaseMigrationsStub.getDataBaseMigrations().size());
     }

diff --git a/src/test/java/pl/timsixth/databasesapi/tests/migrations/stubs/DataBaseMigrationsStub.java b/src/test/java/pl/timsixth/databasesapi/tests/migrations/stubs/DataBaseMigrationsStub.java
@@ -54,7 +54,7 @@ public void updateMigration(DataBaseMigration dataBaseMigration, IMigration migr
     }
 
     @Override
-    public void migrationsRollback() throws ExecutionException, InterruptedException {
+    public void rollbackMigrations() throws ExecutionException, InterruptedException {
         String truncateMigrationTable = "TRUNCATE " + MIGRATION_TABLE_NAME;
 
         try {