Wrote some more test for the model class

Signed-off-by: Gertjan Roke <[email protected]>

Author: GertjanRoke

README.md

@@ -22,9 +22,9 @@ composer require gertjanroke/laravel-db-model
 
 namespace App\Models;
 
-use GertjanRoke\LaravelDbModel\DbModel;
+use GertjanRoke\LaravelDbModel\DBModel;
 
-class Post extends DbModel
+class Post extends DBModel
 {
     public $table = 'posts';
 

composer.json

@@ -18,6 +18,8 @@
     "require": {
         "php": "^7.4|^8.0",
         "spatie/laravel-package-tools": "^1.4.3",
+        "illuminate/collections": "^8.37",
+        "illuminate/support": "^8.37",
         "illuminate/database": "^8.37"
     },
     "require-dev": {

src/DBModel.php

@@ -0,0 +1,68 @@
+<?php
+
+namespace GertjanRoke\LaravelDbModel;
+
+use Closure;
+use Illuminate\Database\Query\Builder;
+use Illuminate\Support\Collection;
+use Illuminate\Support\Facades\DB;
+use Illuminate\Support\Str;
+
+/**
+ * @mixin DB
+ */
+class DBModel
+{
+    /**
+     * @var Closure|Builder|string
+     */
+    public $table;
+
+    public string $as = '';
+
+    public string $connection = '';
+
+    protected Builder $db;
+
+    public function __construct()
+    {
+        $this->db = DB::connection($this->connection ?: null)->table($this->getTable(), $this->as ?: null);
+    }
+
+    /**
+     * @return Closure|Builder|string
+     */
+    public function getTable()
+    {
+        return $this->table ?? Str::snake(Str::pluralStudly(class_basename($this)));
+    }
+
+    public function getDB(): Builder
+    {
+        return $this->db;
+    }
+
+    /**
+     * @param string $method
+     * @param array $parameters
+     *
+     * @return Builder|self|Collection
+     */
+    public function __call($method, $parameters)
+    {
+        $response = $this->db->$method(...$parameters);
+
+        return ($response instanceof Builder) ? $this : $response;
+    }
+
+    /**
+     * @param string $method
+     * @param array $parameters
+     *
+     * @return Builder|self|Collection
+     */
+    public static function __callStatic($method, $parameters)
+    {
+        return (new static())->$method(...$parameters);
+    }
+}

src/DbModel.php

@@ -1,27 +1,29 @@
 <?php
 
-use GertjanRoke\LaravelDbModel\Exceptions\MissingTableNameException;
 use GertjanRoke\LaravelDbModel\Tests\Models\Model;
 use GertjanRoke\LaravelDbModel\Tests\Models\ModelWithMySqlConnection;
 use GertjanRoke\LaravelDbModel\Tests\Models\ModelWithoutConnection;
 use GertjanRoke\LaravelDbModel\Tests\Models\ModelWithoutTableName;
 
-it('throws an exception when no table name is set', function () {
-    ModelWithoutTableName::where('column', 1);
-})->throws(MissingTableNameException::class);
-
 it('can prefill the connection', function () {
-    $query = ModelWithoutConnection::where('column', 1);
+    $model = new ModelWithoutConnection();
 
-    expect(class_basename($query->connection))->toStartWith('SQLiteConnection');
+    expect(class_basename($model->getDB()->connection))->toStartWith('SQLiteConnection');
 
-    $query = ModelWithMySqlConnection::where('column', 1);
+    $model = new ModelWithMySqlConnection();
 
-    expect(class_basename($query->connection))->toStartWith('MySqlConnection');
+    expect(class_basename($model->getDB()->connection))->toStartWith('MySqlConnection');
 });
 
 it('can prefill the table name', function () {
-    $query = Model::where('column', 1);
+    $model = new Model();
+
+    expect($model->getDB()->from)->toBe('models');
+});
+
+it('can make the table name from the class name', function () {
+    $model = new ModelWithoutTableName();
+
 
-    expect($query->from)->toBe('models');
+    expect($model->getDB()->from)->toBe('model_without_table_names');
 });

src/Exceptions/MissingTableNameException.php

@@ -0,0 +1,23 @@
+<?php
+
+use GertjanRoke\LaravelDbModel\Tests\Models\Model;
+use Illuminate\Foundation\Testing\RefreshDatabase;
+use Illuminate\Support\Collection;
+
+uses(RefreshDatabase::class);
+
+it('will return it self when the method called returns the builder class', function () {
+    $model = new Model();
+
+    $returned = $model->where('column', true);
+
+    expect($returned)->toBeInstanceOf($model::class);
+});
+
+it('will return the collection when the db class returns a collection', function () {
+    $model = new Model();
+
+    $returned = $model->get();
+
+    expect($returned)->toBeInstanceOf(Collection::class);
+});

tests/ConstructorTest.php

@@ -2,9 +2,9 @@
 
 namespace GertjanRoke\LaravelDbModel\Tests\Models;
 
-use GertjanRoke\LaravelDbModel\DbModel;
+use GertjanRoke\LaravelDbModel\DBModel;
 
-class Model extends DbModel
+class Model extends DBModel
 {
     public $table = 'models';
 }

tests/MethodCallingTest.php

@@ -2,9 +2,9 @@
 
 namespace GertjanRoke\LaravelDbModel\Tests\Models;
 
-use GertjanRoke\LaravelDbModel\DbModel;
+use GertjanRoke\LaravelDbModel\DBModel;
 
-class ModelWithMySqlConnection extends DbModel
+class ModelWithMySqlConnection extends DBModel
 {
     public $table = 'model-with-mysql-connection';
 

tests/Models/Model.php

@@ -2,9 +2,9 @@
 
 namespace GertjanRoke\LaravelDbModel\Tests\Models;
 
-use GertjanRoke\LaravelDbModel\DbModel;
+use GertjanRoke\LaravelDbModel\DBModel;
 
-class ModelWithoutConnection extends DbModel
+class ModelWithoutConnection extends DBModel
 {
     public $table = 'model-without-connection';
 }

tests/Models/ModelWithMySqlConnection.php

@@ -2,9 +2,9 @@
 
 namespace GertjanRoke\LaravelDbModel\Tests\Models;
 
-use GertjanRoke\LaravelDbModel\DbModel;
+use GertjanRoke\LaravelDbModel\DBModel;
 
-class ModelWithoutTableName extends DbModel
+class ModelWithoutTableName extends DBModel
 {
     //
 }

tests/Models/ModelWithoutConnection.php

@@ -3,6 +3,7 @@
 namespace GertjanRoke\LaravelDbModel\Tests;
 
 use GertjanRoke\LaravelDbModel\LaravelDbModelServiceProvider;
+use GertjanRoke\LaravelDbModel\Tests\migrations\ModelTableMigration;
 use Orchestra\Testbench\TestCase as Orchestra;
 
 class TestCase extends Orchestra
@@ -17,5 +18,7 @@ protected function getPackageProviders($app)
     public function getEnvironmentSetUp($app)
     {
         config()->set('database.default', 'testing');
+
+        (new ModelTableMigration())->up();
     }
 }

tests/Models/ModelWithoutTableName.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace GertjanRoke\LaravelDbModel\Tests\migrations;
+
+use Illuminate\Database\Schema\Blueprint;
+use Illuminate\Support\Facades\Schema;
+
+class ModelTableMigration
+{
+    public function up()
+    {
+        Schema::create('models', function(Blueprint $table) {
+            $table->id();
+            $table->string('title')->nullable();
+            $table->timestamps();
+        });
+    }
+}