Test: add tests for block color property

jomei · Aug 8, 2022 · 26792ea · 26792ea
1 parent 2a903bc
commit 26792ea
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 13 deletions.
diff --git a/block_test.go b/block_test.go
@@ -80,12 +80,13 @@ func TestBlockClient(t *testing.T) {
 							Heading2: struct {
 								RichText []notionapi.RichText `json:"rich_text"`
 								Children notionapi.Blocks     `json:"children,omitempty"`
+								Color    string               `json:"color,omitempty"`
 							}{[]notionapi.RichText{
 								{
 									Type: notionapi.ObjectTypeText,
 									Text: notionapi.Text{Content: "Hello"},
 								},
-							}, nil,
+							}, nil, "",
 							},
 						},
 					},
@@ -121,6 +122,7 @@ func TestBlockClient(t *testing.T) {
 										PlainText: "AAAAAA",
 									},
 								},
+								Color: "blue",
 							},
 						},
 					},
@@ -238,6 +240,7 @@ func TestBlockClient(t *testing.T) {
 								Text: notionapi.Text{Content: "Hello"},
 							},
 						},
+						Color: notionapi.ColorYellow.String(),
 					},
 				},
 				want: &notionapi.ParagraphBlock{
@@ -259,6 +262,7 @@ func TestBlockClient(t *testing.T) {
 								PlainText:   "Hello",
 							},
 						},
+						Color: notionapi.ColorYellow.String(),
 					},
 				},
 				wantErr: false,
@@ -344,6 +348,7 @@ func TestBlockArrayUnmarshal(t *testing.T) {
 								Type:  "emoji",
 								Emoji: &emoji,
 							},
+							Color: notionapi.ColorBlue.String(),
 						},
 					},
 					&notionapi.Heading1Block{
@@ -369,6 +374,7 @@ func TestBlockArrayUnmarshal(t *testing.T) {
 									PlainText: "History 340",
 								},
 							},
+							Color: notionapi.ColorBrownBackground.String(),
 						},
 					},
 					&notionapi.ChildDatabaseBlock{
@@ -423,6 +429,7 @@ func TestBlockArrayUnmarshal(t *testing.T) {
 									PlainText: "Assignment Submission",
 								},
 							},
+							Color: notionapi.ColorDefault.String(),
 						},
 					},
 					&notionapi.ParagraphBlock{
@@ -448,6 +455,7 @@ func TestBlockArrayUnmarshal(t *testing.T) {
 									PlainText: "All essays and papers are due in lecture (due dates are listed on the schedule). No electronic copies will be accepted!",
 								},
 							},
+							Color: notionapi.ColorRed.String(),
 						},
 					},
 				},

diff --git a/page_test.go b/page_test.go
@@ -457,10 +457,7 @@ func TestPageCreateRequest_MarshallJSON(t *testing.T) {
 							Object: notionapi.ObjectTypeBlock,
 							Type:   notionapi.BlockTypeHeading2,
 						},
-						Heading2: struct {
-							RichText []notionapi.RichText `json:"rich_text"`
-							Children notionapi.Blocks     `json:"children,omitempty"`
-						}{
+						Heading2: notionapi.Heading{
 							RichText: []notionapi.RichText{
 								{
 									Type: notionapi.ObjectTypeText,

diff --git a/testdata/block_append_children.json b/testdata/block_append_children.json
@@ -36,10 +36,11 @@
             "plain_text": "AAAAAA",
             "href": null
           }
-        ]
+        ],
+        "color": "blue"
       }
     }
   ],
   "next_cursor": null,
   "has_more": false
-}
+}
diff --git a/testdata/block_array_unmarshal.json b/testdata/block_array_unmarshal.json
@@ -45,7 +45,8 @@
         "icon": {
             "type": "emoji",
             "emoji": "📌"
-        }
+        },
+        "color": "blue"
     }
 }, {
     "object": "block",
@@ -76,7 +77,8 @@
                 "color": "default"
             },
             "plain_text": "History 340"
-        }]
+        }],
+        "color":"brown_background"
     }
 }, {
     "object": "block",
@@ -140,7 +142,8 @@
                 "color": "default"
             },
             "plain_text": "Assignment Submission"
-        }]
+        }],
+        "color":"default"
     }
 }, {
     "object": "block",
@@ -171,6 +174,7 @@
                 "color": "default"
             },
             "plain_text": "All essays and papers are due in lecture (due dates are listed on the schedule). No electronic copies will be accepted!"
-        }]
+        }],
+        "color":"red"
     }
-}]
+}]
diff --git a/testdata/block_update.json b/testdata/block_update.json
@@ -25,6 +25,7 @@
         "plain_text": "Hello",
         "href": null
       }
-    ]
+    ],
+    "color": "yellow"
   }
 }