func (t *TestCase) GoCode() string { var buf bytes.Buffer if err := tplTestCase.Execute(&buf, t); err != nil { panic(err) } return util.GoFmt(buf.String()) }
func (s *Shape) GoCode() string { code := s.Docstring() + "type " + s.ShapeName + " " switch s.Type { case "structure": code += "struct {\n" for _, n := range s.MemberNames() { m := s.MemberRefs[n] code += m.Docstring() if (m.Streaming || m.Shape.Streaming) && s.Payload == n { rtype := "io.ReadSeeker" if len(s.refs) > 1 { rtype = "aws.ReaderSeekCloser" } else if strings.HasSuffix(s.ShapeName, "Output") { rtype = "io.ReadCloser" } s.API.imports["io"] = true code += n + " " + rtype + " " + m.GoTags(false, s.IsRequired(n)) + "\n\n" } else { code += n + " " + m.GoType() + " " + m.GoTags(false, s.IsRequired(n)) + "\n\n" } } metaStruct := "metadata" + s.ShapeName ref := &ShapeRef{ShapeName: s.ShapeName, API: s.API, Shape: s} code += "\n" + metaStruct + " `json:\"-\", xml:\"-\"`\n" code += "}\n\n" code += "type " + metaStruct + " struct {\n" code += "SDKShapeTraits bool " + ref.GoTags(true, false) code += "}" default: panic("Cannot generate toplevel shape for " + s.Type) } return util.GoFmt(code) }
// Returns a string representing the Operation's interface{} functional signature. func (o *Operation) InterfaceSignature() string { var buf bytes.Buffer err := tplInfSig.Execute(&buf, o) if err != nil { panic(err) } return strings.TrimSpace(util.GoFmt(buf.String())) }
func (o *Operation) GoCode() string { var buf bytes.Buffer err := tplOperation.Execute(&buf, o) if err != nil { panic(err) } return strings.TrimSpace(util.GoFmt(buf.String())) }
func (a *API) APIGoCode() string { a.resetImports() a.imports["sync"] = true var buf bytes.Buffer err := tplAPI.Execute(&buf, a) if err != nil { panic(err) } code := a.importsGoCode() + strings.TrimSpace(buf.String()) return util.GoFmt(code) }
func (t *TestSuite) write() { _, d, _, _ := runtime.Caller(1) file := filepath.Join(path.Dir(d), "..", "..", "..", "service", t.PackageName, "integration_test.go") var buf bytes.Buffer if err := tplTestSuite.Execute(&buf, t); err != nil { panic(err) } b := []byte(util.GoFmt(buf.String())) ioutil.WriteFile(file, b, 0644) }
func (a *API) ServiceGoCode() string { a.resetImports() a.imports["github.com/datacratic/aws-sdk-go/internal/signer/v4"] = true a.imports["github.com/datacratic/aws-sdk-go/internal/protocol/"+a.ProtocolPackage()] = true var buf bytes.Buffer err := tplService.Execute(&buf, a) if err != nil { panic(err) } code := a.importsGoCode() + buf.String() return util.GoFmt(code) }
func (t *TestSuite) TestSuite() string { var buf bytes.Buffer t.title = reStripSpace.ReplaceAllStringFunc(t.Description, func(x string) string { return strings.ToUpper(x[1:]) }) t.title = regexp.MustCompile(`\W`).ReplaceAllString(t.title, "") for idx, c := range t.Cases { c.TestSuite = t buf.WriteString(c.TestCase(idx) + "\n") } return util.GoFmt(buf.String()) }
func (a *API) ExampleGoCode() string { exs := []string{} for _, o := range a.OperationList() { exs = append(exs, o.Example()) } code := fmt.Sprintf("import (\n%q\n%q\n%q\n\n%q\n%q\n%q\n)\n\n"+ "var _ time.Duration\nvar _ bytes.Buffer\n\n%s", "bytes", "fmt", "time", "github.com/datacratic/aws-sdk-go/aws", "github.com/datacratic/aws-sdk-go/aws/awsutil", "github.com/datacratic/aws-sdk-go/service/"+a.PackageName(), strings.Join(exs, "\n\n"), ) return util.GoFmt(code) }
func (i *TestCase) TestCase(idx int) string { var buf bytes.Buffer opName := i.API.StructName() + i.TestSuite.title + "Case" + strconv.Itoa(idx+1) if i.Params != nil { // input test // query test should sort body as form encoded values switch i.API.Metadata.Protocol { case "query", "ec2": m, _ := url.ParseQuery(i.InputTest.Body) i.InputTest.Body = m.Encode() case "rest-xml": i.InputTest.Body = util.SortXML(bytes.NewReader([]byte(i.InputTest.Body))) case "json", "rest-json": i.InputTest.Body = strings.Replace(i.InputTest.Body, " ", "", -1) } input := tplInputTestCaseData{ TestCase: i, OpName: strings.ToUpper(opName[0:1]) + opName[1:], ParamsString: helpers.ParamsStructFromJSON(i.Params, i.Given.InputRef.Shape, false), } if err := tplInputTestCase.Execute(&buf, input); err != nil { panic(err) } } else { output := tplOutputTestCaseData{ TestCase: i, Body: fmt.Sprintf("%q", i.OutputTest.Body), OpName: strings.ToUpper(opName[0:1]) + opName[1:], Assertions: utilassert.GenerateAssertions(i.Data, i.Given.OutputRef.Shape, "out"), } if err := tplOutputTestCase.Execute(&buf, output); err != nil { panic(err) } } return util.GoFmt(buf.String()) }
// Returns the go code for the service's API operations as an interface{}. // Assumes that the interface is being created in a different package than // the service API's package. func (a *API) InterfaceGoCode() string { a.resetImports() a.imports = map[string]bool{ "github.com/datacratic/aws-sdk-go/service/" + a.PackageName(): true, } var buf bytes.Buffer err := tplInterface.Execute(&buf, &struct { StructName string OperationList []*Operation }{ StructName: a.StructName() + "API", OperationList: a.OperationList(), }) if err != nil { panic(err) } code := a.importsGoCode() + strings.TrimSpace(buf.String()) return util.GoFmt(code) }
func GenerateTestSuite(filename string) string { inout := "Input" if strings.Contains(filename, "output/") { inout = "Output" } var suites []TestSuite f, err := os.Open(filename) if err != nil { panic(err) } err = json.NewDecoder(f).Decode(&suites) if err != nil { panic(err) } var buf bytes.Buffer buf.WriteString("package " + suites[0].ProtocolPackage() + "_test\n\n") var innerBuf bytes.Buffer innerBuf.WriteString("//\n// Tests begin here\n//\n\n\n") for i, suite := range suites { svcPrefix := inout + "Service" + strconv.Itoa(i+1) suite.API.Metadata.ServiceAbbreviation = svcPrefix + "ProtocolTest" suite.API.Operations = map[string]*api.Operation{} for idx, c := range suite.Cases { c.Given.ExportedName = svcPrefix + "TestCaseOperation" + strconv.Itoa(idx+1) suite.API.Operations[c.Given.ExportedName] = c.Given } suite.API.NoInflections = true // don't require inflections suite.API.NoInitMethods = true // don't generate init methods suite.API.Setup() suite.API.Metadata.EndpointPrefix = suite.API.PackageName() for n, s := range suite.API.Shapes { s.Rename(svcPrefix + "TestShape" + n) } svcCode := addImports(suite.API.ServiceGoCode()) if i == 0 { importMatch := reImportRemoval.FindStringSubmatch(svcCode) buf.WriteString(importMatch[0] + "\n\n") buf.WriteString(preamble + "\n\n") } svcCode = removeImports(svcCode) svcCode = strings.Replace(svcCode, "func New(", "func New"+suite.API.StructName()+"(", -1) buf.WriteString(svcCode + "\n\n") apiCode := removeImports(suite.API.APIGoCode()) apiCode = strings.Replace(apiCode, "var oprw sync.Mutex", "", -1) apiCode = strings.Replace(apiCode, "oprw.Lock()", "", -1) apiCode = strings.Replace(apiCode, "defer oprw.Unlock()", "", -1) buf.WriteString(apiCode + "\n\n") innerBuf.WriteString(suite.TestSuite() + "\n") } return util.GoFmt(buf.String() + innerBuf.String()) }
func ParamsStructFromJSON(value interface{}, shape *api.Shape, prefixPackageName bool) string { f := paramFiller{prefixPackageName: prefixPackageName} return util.GoFmt(f.paramsStructAny(value, shape)) }