Using dedicated Notion instance for testing

Author: michaelhall28

conftest.py

@@ -22,9 +22,24 @@ def cover_url() -> str:
 
 @fixture
 def example_page_id() -> str:
-    return "e439b7a7296d45b98805f24c9cfc2115"
+    return "2cef2075b1dc803a8ec3d142ac105817"
+
+
+@fixture(scope="session")
+def database_id() -> str:
+    return "2cef2075b1dc8023b0ece0dbf9b278f7"
+
+
+@fixture(scope="session")
+def data_source_id1() ->  str:
+    return "2cef2075b1dc80bab834000b42b068d7"
+
+
+@fixture
+def data_source_id2() -> str:
+    return "2cef2075b1dc80048d8d000b1b75df8f"
 
 
 @fixture
-def example_page_id_2() -> str:
-    return "d5bce0a0fe6248d0a120c6c693d9b597"
+def block_id() -> str:
+    return "2cef2075b1dc80a59d6ad9ed97846ff8"

python_notion_api/async_api/tests/test_async_api.py

@@ -6,29 +6,25 @@
 from python_notion_api.async_api.notion_database import NotionDatabase
 from python_notion_api.async_api.notion_page import NotionPage
 
-TEST_DATABASE_ID = "401076f6c7c04ae796bf3e4c847361e1"
-TEST_DATA_SOURCE_ID = "924fbc0cb38f4a09ac2f967266f5c743"
-TEST_BLOCK_ID = "f572e889cd374edbbd15d8bf13174bbc"
-
 
 @async_fixture
-async def page(async_api, example_page_id_2):
-    return await async_api.get_page(page_id=example_page_id_2)
+async def page(async_api, example_page_id):
+    return await async_api.get_page(page_id=example_page_id)
 
 
 @async_fixture
-async def database(async_api):
-    return await async_api.get_database(database_id=TEST_DATABASE_ID)
+async def database(async_api, database_id):
+    return await async_api.get_database(database_id=database_id)
 
 
 @async_fixture
-async def data_source(async_api):
-    return await async_api.get_data_source(data_source_id=TEST_DATA_SOURCE_ID)
+async def data_source(async_api, data_source_id1):
+    return await async_api.get_data_source(data_source_id=data_source_id1)
 
 
 @async_fixture
-async def block(async_api):
-    return await async_api.get_block(block_id=TEST_BLOCK_ID)
+async def block(async_api, block_id):
+    return await async_api.get_block(block_id=block_id)
 
 
 @mark.asyncio

python_notion_api/async_api/tests/test_async_block.py

@@ -6,14 +6,12 @@
 from python_notion_api.async_api.notion_block import NotionBlock
 from python_notion_api.models import ParagraphBlock, RichTextObject
 
-TEST_BLOCK_ID = "f572e889cd374edbbd15d8bf13174bbc"
-
 
 @mark.asyncio
 class TestAsyncBlock:
     @async_fixture
-    async def block(self, async_api):
-        block = NotionBlock(block_id=TEST_BLOCK_ID, api=async_api)
+    async def block(self, async_api, block_id):
+        block = NotionBlock(block_id=block_id, api=async_api)
         await block.reload()
         return block
 

python_notion_api/async_api/tests/test_async_data_source.py

@@ -6,15 +6,13 @@
 from python_notion_api.async_api.notion_data_source import NotionDataSource
 from python_notion_api.async_api.notion_page import NotionPage
 
-TEST_DATA_SOURCE_ID = "924fbc0cb38f4a09ac2f967266f5c743"
-
 
 @mark.asyncio
 class TestAsyncDataSource:
     @async_fixture
-    async def data_source(self, async_api):
+    async def data_source(self, async_api, data_source_id1):
         data_source = NotionDataSource(
-            data_source_id=TEST_DATA_SOURCE_ID, api=async_api
+            data_source_id=data_source_id1, api=async_api
         )
         await data_source.reload()
         return data_source

python_notion_api/async_api/tests/test_async_database.py

@@ -4,18 +4,12 @@
 from python_notion_api.async_api.notion_database import NotionDatabase
 from python_notion_api.models.common import DataSourceObject
 
-TEST_DATABASE_ID = "401076f6c7c04ae796bf3e4c847361e1"
-TEST_DATA_SOURCE_IDS = [
-    "28d751c8a89180ed80c8000b05bd4bb1",
-    "924fbc0cb38f4a09ac2f967266f5c743",
-]
-
 
 @mark.asyncio
 class TestAsyncDatabase:
     @async_fixture
-    async def database(self, async_api):
-        database = NotionDatabase(database_id=TEST_DATABASE_ID, api=async_api)
+    async def database(self, async_api, database_id):
+        database = NotionDatabase(database_id=database_id, api=async_api)
         await database.reload()
         return database
 
@@ -33,16 +27,19 @@ async def test_get_title(self, database):
         title = database.title
         assert title is not None
 
-    async def test_get_data_sources(self, database):
+    async def test_get_data_sources(self, database, data_source_id1,
+                                    data_source_id2):
         data_sources = database.data_sources
         assert isinstance(data_sources, list)
 
         data_sources.sort(key=lambda x: x.data_source_id)
 
-        for i, data_source in enumerate(data_sources):
-            assert isinstance(data_source, DataSourceObject)
-            assert (
-                data_source.data_source_id.replace("-", "")
-                == TEST_DATA_SOURCE_IDS[i]
-            )
-            assert data_source.data_source_name is not None
+        assert len(data_sources) == 2
+        assert isinstance(data_sources[0], DataSourceObject)
+        assert data_sources[0].data_source_id.replace("-", "") == data_source_id2
+        assert data_sources[0].data_source_name is not None
+        assert isinstance(data_sources[1], DataSourceObject)
+        assert data_sources[1].data_source_id.replace(
+            "-", ""
+            ) == data_source_id1
+        assert data_sources[1].data_source_name is not None

python_notion_api/async_api/tests/test_async_page.py

@@ -12,8 +12,8 @@
 @mark.asyncio
 class TestAsyncPage:
     @async_fixture
-    async def page(self, async_api, example_page_id_2):
-        async_page = NotionPage(page_id=example_page_id_2, api=async_api)
+    async def page(self, async_api, example_page_id):
+        async_page = NotionPage(page_id=example_page_id, api=async_api)
         await async_page.reload()
         return async_page
 

python_notion_api/async_api/tests/test_properties.py

@@ -15,16 +15,15 @@
 )
 from python_notion_api.models.sorts import Sort
 
-TEST_DB = "401076f6c7c04ae796bf3e4c847361e1"
-TEST_DS = "924fbc0cb38f4a09ac2f967266f5c743"
+
 TEST_TITLE = f"API Test {datetime.now(UTC).isoformat()}"
 TEST_TEXT = "Test text is boring"
 TEST_NUMBER = 12.5
 TEST_SELECT = "foo"
 TEST_STATUS = "In progress"
 TEST_MULTI_SELECT = ["foo", "bar", "baz"]
 TEST_DATE = datetime.now()
-TEST_PEOPLE = ["fa9e1df9-7c24-427c-9c20-eac629565fe4"]
+TEST_PEOPLE = ["2ced872b-594c-8176-a765-0002860b6fdc"]
 TEST_FILES = [File(name="foo.pdf", url="http://example.com/file")]
 TEST_CHECKBOX = True
 TEST_URL = "http://example.com"
@@ -33,22 +32,22 @@
 
 
 @async_fixture
-async def database(async_api):
-    return await async_api.get_database(database_id=TEST_DB)
+async def database(async_api, database_id):
+    return await async_api.get_database(database_id=database_id)
 
 
 @async_fixture
-async def data_source(async_api):
-    return await async_api.get_data_source(data_source_id=TEST_DS)
+async def data_source(async_api, data_source_id1):
+    return await async_api.get_data_source(data_source_id=data_source_id1)
 
 
 @mark.asyncio
 class TestCore:
-    async def test_get_database(self, database):
-        assert database.database_id == TEST_DB
+    async def test_get_database(self, database, database_id):
+        assert database.database_id == database_id
 
-    async def test_get_data_source(self, data_source):
-        assert data_source.data_source_id == TEST_DS
+    async def test_get_data_source(self, data_source, data_source_id1):
+        assert data_source.data_source_id == data_source_id1
 
     async def test_create_empty_page(self, data_source):
         new_page = await data_source.create_page()
@@ -81,18 +80,18 @@ def api(cls):
         return cls.async_api
 
     @async_fixture(scope="class")
-    async def data_source(cls, api):
+    async def data_source(cls, api, data_source_id1):
         if not hasattr(cls, "async_data_source"):
             cls.async_data_source = await cls.async_api.get_data_source(
-                data_source_id=TEST_DS
+                data_source_id=data_source_id1
             )
         return cls.async_data_source
 
     @async_fixture(scope="class")
-    async def database(cls, api):
+    async def database(cls, api, database_id):
         if not hasattr(cls, "async_database"):
             cls.async_database = await cls.async_api.get_database(
-                database_id=TEST_DB
+                database_id=database_id
             )
         return cls.async_database
 
@@ -198,18 +197,18 @@ async def test_get_unique_id(self, page):
 
     async def test_get_by_id(self, page):
         await page.set("Email", TEST_EMAIL)
-        email = await page.get("%3E%5Ehh", cache=False)
+        email = await page.get("FEe%40", cache=False)
         assert email == TEST_EMAIL
 
     async def test_set_by_id(self, page):
-        await page.set("%3E%5Ehh", TEST_EMAIL)
+        await page.set("FEe%40", TEST_EMAIL)
         email = await page.get("Email", cache=False)
         assert email == TEST_EMAIL
 
     async def test_update(self, page):
         await page.update(
             properties={
-                "%3E%5Ehh": TEST_EMAIL,
+                "FEe%40": TEST_EMAIL,
                 "Phone": TEST_PHONE,
                 "Multi-select": None,
             }
@@ -233,17 +232,15 @@ async def test_reload(self, page):
 
 @mark.asyncio
 class TestRollups:
-    NUMBER_PAGE_ID = "25e800a118414575ab30a8dc42689b74"
-    DATE_PAGE_ID = "e38bb90faf8a436895f089fed2446cc6"
-    EMPTY_ROLLUP_PAGE_ID = "2b5efae5bad24df884b4f953e3788b64"
+    EMPTY_ROLLUP_PAGE_ID = "2cef2075b1dc80b5b67edc58426e92f2"
 
-    async def test_number_rollup(self, async_api):
-        number_page = await async_api.get_page(self.NUMBER_PAGE_ID)
+    async def test_number_rollup(self, async_api, example_page_id):
+        number_page = await async_api.get_page(example_page_id)
         num = await number_page.get("Number rollup")
         assert num == 10
 
-    async def test_date_rollup(self, async_api):
-        date_page = await async_api.get_page(self.DATE_PAGE_ID)
+    async def test_date_rollup(self, async_api, example_page_id):
+        date_page = await async_api.get_page(example_page_id)
         date = await date_page.get("Date rollup")
         assert isinstance(date.start, datetime)
 

python_notion_api/async_api/tests/test_query.py

@@ -7,7 +7,8 @@
 async def main():
     api = AsyncNotionAPI(access_token=os.environ["NOTION_TOKEN"])
 
-    db = await api.get_database(database_id="c0802577c79645e5af855f0ca46148b2")
+    db = await api.get_data_source(
+        data_source_id="2cef2075b1dc80bab834000b42b068d7")
 
     async for page in db.query():
         print(await page.get("title"))