mirror of
https://github.com/netbirdio/gvisor.git
synced 2026-05-22 17:12:49 -07:00
Rahat MahmoodandgVisor bot
fee2c07728
Previously, the go-marshal-generated tests did not respect build tags. This can cause the test to unbuildable under some build configurations, as the original types the tests refer to may not be defined. This CL copies the build tags from the input files to the test, similar to the generated library; however test packages have an additional constraint. A test package cannot be totally empty (i.e. have no test/example/benchmark defined), otherwise the go compiler returns an error. To ensure the generated test package always contains a testable entity under all build configurations, we now emit an extra test file with no build tags that contains a single no-op example. PiperOrigin-RevId: 334496821
74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
// Copyright 2019 The gVisor Authors.
|
|
//
|
|
// 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.
|
|
|
|
// go_marshal is a code generation utility for automatically generating code to
|
|
// marshal go data structures to memory.
|
|
//
|
|
// This binary is typically run as part of the build process, and is invoked by
|
|
// the go_marshal bazel rule defined in defs.bzl.
|
|
//
|
|
// See README.md.
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"gvisor.dev/gvisor/tools/go_marshal/gomarshal"
|
|
)
|
|
|
|
var (
|
|
pkg = flag.String("pkg", "", "output package")
|
|
output = flag.String("output", "", "output file")
|
|
outputTest = flag.String("output_test", "", "output file for tests")
|
|
outputTestUnconditional = flag.String("output_test_unconditional", "", "output file for unconditional tests")
|
|
imports = flag.String("imports", "", "comma-separated list of extra packages to import in generated code")
|
|
)
|
|
|
|
func main() {
|
|
flag.Usage = func() {
|
|
fmt.Fprintf(os.Stderr, "Usage: %s <input go src files>\n", os.Args[0])
|
|
flag.PrintDefaults()
|
|
}
|
|
flag.Parse()
|
|
if len(flag.Args()) == 0 {
|
|
flag.Usage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *pkg == "" {
|
|
flag.Usage()
|
|
fmt.Fprint(os.Stderr, "Flag -pkg must be provided.\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
var extraImports []string
|
|
if len(*imports) > 0 {
|
|
// Note: strings.Split(s, sep) returns s if sep doesn't exist in s. Thus
|
|
// we check for an empty imports list to avoid emitting an empty string
|
|
// as an import.
|
|
extraImports = strings.Split(*imports, ",")
|
|
}
|
|
g, err := gomarshal.NewGenerator(flag.Args(), *output, *outputTest, *outputTestUnconditional, *pkg, extraImports)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := g.Run(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|