Add calls and API to get user's fleet, determined by rank #35

tests/PilotTest.php

@@ -0,0 +1,66 @@
+<?php
+
+use App\Services\UserService;
+use Tests\TestData;
+
+class PilotTest extends TestCase
+{
+    protected $userSvc;
+
+    public function setUp()
+    {
+        parent::setUp();
+        $this->userSvc = app(UserService::class);
+    }
+
+    public function testRankSubfleets()
+    {
+        # Add subfleets and aircraft, but also add another
+        # set of subfleets
+        $subfleetA = TestData::createSubfleetWithAircraft();
+        TestData::createSubfleetWithAircraft();
+
+        $rank = TestData::createRank(10, [$subfleetA['subfleet']->id]);
+
+        $user = factory(App\Models\User::class)->create([
+            'rank_id' => $rank->id,
+        ]);
+
+        $added_aircraft = $subfleetA['aircraft']->pluck('id');
+
+        $subfleets = $this->userSvc->getAllowableSubfleets($user);
+        $this->assertEquals(1, $subfleets->count());
+
+        $subfleet = $subfleets[0];
+        $all_aircraft = $subfleet->aircraft->pluck('id');
+        $this->assertEquals($added_aircraft, $all_aircraft);
+
+        /**
+         * Check via API
+         */
+        $resp = $this->get('/api/user/fleet', [], $user);
+        $body = $resp->json();
+
+        # Get the subfleet that's been added in
+        $subfleet_from_api = $body['data'][0];
+        $this->assertEquals($subfleet->id, $subfleet_from_api['id']);
+
+        # Get all the aircraft from that subfleet
+        $aircraft_from_api = collect($subfleet_from_api['aircraft'])->pluck('id');
+        $this->assertEquals($added_aircraft, $aircraft_from_api);
+
+        /**
+         * Check the user ID call
+         */
+        $resp = $this->get('/api/users/'.$user->id.'/fleet', [], $user);
+        $body = $resp->json();
+
+        # Get the subfleet that's been added in
+        $subfleet_from_api = $body['data'][0];
+        $this->assertEquals($subfleet->id, $subfleet_from_api['id']);
+
+        # Get all the aircraft from that subfleet
+        $aircraft_from_api = collect($subfleet_from_api['aircraft'])->pluck('id');
+        $this->assertEquals($added_aircraft, $aircraft_from_api);
+    }
+}

tests/SubfleetTest.php

@@ -1,7 +1,5 @@
 <?php
 
-use App\Models\Fare;
-use App\Models\Subfleet;
 
 class SubfleetTest extends TestCase
 {

tests/TestData.php

@@ -0,0 +1,54 @@
+<?php
+/**
+ *
+ */
+
+namespace Tests;
+
+class TestData
+{
+    /**
+     * Create a rank and associate the given subfleet IDs with it
+     * @param int $hours
+     * @param array $subfleet_ids
+     * @return mixed
+     */
+    public static function createRank($hours=0, $subfleet_ids=[])
+    {
+        $attrs = [];
+
+        if($hours === null) {
+            $attrs['hours'] = $hours;
+        }
+
+        $rank = factory(\App\Models\Rank::class)->create($attrs);
+        if(!empty($subfleet_ids)) {
+            $rank->subfleets()->syncWithoutDetaching($subfleet_ids);
+        }
+
+        return $rank;
+    }
+
+    /**
+     * Create a subfleet with a number of aircraft assigned
+     * @param null $aircraft_count
+     * @return mixed
+     */
+    public static function createSubfleetWithAircraft($aircraft_count = null)
+    {
+        $subfleet = factory(\App\Models\Subfleet::class)->create();
+
+        if($aircraft_count === null) {
+            $aircraft_count = \random_int(2, 10);
+        }
+
+        $aircraft = factory(\App\Models\Aircraft::class, $aircraft_count)->create([
+            'subfleet_id' => $subfleet->id
+        ]);
+
+        return [
+            'subfleet' => $subfleet,
+            'aircraft' => $aircraft,
+        ];
+    }
+}