Fix test manage state permission (#1334)

AzBuilder · Sep 25, 2024 · 13cafef · 13cafef
1 parent de4b739
commit 13cafef
Showing 1 changed file with 41 additions and 2 deletions.
diff --git a/api/src/test/java/org/terrakube/api/LocalStorageTests.java b/api/src/test/java/org/terrakube/api/LocalStorageTests.java
@@ -2,11 +2,16 @@
 
 import org.apache.commons.io.FileUtils;
 import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
+import org.terrakube.api.repository.TeamRepository;
+import org.terrakube.api.rs.team.Team;
 
 import java.io.File;
 import java.io.IOException;
 import java.nio.charset.Charset;
+import java.util.Optional;
+import java.util.UUID;
 
 import static io.restassured.RestAssured.given;
 
@@ -16,19 +21,27 @@ public class LocalStorageTests extends ServerApplicationTests {
     private static final String STATE_DIRECTORY = "%s/.terraform-spring-boot/local/state/%s/%s/%s/%s/terraformLibrary.tfPlan";
     private static final String STATE_DIRECTORY_JSON = "%s/.terraform-spring-boot/local/state/%s/%s/state/%s.json";
 
+    @Autowired
+    TeamRepository teamRepository;
+
     @Test
     void testLocalStorageJSON() throws IOException {
         FileUtils.writeStringToFile(
                 new File(
-                        String.format(STATE_DIRECTORY_JSON, FileUtils.getUserDirectoryPath(), "1", "1", "1")),
+                        String.format(STATE_DIRECTORY_JSON, FileUtils.getUserDirectoryPath(), "d9b58bd3-f3fc-4056-a026-1163297e80a8", "5ed411ca-7ab8-4d2f-b591-02d0d5788afc", "1")),
                 "SAMPLE",
                 Charset.defaultCharset().toString()
         );
 
+        Optional<Team> teamOptional = teamRepository.findById(UUID.fromString("58529721-425e-44d7-8b0d-1d515043c2f7"));
+        Team team = teamOptional.get();
+        team.setManageState(true);
+        teamRepository.save(team);
+
         given()
                 .headers("Authorization", "Bearer " + generatePAT("TERRAKUBE_DEVELOPERS"))
                 .when()
-                .get("/tfstate/v1/organization/1/workspace/1/state/1.json")
+                .get("/tfstate/v1/organization/d9b58bd3-f3fc-4056-a026-1163297e80a8/workspace/5ed411ca-7ab8-4d2f-b591-02d0d5788afc/state/1.json")
                 .then()
                 .assertThat()
                 .log()
@@ -37,6 +50,32 @@ void testLocalStorageJSON() throws IOException {
 
     }
 
+    @Test
+    void testLocalStorageJSONWithoutManageStatePermission() throws IOException {
+        FileUtils.writeStringToFile(
+                new File(
+                        String.format(STATE_DIRECTORY_JSON, FileUtils.getUserDirectoryPath(), "d9b58bd3-f3fc-4056-a026-1163297e80a8", "5ed411ca-7ab8-4d2f-b591-02d0d5788afc", "1")),
+                "SAMPLE",
+                Charset.defaultCharset().toString()
+        );
+
+        Optional<Team> teamOptional = teamRepository.findById(UUID.fromString("58529721-425e-44d7-8b0d-1d515043c2f7"));
+        Team team = teamOptional.get();
+        team.setManageState(false);
+        teamRepository.save(team);
+
+        given()
+                .headers("Authorization", "Bearer " + generatePAT("TERRAKUBE_DEVELOPERS"))
+                .when()
+                .get("/tfstate/v1/organization/d9b58bd3-f3fc-4056-a026-1163297e80a8/workspace/5ed411ca-7ab8-4d2f-b591-02d0d5788afc/state/1.json")
+                .then()
+                .assertThat()
+                .log()
+                .all()
+                .statusCode(HttpStatus.FORBIDDEN.value());
+
+    }
+
     @Test
     void testLocalStorageBinaryState() throws IOException {
         FileUtils.writeStringToFile(