feat: Add compression option ZSTD. (#1890)

* Add ZSTD to compression types

* feat: adds tests re Compression types

* revise datatype from Enum to object

* adds license text and docstring

* change object back to enum datatype

* updates compression object comparison

* updates Compression class

* jsonify and sort the input and output for testing

* Update tests/unit/job/test_extract.py

* moved json import statement

* removed enums test and file

---------

Co-authored-by: Ethan Steinberg <[email protected]>
Co-authored-by: Tim Sweña (Swast) <[email protected]>

Author: 3 people

google/cloud/bigquery/enums.py

@@ -22,7 +22,7 @@ class AutoRowIDs(enum.Enum):
     GENERATE_UUID = enum.auto()
 
 
-class Compression(object):
+class Compression(str, enum.Enum):
     """The compression type to use for exported files. The default value is
     :attr:`NONE`.
 
@@ -39,6 +39,9 @@ class Compression(object):
     SNAPPY = "SNAPPY"
     """Specifies SNAPPY format."""
 
+    ZSTD = "ZSTD"
+    """Specifies ZSTD format."""
+
     NONE = "NONE"
     """Specifies no compression."""
 

tests/unit/job/test_extract.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import json
 from unittest import mock
 
 from ..helpers import make_connection
@@ -45,9 +46,8 @@ def test_to_api_repr(self):
         config.print_header = False
         config._properties["extract"]["someNewField"] = "some-value"
         config.use_avro_logical_types = True
-        resource = config.to_api_repr()
-        self.assertEqual(
-            resource,
+        resource = json.dumps(config.to_api_repr(), sort_keys=True)
+        expected = json.dumps(
             {
                 "extract": {
                     "compression": "SNAPPY",
@@ -58,6 +58,12 @@ def test_to_api_repr(self):
                     "useAvroLogicalTypes": True,
                 }
             },
+            sort_keys=True,
+        )
+
+        self.assertEqual(
+            resource,
+            expected,
         )
 
     def test_from_api_repr(self):