tuned test to bump coverage

Signed-off-by: quobix <dave@quobix.com>
pb33f · Sep 21, 2023 · 3c9415b · 3c9415b
1 parent 96e7818
commit 3c9415b
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/datamodel/low/v3/create_document_test.go b/datamodel/low/v3/create_document_test.go
@@ -80,6 +80,64 @@ func TestCircularReferenceError(t *testing.T) {
 	assert.Len(t, err, 3)
 }
 
+func TestCircularReference_IgnoreArray(t *testing.T) {
+
+	spec := `openapi: 3.1.0
+components:
+  schemas:
+    ProductCategory:
+      type: "object"
+      properties:
+        name:
+          type: "string"
+        children:
+          type: "array"
+          items:
+            $ref: "#/components/schemas/ProductCategory"
+          description: "Array of sub-categories in the same format."
+      required:
+        - "name"
+        - "children"`
+
+	info, _ := datamodel.ExtractSpecInfo([]byte(spec))
+	circDoc, err := CreateDocumentFromConfig(info, &datamodel.DocumentConfiguration{
+		AllowFileReferences:           false,
+		AllowRemoteReferences:         false,
+		IgnoreArrayCircularReferences: true,
+	})
+	assert.NotNil(t, circDoc)
+	assert.Len(t, err, 0)
+}
+
+func TestCircularReference_IgnorePoly(t *testing.T) {
+
+	spec := `openapi: 3.1.0
+components:
+  schemas:
+    ProductCategory:
+      type: "object"
+      properties:
+        name:
+          type: "string"
+        children:
+          type: "object"
+          anyOf:
+            - $ref: "#/components/schemas/ProductCategory"
+          description: "Array of sub-categories in the same format."
+      required:
+        - "name"
+        - "children"`
+
+	info, _ := datamodel.ExtractSpecInfo([]byte(spec))
+	circDoc, err := CreateDocumentFromConfig(info, &datamodel.DocumentConfiguration{
+		AllowFileReferences:                 false,
+		AllowRemoteReferences:               false,
+		IgnorePolymorphicCircularReferences: true,
+	})
+	assert.NotNil(t, circDoc)
+	assert.Len(t, err, 0)
+}
+
 func BenchmarkCreateDocument_Stripe(b *testing.B) {
 	data, _ := os.ReadFile("../../../test_specs/stripe.yaml")
 	info, _ := datamodel.ExtractSpecInfo(data)