Refactor test

Author: Barry-Zou

tests/Integration/LaborTest.php

@@ -20,10 +20,14 @@
 use Square\Labor\Shifts\Requests\UpdateShiftRequest;
 use Square\Labor\WorkweekConfigs\Requests\ListWorkweekConfigsRequest;
 use Square\SquareClient;
-use Square\Types\CreateTeamMemberRequest;
 use Square\Types\BreakType;
 use Square\Types\Money;
+use Square\TeamMembers\Requests\SearchTeamMembersRequest;
+use Square\Types\SearchTeamMembersQuery;
+use Square\Types\SearchTeamMembersFilter;
 use Square\Types\Shift;
+use Square\Types\ShiftQuery;
+use Square\Types\ShiftFilter;
 use Square\Types\ShiftWage;
 use Square\Types\TeamMember;
 
@@ -42,19 +46,35 @@ class LaborTest extends TestCase
     public static function setUpBeforeClass(): void
     {
         self::$client = Helpers::createClient();
-        self::$locationId = Helpers::createLocation(self::$client );
 
-        // Create team member for testing
-        $teamResponse = self::$client->teamMembers->create(new CreateTeamMemberRequest([
-            'idempotencyKey' => uniqid(),
-            'teamMember' => new TeamMember([
-                'givenName' => 'Sherlock',
-                'familyName' => 'Holmes',
-            ]),
+        // Get first available location
+        $locations = self::$client->locations->list();
+        $locationsList = $locations->getLocations();
+        if ($locationsList === null || count($locationsList) === 0) {
+            throw new RuntimeException('No locations available for testing');
+        }
+        $locationId = $locationsList[0]->getId();
+        if ($locationId === null) {
+            throw new RuntimeException('Location ID not present');
+        }
+        self::$locationId = $locationId;
+
+        // Get first available team member at this location
+        $teamMembersResponse = self::$client->teamMembers->search(new SearchTeamMembersRequest([
+            'query' => [
+                'filter' => [
+                    'locationIds' => [self::$locationId],
+                    'status' => 'ACTIVE',
+                ],
+            ],
         ]));
-        $memberId = $teamResponse->getTeamMember()?->getId();
-        if(!$memberId) {
-            throw new RuntimeException('Failed to create team member.');
+        $teamMembers = $teamMembersResponse->getTeamMembers();
+        if ($teamMembers === null || count($teamMembers) === 0) {
+            throw new RuntimeException('No team members available at location ' . self::$locationId);
+        }
+        $memberId = $teamMembers[0]->getId();
+        if ($memberId === null) {
+            throw new RuntimeException('Team member ID not present');
         }
         self::$memberId = $memberId;
 
@@ -96,10 +116,12 @@ public static function tearDownAfterClass(): void
         try {
             self::$client->labor->shifts->delete(new DeleteShiftsRequest(['id' => self::$shiftId]));
         } catch (Exception) {
+            // Test may have already deleted the shift
         }
         try {
             self::$client->labor->breakTypes->delete(new DeleteBreakTypesRequest(['id' => self::$breakId]));
         } catch (Exception) {
+            // Test may have already deleted the break
         }
     }
 
@@ -207,22 +229,38 @@ public function testUpdateShift(): void
      */
     public function testDeleteShift(): void
     {
-        // create team member
-        $teamMemberResponse = self::$client->teamMembers->create(new CreateTeamMemberRequest([
-            'idempotencyKey' => uniqid(),
-            'teamMember' => new TeamMember([
-                'givenName' => 'Sherlock',
-                'familyName' => 'Holmes',
+        // First search for existing shifts for this team member
+        $searchRequest = new SearchShiftsRequest([
+            'query' => new ShiftQuery([
+                'filter' => new ShiftFilter([
+                    'teamMemberIds' => [self::$memberId],
+                ]),
             ]),
-        ]));
+            'limit' => 100,
+        ]);
+        $existingShifts = self::$client->labor->shifts->search($searchRequest);
+
+        // Delete any existing shifts
+        if ($existingShifts->getShifts()) {
+            foreach ($existingShifts->getShifts() as $shift) {
+                if ($shift->getId()) {
+                    self::$client->labor->shifts->delete(new DeleteShiftsRequest(['id' => $shift->getId()]));
+                }
+            }
+        }
+
+        // Start the shift 10 seconds from now and end it 20 seconds from now
+        $startTime = new DateTime('+10 seconds');
+        $endTime = new DateTime('+20 seconds');
 
-        // create shift
+        // Create shift
         $shiftResponse = self::$client->labor->shifts->create(new CreateShiftRequest([
             'idempotencyKey' => uniqid(),
             'shift' => new Shift([
-                'startAt' => self::formatDateString(new DateTime()),
+                'startAt' => self::formatDateString($startTime),
+                'endAt' => self::formatDateString($endTime),
                 'locationId' => self::$locationId,
-                'teamMemberId' => $teamMemberResponse->getTeamMember()?->getId(),
+                'teamMemberId' => self::$memberId,
             ]),
         ]));
         $shift = $shiftResponse->getShift();
@@ -233,6 +271,10 @@ public function testDeleteShift(): void
         if(!$shiftId) {
             throw new RuntimeException('Shift ID is null.');
         }
+
+        // Add a small delay to ensure the shift is fully created
+        sleep(1);
+
         $response = self::$client->labor->shifts->delete(new DeleteShiftsRequest(['id' => $shiftId]));
         $this->assertNotNull($response);
     }
@@ -284,4 +326,4 @@ private static function formatDateString(DateTime $date): string
     {
         return $date->format(DateTimeInterface::ATOM);
     }
-}
+}