add tests

Author: sdebruyn

tests/functional/adapter/test_seed.py

@@ -1,5 +1,8 @@
+import os
+
 import pytest
 from dbt.tests.adapter.simple_seed.seeds import seeds__expected_sql
+from dbt.tests.adapter.simple_seed.test_seed import SeedConfigBase
 from dbt.tests.adapter.simple_seed.test_seed import TestBasicSeedTests as BaseBasicSeedTests
 from dbt.tests.adapter.simple_seed.test_seed import (
     TestSeedConfigFullRefreshOff as BaseSeedConfigFullRefreshOff,
@@ -20,7 +23,7 @@
     seeds__disabled_in_config_csv,
     seeds__enabled_in_config_csv,
 )
-from dbt.tests.util import get_connection
+from dbt.tests.util import get_connection, run_dbt
 
 from dbt.adapters.sqlserver import SQLServerAdapter
 
@@ -180,3 +183,43 @@ def setUp(self, project):
 
 class TestSeedSpecificFormatsSQLServer(BaseSeedSpecificFormats):
     pass
+
+
+class TestSeedBatchSizeMaxSQLServer(SeedConfigBase):
+    @pytest.fixture(scope="class")
+    def seeds(self, test_data_dir):
+        return {
+            "five_columns.csv": """
+seed_id,first_name,email,ip_address,birthday
+1,Larry,[email protected],69.135.206.194,2008-09-12 19:08:31
+2,Larry,[email protected],64.210.133.162,1978-05-09 04:15:14
+3,Anna,[email protected],168.104.64.114,2011-10-16 04:07:57
+        """
+        }
+
+    def test_max_batch_size(self, project, logs_dir):
+        run_dbt(["seed"])
+        with open(os.path.join(logs_dir, "dbt.log"), "r") as fp:
+            logs = "".join(fp.readlines())
+
+        assert "Inserting batches of 400 records" in logs
+
+
+class TestSeedBatchSizeCustomSQLServer(SeedConfigBase):
+    @pytest.fixture(scope="class")
+    def seeds(self, test_data_dir):
+        return {
+            "six_columns.csv": """
+seed_id,first_name,last_name,email,ip_address,birthday
+1,Larry,King,[email protected],69.135.206.194,2008-09-12 19:08:31
+2,Larry,Perkins,[email protected],64.210.133.162,1978-05-09 04:15:14
+3,Anna,Montgomery,[email protected],168.104.64.114,2011-10-16 04:07:57
+            """
+        }
+
+    def test_custom_batch_size(self, project, logs_dir):
+        run_dbt(["seed"])
+        with open(os.path.join(logs_dir, "dbt.log"), "r") as fp:
+            logs = "".join(fp.readlines())
+
+        assert "Inserting batches of 350 records" in logs