From 9933a4988a4d755e37475b528ec3742c1b107ece Mon Sep 17 00:00:00 2001
From: victor-meng <56978073+victor-meng@users.noreply.github.com>
Date: Tue, 13 Oct 2020 15:49:30 -0700
Subject: [PATCH] Use SDK calls to read and update shared throughput for Tables
 API (#278)

RP does not supporting reading or updating database level throughput for Tables API so we have to switch back to using SDK calls for these operations.
---
 src/Common/dataAccess/readDatabaseOffer.ts |  8 +++-----
 src/Common/dataAccess/updateOffer.ts       | 15 ++++++++-------
 src/Explorer/Tree/Database.ts              |  6 +-----
 3 files changed, 12 insertions(+), 17 deletions(-)

diff --git a/src/Common/dataAccess/readDatabaseOffer.ts b/src/Common/dataAccess/readDatabaseOffer.ts
index 0aec98f51..2a6f67694 100644
--- a/src/Common/dataAccess/readDatabaseOffer.ts
+++ b/src/Common/dataAccess/readDatabaseOffer.ts
@@ -17,14 +17,12 @@ import { userContext } from "../../UserContext";
 export const readDatabaseOffer = async (
   params: DataModels.ReadDatabaseOfferParams
 ): Promise<DataModels.OfferWithHeaders> => {
-  if (userContext.defaultExperience === DefaultAccountExperienceType.Table) {
-    throw new Error("Reading database offer is not allowed for tables accounts");
-  }
-
   const clearMessage = logConsoleProgress(`Querying offer for database ${params.databaseId}`);
   let offerId = params.offerId;
   if (!offerId) {
-    offerId = await (window.authType === AuthType.AAD && !userContext.useSDKOperations
+    offerId = await (window.authType === AuthType.AAD &&
+    !userContext.useSDKOperations &&
+    userContext.defaultExperience !== DefaultAccountExperienceType.Table
       ? getDatabaseOfferIdWithARM(params.databaseId)
       : getDatabaseOfferIdWithSDK(params.databaseResourceId));
     if (!offerId) {
diff --git a/src/Common/dataAccess/updateOffer.ts b/src/Common/dataAccess/updateOffer.ts
index 733c183c4..3b276bb50 100644
--- a/src/Common/dataAccess/updateOffer.ts
+++ b/src/Common/dataAccess/updateOffer.ts
@@ -60,9 +60,14 @@ export const updateOffer = async (params: UpdateOfferParams): Promise<Offer> =>
 
   try {
     if (window.authType === AuthType.AAD && !userContext.useSDKOperations) {
-      updatedOffer = await (params.collectionId
-        ? updateCollectionOfferWithARM(params)
-        : updateDatabaseOfferWithARM(params));
+      if (params.collectionId) {
+        updatedOffer = await updateCollectionOfferWithARM(params);
+      } else if (userContext.defaultExperience === DefaultAccountExperienceType.Table) {
+        // update table's database offer with SDK since RP doesn't support it
+        updatedOffer = await updateOfferWithSDK(params);
+      } else {
+        updatedOffer = await updateDatabaseOfferWithARM(params);
+      }
     } else {
       updatedOffer = await updateOfferWithSDK(params);
     }
@@ -115,10 +120,6 @@ const updateCollectionOfferWithARM = async (params: UpdateOfferParams): Promise<
 };
 
 const updateDatabaseOfferWithARM = async (params: UpdateOfferParams): Promise<Offer> => {
-  if (userContext.defaultExperience === DefaultAccountExperienceType.Table) {
-    throw new Error("Updating database offer is not allowed for tables accounts");
-  }
-
   try {
     switch (userContext.defaultExperience) {
       case DefaultAccountExperienceType.DocumentDB:
diff --git a/src/Explorer/Tree/Database.ts b/src/Explorer/Tree/Database.ts
index 72fa38a00..63bcc4f2b 100644
--- a/src/Explorer/Tree/Database.ts
+++ b/src/Explorer/Tree/Database.ts
@@ -201,11 +201,7 @@ export default class Database implements ViewModels.Database {
   }
 
   public async loadOffer(): Promise<void> {
-    if (
-      !this.container.isServerlessEnabled() &&
-      this.container.defaultExperience() !== DefaultAccountExperienceType.Table &&
-      !this.offer()
-    ) {
+    if (!this.container.isServerlessEnabled() && !this.offer()) {
       const params: DataModels.ReadDatabaseOfferParams = {
         databaseId: this.id(),
         databaseResourceId: this.self