From 2bd0b223a68657fa13870e2dbe656cd622a78ac6 Mon Sep 17 00:00:00 2001
From: Raul Ochoa <rochoaf@gmail.com>
Date: Mon, 7 Jul 2014 10:36:51 +0200
Subject: [PATCH] CDB_SetUserQuotaInBytes returning set value instead of
 previous value.

---
 scripts-available/CDB_Quota.sql | 10 +---------
 test/CDB_QuotaTest_expect       |  4 ++--
 2 files changed, 3 insertions(+), 11 deletions(-)

diff --git a/scripts-available/CDB_Quota.sql b/scripts-available/CDB_Quota.sql
index 376bed0..fb718fd 100644
--- a/scripts-available/CDB_Quota.sql
+++ b/scripts-available/CDB_Quota.sql
@@ -84,26 +84,18 @@ CREATE OR REPLACE FUNCTION CDB_SetUserQuotaInBytes(schema_name text, bytes int8)
 RETURNS int8 AS
 $$
 DECLARE
-  current_quota int8;
-  schema_ok boolean;
   sql text;
 BEGIN
   IF cartodb.schema_exists(schema_name::text) = false THEN
     RAISE EXCEPTION 'Invalid schema name "%"', schema_name::text;
   END IF;
 
-  BEGIN
-    EXECUTE FORMAT('SELECT %I._CDB_UserQuotaInBytes();', schema_name::text) INTO current_quota;
-  EXCEPTION WHEN undefined_function THEN
-    current_quota := 0;
-  END;
-
   sql := 'CREATE OR REPLACE FUNCTION "' || schema_name::text || '"._CDB_UserQuotaInBytes() '
     || 'RETURNS int8 AS $X$ SELECT ' || bytes
     || '::int8 $X$ LANGUAGE sql IMMUTABLE';
   EXECUTE sql;
 
-  return current_quota;
+  return bytes;
 END
 $$
 LANGUAGE 'plpgsql' VOLATILE STRICT;
diff --git a/test/CDB_QuotaTest_expect b/test/CDB_QuotaTest_expect
index 492c8d7..83bb0e2 100644
--- a/test/CDB_QuotaTest_expect
+++ b/test/CDB_QuotaTest_expect
@@ -6,9 +6,9 @@ ERROR:  Quota exceeded by 3.9990234375KB
 0
 
 INSERT 0 1024
-0
-ERROR:  Quota exceeded by 103.9921875KB
 8
+ERROR:  Quota exceeded by 103.9921875KB
+0
 INSERT 0 1
 DROP TABLE
 SET