BAEL-1248 google sheets api example (#3189)

* google sheets api example * add dependencies * rename test * update test * add comment * trigger build * trigger build * Formatting to standards * Getting rid of stray tabs * Applying formatting rules to source files as well * trigger build
2017-12-14 06:26:03 +02:00
parent 471ed5d992
commit 18c8747c98
5 changed files with 228 additions and 3 deletions
@@ -0,0 +1,140 @@
+package com.baeldung.google.sheets;
+
+import java.io.IOException;
+import java.security.GeneralSecurityException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import com.google.api.services.sheets.v4.Sheets;
+import com.google.api.services.sheets.v4.model.AppendValuesResponse;
+import com.google.api.services.sheets.v4.model.BatchGetValuesResponse;
+import com.google.api.services.sheets.v4.model.BatchUpdateSpreadsheetRequest;
+import com.google.api.services.sheets.v4.model.BatchUpdateValuesRequest;
+import com.google.api.services.sheets.v4.model.BatchUpdateValuesResponse;
+import com.google.api.services.sheets.v4.model.CopyPasteRequest;
+import com.google.api.services.sheets.v4.model.GridRange;
+import com.google.api.services.sheets.v4.model.Request;
+import com.google.api.services.sheets.v4.model.Spreadsheet;
+import com.google.api.services.sheets.v4.model.SpreadsheetProperties;
+import com.google.api.services.sheets.v4.model.UpdateSpreadsheetPropertiesRequest;
+import com.google.api.services.sheets.v4.model.UpdateValuesResponse;
+import com.google.api.services.sheets.v4.model.ValueRange;
+
+import static org.assertj.core.api.Assertions.*;
+
+public class GoogleSheetsIntegrationTest {
+
+    private static Sheets sheetsService;
+
+    // this id can be replaced with your spreadsheet id 
+    // otherwise be advised that multiple people may run this test and update the public spreadsheet
+    private static final String SPREADSHEET_ID = "1sILuxZUnyl_7-MlNThjt765oWshN3Xs-PPLfqYe4DhI";
+
+    @BeforeClass
+    public static void setup() throws GeneralSecurityException, IOException {
+        sheetsService = SheetsServiceUtil.getSheetsService();
+    }
+
+    @Test
+    public void whenWriteSheet_thenReadSheetOk() throws IOException {
+        ValueRange body = new ValueRange()
+          .setValues(Arrays.asList(
+                       Arrays.asList("Expenses January"), 
+                       Arrays.asList("books", "30"), 
+                       Arrays.asList("pens", "10"),
+                       Arrays.asList("Expenses February"), 
+                       Arrays.asList("clothes", "20"),
+                       Arrays.asList("shoes", "5")));
+        UpdateValuesResponse result = sheetsService.spreadsheets().values()
+          .update(SPREADSHEET_ID, "A1", body)
+          .setValueInputOption("RAW")
+          .execute();
+        
+        List<ValueRange> data = new ArrayList<>();
+        data.add(new ValueRange()
+          .setRange("D1")
+          .setValues(Arrays.asList(
+                       Arrays.asList("January Total", "=B2+B3"))));
+        data.add(new ValueRange()
+          .setRange("D4")
+          .setValues(Arrays.asList(
+                       Arrays.asList("February Total", "=B5+B6"))));
+
+        BatchUpdateValuesRequest batchBody = new BatchUpdateValuesRequest()
+          .setValueInputOption("USER_ENTERED")
+          .setData(data);
+        BatchUpdateValuesResponse batchResult =
+          sheetsService.spreadsheets().values()
+            .batchUpdate(SPREADSHEET_ID, batchBody)
+            .execute();
+        
+        List<String> ranges = Arrays.asList("E1","E4");
+        BatchGetValuesResponse readResult = 
+          sheetsService.spreadsheets().values()
+            .batchGet(SPREADSHEET_ID)
+            .setRanges(ranges)
+            .execute();
+      
+        ValueRange januaryTotal = readResult.getValueRanges().get(0);
+        assertThat(januaryTotal.getValues().get(0).get(0)).isEqualTo("40");
+
+        ValueRange febTotal = readResult.getValueRanges().get(1);
+        assertThat(febTotal.getValues().get(0).get(0)).isEqualTo("25");
+
+        ValueRange appendBody = new ValueRange()
+          .setValues(Arrays.asList(
+            Arrays.asList("Total", "=E1+E4")));
+        AppendValuesResponse appendResult =
+          sheetsService.spreadsheets().values()
+            .append(SPREADSHEET_ID, "A1", appendBody)
+            .setValueInputOption("USER_ENTERED")
+            .setInsertDataOption("INSERT_ROWS")
+            .setIncludeValuesInResponse(true)
+            .execute();
+
+        ValueRange total = appendResult.getUpdates().getUpdatedData();
+        assertThat(total.getValues().get(0).get(1)).isEqualTo("65");
+    }
+    
+
+    @Test
+    public void whenUpdateSpreadSheetTitle_thenOk() throws IOException {
+        
+        UpdateSpreadsheetPropertiesRequest updateRequest = new UpdateSpreadsheetPropertiesRequest()
+          .setFields("*")
+          .setProperties(new SpreadsheetProperties().setTitle("Expenses"));
+                
+        CopyPasteRequest copyRequest = new CopyPasteRequest()
+          .setSource(new GridRange().setSheetId(0)
+            .setStartColumnIndex(0).setEndColumnIndex(2)
+            .setStartRowIndex(0).setEndRowIndex(1))
+          .setDestination(new GridRange().setSheetId(1)
+            .setStartColumnIndex(0).setEndColumnIndex(2)
+            .setStartRowIndex(0).setEndRowIndex(1))
+          .setPasteType("PASTE_VALUES");
+        
+        List<Request> requests = new ArrayList<>();
+     
+        requests.add(new Request().setCopyPaste(copyRequest));
+        requests.add(new Request().setUpdateSpreadsheetProperties(updateRequest));
+
+        BatchUpdateSpreadsheetRequest body =
+                new BatchUpdateSpreadsheetRequest().setRequests(requests);
+
+        sheetsService.spreadsheets().batchUpdate(SPREADSHEET_ID, body).execute();
+    }
+    
+    @Test
+    public void whenCreateSpreadSheet_thenIdOk() throws IOException {
+        Spreadsheet spreadSheet = new Spreadsheet()
+          .setProperties(new SpreadsheetProperties().setTitle("My Spreadsheet"));
+        Spreadsheet result = sheetsService.spreadsheets().create(spreadSheet).execute();
+
+        assertThat(result.getSpreadsheetId()).isNotNull();   
+    }
+
+}