2020-03-20 16:17:58 +01:00
// Copyright 2016-2020, Pulumi Corporation.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Pulling out some of the repeated strings tokens into constants would harm readability, so we just ignore the
// goconst linter's warning.
//
// nolint: lll, goconst
package docs
import (
"testing"
2021-04-19 23:05:23 +02:00
"github.com/pulumi/pulumi/pkg/v3/codegen/internal/test"
2021-03-17 14:20:05 +01:00
"github.com/pulumi/pulumi/pkg/v3/codegen/schema"
2020-03-20 16:17:58 +01:00
"github.com/stretchr/testify/assert"
)
const (
unitTestTool = "Pulumi Resource Docs Unit Test"
providerPackage = "prov"
2020-05-09 01:25:28 +02:00
codeFence = "```"
2020-03-20 16:17:58 +01:00
)
var (
simpleProperties = map [ string ] schema . PropertySpec {
"stringProp" : {
Description : "A string prop." ,
TypeSpec : schema . TypeSpec {
Type : "string" ,
} ,
} ,
"boolProp" : {
Description : "A bool prop." ,
TypeSpec : schema . TypeSpec {
Type : "boolean" ,
} ,
} ,
}
// testPackageSpec represents a fake package spec for a Provider used for testing.
testPackageSpec schema . PackageSpec
)
func initTestPackageSpec ( t * testing . T ) {
t . Helper ( )
2021-07-14 01:41:40 +02:00
pythonMapCase := map [ string ] schema . RawMessage {
"python" : schema . RawMessage ( ` { "mapCase":false} ` ) ,
2020-03-20 16:17:58 +01:00
}
testPackageSpec = schema . PackageSpec {
Name : providerPackage ,
Description : "A fake provider package used for testing." ,
Meta : & schema . MetadataSpec {
ModuleFormat : "(.*)(?:/[^/]*)" ,
} ,
2020-09-16 22:47:40 +02:00
Types : map [ string ] schema . ComplexTypeSpec {
2020-03-20 16:17:58 +01:00
// Package-level types.
"prov:/getPackageResourceOptions:getPackageResourceOptions" : {
2020-09-16 22:47:40 +02:00
ObjectTypeSpec : schema . ObjectTypeSpec {
Description : "Options object for the package-level function getPackageResource." ,
Type : "object" ,
Properties : simpleProperties ,
} ,
2020-03-20 16:17:58 +01:00
} ,
// Module-level types.
"prov:module/getModuleResourceOptions:getModuleResourceOptions" : {
2020-09-16 22:47:40 +02:00
ObjectTypeSpec : schema . ObjectTypeSpec {
Description : "Options object for the module-level function getModuleResource." ,
Type : "object" ,
Properties : simpleProperties ,
} ,
2020-03-20 16:17:58 +01:00
} ,
"prov:module/ResourceOptions:ResourceOptions" : {
2020-09-16 22:47:40 +02:00
ObjectTypeSpec : schema . ObjectTypeSpec {
Description : "The resource options object." ,
Type : "object" ,
Properties : map [ string ] schema . PropertySpec {
"stringProp" : {
Description : "A string prop." ,
Language : pythonMapCase ,
TypeSpec : schema . TypeSpec {
Type : "string" ,
} ,
2020-03-20 16:17:58 +01:00
} ,
2020-09-16 22:47:40 +02:00
"boolProp" : {
Description : "A bool prop." ,
Language : pythonMapCase ,
TypeSpec : schema . TypeSpec {
Type : "boolean" ,
} ,
2020-03-20 16:17:58 +01:00
} ,
2020-09-16 22:47:40 +02:00
"recursiveType" : {
Description : "I am a recursive type." ,
Language : pythonMapCase ,
TypeSpec : schema . TypeSpec {
Ref : "#/types/prov:module/ResourceOptions:ResourceOptions" ,
} ,
2020-04-23 01:49:57 +02:00
} ,
} ,
2020-03-20 16:17:58 +01:00
} ,
} ,
"prov:module/ResourceOptions2:ResourceOptions2" : {
2020-09-16 22:47:40 +02:00
ObjectTypeSpec : schema . ObjectTypeSpec {
Description : "The resource options object." ,
Type : "object" ,
Properties : map [ string ] schema . PropertySpec {
"uniqueProp" : {
Description : "This is a property unique to this type." ,
Language : pythonMapCase ,
TypeSpec : schema . TypeSpec {
Type : "number" ,
} ,
2020-03-20 16:17:58 +01:00
} ,
} ,
} ,
} ,
} ,
Resources : map [ string ] schema . ResourceSpec {
"prov:module/resource:Resource" : {
2020-04-21 13:28:44 +02:00
ObjectTypeSpec : schema . ObjectTypeSpec {
2020-05-09 01:25:28 +02:00
Description : ` This is a module - level resource called Resource .
{ { % examples % } }
# # Example Usage
{ { % example % } }
# # # Basic Example
` + codeFence + ` typescript
// Some TypeScript code.
` + codeFence + `
` + codeFence + ` python
# Some Python code .
` + codeFence + `
{ { % / example % } }
{ { % example % } }
# # # Custom Sub - Domain Example
` + codeFence + ` typescript
// Some typescript code
` + codeFence + `
` + codeFence + ` python
# Some Python code .
` + codeFence + `
{ { % / example % } }
{ { % / examples % } }
2020-11-09 15:12:58 +01:00
# # Import
The import docs would be here
` + codeFence + ` sh
$ pulumi import prov : module / resource : Resource test test
` + codeFence + `
2020-05-09 01:25:28 +02:00
` ,
2020-04-21 13:28:44 +02:00
} ,
2020-03-20 16:17:58 +01:00
InputProperties : map [ string ] schema . PropertySpec {
"integerProp" : {
Description : "This is integerProp's description." ,
TypeSpec : schema . TypeSpec {
Type : "integer" ,
} ,
} ,
"stringProp" : {
Description : "This is stringProp's description." ,
TypeSpec : schema . TypeSpec {
Type : "string" ,
} ,
} ,
"boolProp" : {
Description : "A bool prop." ,
TypeSpec : schema . TypeSpec {
Type : "boolean" ,
} ,
} ,
"optionsProp" : {
TypeSpec : schema . TypeSpec {
Ref : "#/types/prov:module/ResourceOptions:ResourceOptions" ,
} ,
} ,
"options2Prop" : {
TypeSpec : schema . TypeSpec {
Ref : "#/types/prov:module/ResourceOptions2:ResourceOptions2" ,
} ,
} ,
2020-04-23 01:49:57 +02:00
"recursiveType" : {
Description : "I am a recursive type." ,
TypeSpec : schema . TypeSpec {
Ref : "#/types/prov:module/ResourceOptions:ResourceOptions" ,
} ,
} ,
2020-03-20 16:17:58 +01:00
} ,
} ,
2020-04-21 13:28:44 +02:00
"prov:/packageLevelResource:PackageLevelResource" : {
ObjectTypeSpec : schema . ObjectTypeSpec {
Description : "This is a package-level resource." ,
} ,
InputProperties : map [ string ] schema . PropertySpec {
"prop" : {
Description : "An input property." ,
TypeSpec : schema . TypeSpec {
Type : "string" ,
} ,
} ,
} ,
} ,
2020-03-20 16:17:58 +01:00
} ,
Functions : map [ string ] schema . FunctionSpec {
// Package-level Functions.
"prov:/getPackageResource:getPackageResource" : {
Description : "A package-level function." ,
Inputs : & schema . ObjectTypeSpec {
Description : "Inputs for getPackageResource." ,
Type : "object" ,
Properties : map [ string ] schema . PropertySpec {
"options" : {
TypeSpec : schema . TypeSpec {
Ref : "#/types/prov:/getPackageResourceOptions:getPackageResourceOptions" ,
} ,
} ,
} ,
} ,
Outputs : & schema . ObjectTypeSpec {
Description : "Outputs for getPackageResource." ,
Properties : simpleProperties ,
Type : "object" ,
} ,
} ,
// Module-level Functions.
"prov:module/getModuleResource:getModuleResource" : {
Description : "A module-level function." ,
Inputs : & schema . ObjectTypeSpec {
Description : "Inputs for getModuleResource." ,
Type : "object" ,
Properties : map [ string ] schema . PropertySpec {
"options" : {
TypeSpec : schema . TypeSpec {
Ref : "#/types/prov:module/getModuleResource:getModuleResource" ,
} ,
} ,
} ,
} ,
Outputs : & schema . ObjectTypeSpec {
Description : "Outputs for getModuleResource." ,
Properties : simpleProperties ,
Type : "object" ,
} ,
} ,
} ,
}
}
2020-04-21 13:28:44 +02:00
func getResourceFromModule ( resource string , mod * modContext ) * schema . Resource {
for _ , r := range mod . resources {
if resourceName ( r ) != resource {
continue
}
return r
}
return nil
}
2021-02-17 05:03:06 +01:00
func getFunctionFromModule ( function string , mod * modContext ) * schema . Function {
for _ , f := range mod . functions {
if tokenToName ( f . Token ) != function {
continue
}
return f
}
return nil
}
func TestFunctionHeaders ( t * testing . T ) {
initTestPackageSpec ( t )
schemaPkg , err := schema . ImportSpec ( testPackageSpec , nil )
assert . NoError ( t , err , "importing spec" )
tests := [ ] struct {
ExpectedTitleTag string
FunctionName string
ModuleName string
ExpectedMetaDesc string
} {
{
FunctionName : "getPackageResource" ,
// Empty string indicates the package-level root module.
ModuleName : "" ,
ExpectedTitleTag : "prov.getPackageResource" ,
ExpectedMetaDesc : "Documentation for the prov.getPackageResource function with examples, input properties, output properties, and supporting types." ,
} ,
{
FunctionName : "getModuleResource" ,
ModuleName : "module" ,
ExpectedTitleTag : "prov.module.getModuleResource" ,
ExpectedMetaDesc : "Documentation for the prov.module.getModuleResource function with examples, input properties, output properties, and supporting types." ,
} ,
}
modules := generateModulesFromSchemaPackage ( unitTestTool , schemaPkg )
for _ , test := range tests {
t . Run ( test . FunctionName , func ( t * testing . T ) {
mod , ok := modules [ test . ModuleName ]
if ! ok {
t . Fatalf ( "could not find the module %s in modules map" , test . ModuleName )
}
f := getFunctionFromModule ( test . FunctionName , mod )
if f == nil {
t . Fatalf ( "could not find %s in modules" , test . FunctionName )
}
h := mod . genFunctionHeader ( f )
assert . Equal ( t , test . ExpectedTitleTag , h . TitleTag )
assert . Equal ( t , test . ExpectedMetaDesc , h . MetaDesc )
} )
}
}
2020-04-21 13:28:44 +02:00
func TestResourceDocHeader ( t * testing . T ) {
initTestPackageSpec ( t )
schemaPkg , err := schema . ImportSpec ( testPackageSpec , nil )
assert . NoError ( t , err , "importing spec" )
tests := [ ] struct {
Name string
ExpectedTitleTag string
ResourceName string
ModuleName string
2020-12-17 19:44:57 +01:00
ExpectedMetaDesc string
2020-04-21 13:28:44 +02:00
} {
{
Name : "PackageLevelResourceHeader" ,
ResourceName : "PackageLevelResource" ,
// Empty string indicates the package-level root module.
ModuleName : "" ,
2021-02-17 05:03:06 +01:00
ExpectedTitleTag : "prov.PackageLevelResource" ,
ExpectedMetaDesc : "Documentation for the prov.PackageLevelResource resource with examples, input properties, output properties, lookup functions, and supporting types." ,
2020-04-21 13:28:44 +02:00
} ,
{
Name : "ModuleLevelResourceHeader" ,
ResourceName : "Resource" ,
ModuleName : "module" ,
2020-12-09 00:04:36 +01:00
ExpectedTitleTag : "prov.module.Resource" ,
2020-12-17 19:44:57 +01:00
ExpectedMetaDesc : "Documentation for the prov.module.Resource resource with examples, input properties, output properties, lookup functions, and supporting types." ,
2020-04-21 13:28:44 +02:00
} ,
}
2021-02-09 18:04:02 +01:00
modules := generateModulesFromSchemaPackage ( unitTestTool , schemaPkg )
2020-04-21 13:28:44 +02:00
for _ , test := range tests {
t . Run ( test . Name , func ( t * testing . T ) {
mod , ok := modules [ test . ModuleName ]
if ! ok {
t . Fatalf ( "could not find the module %s in modules map" , test . ModuleName )
}
r := getResourceFromModule ( test . ResourceName , mod )
if r == nil {
t . Fatalf ( "could not find %s in modules" , test . ResourceName )
}
h := mod . genResourceHeader ( r )
assert . Equal ( t , test . ExpectedTitleTag , h . TitleTag )
2020-12-17 19:44:57 +01:00
assert . Equal ( t , test . ExpectedMetaDesc , h . MetaDesc )
2020-04-21 13:28:44 +02:00
} )
}
}
2020-05-09 01:25:28 +02:00
func TestExamplesProcessing ( t * testing . T ) {
initTestPackageSpec ( t )
description := testPackageSpec . Resources [ "prov:module/resource:Resource" ] . Description
2020-06-18 21:32:15 +02:00
docInfo := decomposeDocstring ( description )
examplesSection := docInfo . examples
2020-11-09 15:12:58 +01:00
importSection := docInfo . importDetails
assert . NotEmpty ( t , importSection )
2020-05-09 01:25:28 +02:00
// The resource under test has two examples and both have TS and Python examples.
assert . Equal ( t , 2 , len ( examplesSection ) )
assert . Equal ( t , "### Basic Example" , examplesSection [ 0 ] . Title )
assert . Equal ( t , "### Custom Sub-Domain Example" , examplesSection [ 1 ] . Title )
expectedLangSnippets := [ ] string { "typescript" , "python" }
otherLangSnippets := [ ] string { "csharp" , "go" }
for _ , e := range examplesSection {
for _ , lang := range expectedLangSnippets {
_ , ok := e . Snippets [ lang ]
assert . True ( t , ok , "Could not find %s snippet" , lang )
}
for _ , lang := range otherLangSnippets {
snippet , ok := e . Snippets [ lang ]
assert . True ( t , ok , "Expected to find default placeholders for other languages" )
assert . Contains ( t , "Coming soon!" , snippet )
}
}
}
2021-04-19 23:05:23 +02:00
func generatePackage ( tool string , pkg * schema . Package , extraFiles map [ string ] [ ] byte ) ( map [ string ] [ ] byte , error ) {
return GeneratePackage ( tool , pkg )
}
func TestGeneratePackage ( t * testing . T ) {
2021-07-07 00:40:53 +02:00
test . TestSDKCodegen ( t , "docs" , generatePackage )
2021-04-19 23:05:23 +02:00
}