1212import com .squareup .square .labor .types .SearchShiftsRequest ;
1313import com .squareup .square .labor .types .UpdateBreakTypeRequest ;
1414import com .squareup .square .labor .types .UpdateShiftRequest ;
15- import com .squareup .square .types .Location ;
16- import com .squareup .square .types .SearchTeamMembersRequest ;
17- import com .squareup .square .types .SearchTeamMembersQuery ;
18- import com .squareup .square .types .SearchTeamMembersFilter ;
19- import com .squareup .square .types .TeamMemberStatus ;
2015import com .squareup .square .types .BreakType ;
2116import com .squareup .square .types .CreateBreakTypeResponse ;
2217import com .squareup .square .types .CreateShiftResponse ;
2318import com .squareup .square .types .DeleteBreakTypeResponse ;
2419import com .squareup .square .types .DeleteShiftResponse ;
2520import com .squareup .square .types .GetBreakTypeResponse ;
2621import com .squareup .square .types .GetShiftResponse ;
22+ import com .squareup .square .types .Location ;
2723import com .squareup .square .types .SearchShiftsResponse ;
24+ import com .squareup .square .types .SearchTeamMembersFilter ;
25+ import com .squareup .square .types .SearchTeamMembersQuery ;
26+ import com .squareup .square .types .SearchTeamMembersRequest ;
2827import com .squareup .square .types .Shift ;
2928import com .squareup .square .types .ShiftWage ;
3029import com .squareup .square .types .TeamMember ;
30+ import com .squareup .square .types .TeamMemberStatus ;
3131import com .squareup .square .types .UpdateBreakTypeResponse ;
3232import com .squareup .square .types .UpdateShiftResponse ;
3333import com .squareup .square .types .WorkweekConfig ;
34-
3534import java .time .OffsetDateTime ;
3635import java .time .format .DateTimeFormatter ;
3736import java .util .Collections ;
@@ -53,28 +52,30 @@ public class LaborTest {
5352 @ BeforeEach
5453 public void before () {
5554 client = TestClientFactory .create ();
56-
55+
5756 // Get first available location
58- List <Location > locations = client .locations ().list ().getLocations ()
59- .orElseThrow (() -> new RuntimeException ("No locations available" ));
57+ List <Location > locations = client .locations ()
58+ .list ()
59+ .getLocations ()
60+ .orElseThrow (() -> new RuntimeException ("No locations available" ));
6061 if (locations .isEmpty ()) {
6162 throw new RuntimeException ("No locations available for testing" );
6263 }
6364 locationId = locations .get (0 ).getId ().orElseThrow (() -> new RuntimeException ("Location ID not present" ));
64-
65+
6566 // Get first available team member at this location
66- List <TeamMember > teamMembers = client .teamMembers ().search (
67- SearchTeamMembersRequest .builder ()
68- .query (SearchTeamMembersQuery .builder ()
69- .filter (SearchTeamMembersFilter .builder ()
70- .locationIds (Collections .singletonList (locationId ))
71- .status (TeamMemberStatus .ACTIVE )
67+ List <TeamMember > teamMembers = client .teamMembers ()
68+ .search (SearchTeamMembersRequest .builder ()
69+ .query (SearchTeamMembersQuery .builder ()
70+ .filter (SearchTeamMembersFilter .builder ()
71+ .locationIds (Collections .singletonList (locationId ))
72+ .status (TeamMemberStatus .ACTIVE )
73+ .build ())
74+ .build ())
7275 .build ())
73- .build ())
74- .build ()
75- ).getTeamMembers ()
76- .orElseThrow (() -> new RuntimeException ("Failed to get team members" ));
77-
76+ .getTeamMembers ()
77+ .orElseThrow (() -> new RuntimeException ("Failed to get team members" ));
78+
7879 if (teamMembers .isEmpty ()) {
7980 throw new RuntimeException ("No team members available at location " + locationId );
8081 }
@@ -238,17 +239,17 @@ public void testDeleteShift() throws Exception {
238239 .limit (100 )
239240 .build ();
240241
241- SearchShiftsResponse existingShifts = client .labor ()
242- .shifts ()
243- .search (searchRequest );
242+ SearchShiftsResponse existingShifts = client .labor ().shifts ().search (searchRequest );
244243
245244 // Delete any existing shifts
246245 if (existingShifts .getShifts ().isPresent ()) {
247246 for (Shift existingShift : existingShifts .getShifts ().get ()) {
248247 if (existingShift .getId ().isPresent ()) {
249248 client .labor ()
250249 .shifts ()
251- .delete (DeleteShiftsRequest .builder ().id (existingShift .getId ().get ()).build ());
250+ .delete (DeleteShiftsRequest .builder ()
251+ .id (existingShift .getId ().get ())
252+ .build ());
252253 }
253254 }
254255 }
@@ -336,4 +337,4 @@ public void testListWorkweekConfigs() {
336337 .list (ListWorkweekConfigsRequest .builder ().build ());
337338 Assertions .assertFalse (response .getItems ().isEmpty ());
338339 }
339- }
340+ }
0 commit comments