Implement TestDocsGenerateBigQuery test (#190)

* Implement TestDocsGenerateBigQuery test

* Remove legacy docs_generate tests

* Add -vv to get info on failing test

tests/functional/adapter/expected_stats.py

@@ -0,0 +1,62 @@
+from dbt.tests.util import AnyString, AnyFloat
+
+
+def bigquery_stats(is_table, partition=None, cluster=None):
+    stats = {}
+
+    if is_table:
+        stats.update(
+            {
+                "num_bytes": {
+                    "id": "num_bytes",
+                    "label": AnyString(),
+                    "value": AnyFloat(),
+                    "description": AnyString(),
+                    "include": True,
+                },
+                "num_rows": {
+                    "id": "num_rows",
+                    "label": AnyString(),
+                    "value": AnyFloat(),
+                    "description": AnyString(),
+                    "include": True,
+                },
+            }
+        )
+
+    if partition is not None:
+        stats.update(
+            {
+                "partitioning_type": {
+                    "id": "partitioning_type",
+                    "label": AnyString(),
+                    "value": partition,
+                    "description": AnyString(),
+                    "include": True,
+                }
+            }
+        )
+
+    if cluster is not None:
+        stats.update(
+            {
+                "clustering_fields": {
+                    "id": "clustering_fields",
+                    "label": AnyString(),
+                    "value": cluster,
+                    "description": AnyString(),
+                    "include": True,
+                }
+            }
+        )
+
+    has_stats = {
+        "id": "has_stats",
+        "label": "Has Stats?",
+        "value": bool(stats),
+        "description": "Indicates whether there are statistics for this table",
+        "include": False,
+    }
+    stats["has_stats"] = has_stats
+
+    return stats

tests/functional/adapter/test_basic.py

@@ -13,6 +13,9 @@
 from dbt.tests.adapter.basic.test_snapshot_timestamp import BaseSnapshotTimestamp
 from dbt.tests.adapter.basic.test_adapter_methods import BaseAdapterMethod
 from dbt.tests.adapter.basic.test_validate_connection import BaseValidateConnection
+from dbt.tests.adapter.basic.test_docs_generate import BaseDocsGenerate
+from dbt.tests.adapter.basic.expected_catalog import base_expected_catalog
+from tests.functional.adapter.expected_stats import bigquery_stats
 
 
 class TestSimpleMaterializationsBigQuery(BaseSimpleMaterializations):
@@ -53,8 +56,26 @@ class TestSnapshotCheckColsBigQuery(BaseSnapshotCheckCols):
 class TestSnapshotTimestampBigQuery(BaseSnapshotTimestamp):
     pass
 
+
 class TestBaseAdapterMethodBigQuery(BaseAdapterMethod):
     pass
 
+
 class TestBigQueryValidateConnection(BaseValidateConnection):
-    pass
+    pass
+
+
+class TestDocsGenerateBigQuery(BaseDocsGenerate):
+    @pytest.fixture(scope="class")
+    def expected_catalog(self, project):
+        return base_expected_catalog(
+            project,
+            role=None,
+            id_type="INT64",
+            text_type="STRING",
+            time_type="DATETIME",
+            view_type="view",
+            table_type="table",
+            model_stats=bigquery_stats(False),
+            seed_stats=bigquery_stats(True),
+        )