fix OwnFadSetDiscretizedActionGeneratorTest

was broken by 5e2035e
poseidon-fisheries · Nov 29, 2023 · 9d83c1a · 9d83c1a
1 parent 7b084ce
commit 9d83c1a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...a/uk/ac/ox/oxfish/fisher/purseseiner/planner/OwnFadSetDiscretizedActionGeneratorTest.java b/...a/uk/ac/ox/oxfish/fisher/purseseiner/planner/OwnFadSetDiscretizedActionGeneratorTest.java
@@ -76,6 +76,8 @@ void test() {
         when(fadManager.getFadMap()).thenReturn(fadMap);
         when(fadManager.getFisher()).thenReturn(fisher);
 
+        when(fisher.isAllowedAtSea()).thenReturn(true);
+        when(fisher.isAllowedToFishHere(any(), any())).thenReturn(true);
         when(fisher.getGear()).thenReturn(gear);
         when(fisher.grabState()).thenReturn(fishState);
         when(fishState.getMap()).thenReturn(map);
@@ -162,6 +164,8 @@ void filterByValue() {
         when(gear.isSafe(any())).thenReturn(true);
         final FadManager fadManager = gear.getFadManager();
         when(fadManager.getFisher()).thenReturn(fisher);
+        when(fisher.isAllowedAtSea()).thenReturn(true);
+        when(fisher.isAllowedToFishHere(any(), any())).thenReturn(true);
         when(fisher.getGear()).thenReturn(gear);
         when(fisher.grabState()).thenReturn(fishState);
         when(fishState.getMap()).thenReturn(map);
@@ -223,6 +227,8 @@ void banLocations() {
             );
         final FadManager fadManager = gear.getFadManager();
         when(fadManager.getFisher()).thenReturn(fisher);
+        when(fisher.isAllowedAtSea()).thenReturn(true);
+        when(fisher.isAllowedToFishHere(any(), any())).thenReturn(true);
         when(fisher.getGear()).thenReturn(gear);
         when(fisher.grabState()).thenReturn(fishState);
         when(fishState.getMap()).thenReturn(map);
@@ -276,4 +282,4 @@ void banLocations() {
         Assertions.assertEquals(1, initialOptions.size());
         Assertions.assertEquals(3, (int) initialOptions.get(0).getValue());
     }
-}
+}