| // Copyright 2025 The Dawn & Tint Authors | |
| // | |
| // Redistribution and use in source and binary forms, with or without | |
| // modification, are permitted provided that the following conditions are met: | |
| // | |
| // 1. Redistributions of source code must retain the above copyright notice, this | |
| // list of conditions and the following disclaimer. | |
| // | |
| // 2. Redistributions in binary form must reproduce the above copyright notice, | |
| // this list of conditions and the following disclaimer in the documentation | |
| // and/or other materials provided with the distribution. | |
| // | |
| // 3. Neither the name of the copyright holder nor the names of its | |
| // contributors may be used to endorse or promote products derived from | |
| // this software without specific prior written permission. | |
| // | |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | |
| // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
| // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE | |
| // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
| // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
| // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER | |
| // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, | |
| // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| package main | |
| import ( | |
| "bytes" | |
| "errors" | |
| "flag" | |
| "fmt" | |
| "runtime" | |
| "testing" | |
| "dawn.googlesource.com/dawn/tools/src/fileutils" | |
| "dawn.googlesource.com/dawn/tools/src/oswrapper" | |
| "github.com/stretchr/testify/require" | |
| ) | |
| func TestGatherWgslFiles(t *testing.T) { | |
| tests := []struct { | |
| name string | |
| inputsDir string | |
| outDir string | |
| setupFS func(t *testing.T, fs oswrapper.FSTestOSWrapper) | |
| wantErr bool | |
| verify func(t *testing.T, fs oswrapper.FSTestOSWrapper) | |
| }{ | |
| { | |
| name: "Basic copy", | |
| inputsDir: "/in", | |
| outDir: "/out", | |
| setupFS: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| require.NoError(t, fs.MkdirAll("/in/subdir", 0777)) | |
| require.NoError(t, fs.MkdirAll("/out", 0777)) | |
| require.NoError(t, fs.WriteFile("/in/a.wgsl", []byte("shader a"), 0666)) | |
| require.NoError(t, fs.WriteFile("/in/b.txt", []byte("not a shader"), 0666)) | |
| require.NoError(t, fs.WriteFile("/in/c.expected.wgsl", []byte("should be ignored"), 0666)) | |
| require.NoError(t, fs.WriteFile("/in/subdir/d.wgsl", []byte("shader d"), 0666)) | |
| }, | |
| wantErr: false, | |
| verify: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| // Check that expected files were created with correct content | |
| contentA, err := fs.ReadFile("/out/a.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "shader a", string(contentA)) | |
| contentD, err := fs.ReadFile("/out/subdir_d.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "shader d", string(contentD)) | |
| // Check that other files were not copied | |
| _, err = fs.Stat("/out/b.txt") | |
| require.Error(t, err, "b.txt should not have been copied") | |
| _, err = fs.Stat("/out/c.expected.wgsl") | |
| require.Error(t, err, "c.expected.wgsl should not have been copied") | |
| }, | |
| }, | |
| { | |
| name: "Complex subdirectories", | |
| inputsDir: "/in", | |
| outDir: "/out", | |
| setupFS: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| require.NoError(t, fs.MkdirAll("/in/a/b", 0777)) | |
| require.NoError(t, fs.MkdirAll("/out", 0777)) | |
| require.NoError(t, fs.WriteFile("/in/a/b/c.wgsl", []byte("shader c"), 0666)) | |
| }, | |
| wantErr: false, | |
| verify: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| content, err := fs.ReadFile("/out/a_b_c.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "shader c", string(content)) | |
| }, | |
| }, | |
| { | |
| name: "Empty input directory", | |
| inputsDir: "/in", | |
| outDir: "/out", | |
| setupFS: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| require.NoError(t, fs.MkdirAll("/in", 0777)) | |
| require.NoError(t, fs.MkdirAll("/out", 0777)) | |
| }, | |
| wantErr: false, | |
| verify: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| isEmpty, err := fileutils.IsEmptyDir("/out", fs) | |
| require.NoError(t, err) | |
| require.True(t, isEmpty) | |
| }, | |
| }, | |
| { | |
| name: "Output directory does not exist", | |
| inputsDir: "/in", | |
| outDir: "/out", | |
| setupFS: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| require.NoError(t, fs.MkdirAll("/in", 0777)) | |
| require.NoError(t, fs.WriteFile("/in/a.wgsl", []byte("shader a"), 0666)) | |
| }, | |
| wantErr: false, | |
| verify: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| // Check that output directory was created | |
| info, err := fs.Stat("/out") | |
| require.NoError(t, err) | |
| require.True(t, info.IsDir()) | |
| // Check that file was copied | |
| content, err := fs.ReadFile("/out/a.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "shader a", string(content)) | |
| }, | |
| }, | |
| { | |
| name: "Input directory does not exist", | |
| inputsDir: "/nonexistent", | |
| outDir: "/out", | |
| setupFS: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| require.NoError(t, fs.MkdirAll("/out", 0777)) | |
| }, | |
| wantErr: true, | |
| }, | |
| { | |
| name: "Copyright notice stripping", | |
| inputsDir: "/in", | |
| outDir: "/out", | |
| setupFS: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| require.NoError(t, fs.MkdirAll("/in", 0777)) | |
| require.NoError(t, fs.MkdirAll("/out", 0777)) | |
| // Case 1: No comments and not copyright | |
| require.NoError(t, fs.WriteFile("/in/case1.wgsl", []byte("fn main() {}"), 0666)) | |
| // Case 2: Comments but no copyright | |
| require.NoError(t, fs.WriteFile("/in/case2.wgsl", []byte("// This is a comment\nfn main() {}"), 0666)) | |
| // Case 3: Copyright but no other comments | |
| copyrightNoComments := []byte("// Copyright 2020 The Dawn & Tint Authors\n//\n// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\nfn main() {}") | |
| require.NoError(t, fs.WriteFile("/in/case3.wgsl", copyrightNoComments, 0666)) | |
| // Case 4: Copyright and other comments | |
| copyrightAndComments := []byte("// Copyright 2020 The Dawn & Tint Authors\n//\n// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n// This is a comment\nfn main() {}") | |
| require.NoError(t, fs.WriteFile("/in/case4.wgsl", copyrightAndComments, 0666)) | |
| }, | |
| wantErr: false, | |
| verify: func(t *testing.T, fs oswrapper.FSTestOSWrapper) { | |
| // For all confirm old _no _comments files are not produced | |
| // Case 1 validation: Copied as-is | |
| content1, err := fs.ReadFile("/out/case1.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "fn main() {}", string(content1)) | |
| _, err = fs.Stat("/out/case1_no_comments.wgsl") | |
| require.Error(t, err) | |
| // Case 2 validation: Copied as-is | |
| content2, err := fs.ReadFile("/out/case2.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "// This is a comment\nfn main() {}", string(content2)) | |
| _, err = fs.Stat("/out/case2_no_comments.wgsl") | |
| require.Error(t, err) | |
| // Case 3 validation: Copied with copyright block removed | |
| content3, err := fs.ReadFile("/out/case3.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "fn main() {}", string(content3)) | |
| _, err = fs.Stat("/out/case3_no_comments.wgsl") | |
| require.Error(t, err) | |
| // Case 4 validation: Copied with copyright block removed, but other comments retained | |
| content4, err := fs.ReadFile("/out/case4.wgsl") | |
| require.NoError(t, err) | |
| require.Equal(t, "// This is a comment\nfn main() {}", string(content4)) | |
| _, err = fs.Stat("/out/case4_no_comments.wgsl") | |
| require.Error(t, err) | |
| }, | |
| }, | |
| } | |
| for _, tc := range tests { | |
| t.Run(tc.name, func(t *testing.T) { | |
| wrapper := oswrapper.CreateFSTestOSWrapper() | |
| if tc.setupFS != nil { | |
| tc.setupFS(t, wrapper) | |
| } | |
| err := gatherWgslFiles(tc.inputsDir, tc.outDir, wrapper) | |
| if tc.wantErr { | |
| require.Error(t, err) | |
| } else { | |
| require.NoError(t, err) | |
| } | |
| if tc.verify != nil { | |
| tc.verify(t, wrapper) | |
| } | |
| }) | |
| } | |
| } | |
| func TestCheckInputFileType(t *testing.T) { | |
| tests := []struct { | |
| name string | |
| content []byte | |
| mode FuzzMode | |
| wantErr bool | |
| errMatch string | |
| }{ | |
| { | |
| name: "Valid WGSL text file", | |
| content: []byte("@fragment\nfn main() {}\n"), | |
| mode: FuzzModeWgsl, | |
| wantErr: false, | |
| }, | |
| { | |
| name: "Invalid WGSL - binary file with null byte", | |
| content: []byte{0x00, 0x01, 0x02, 0x03}, | |
| mode: FuzzModeWgsl, | |
| wantErr: true, | |
| errMatch: "wrong file type detected: expected a text file for WGSL mode, but '/test' appears to be binary. Did you forget to add the -ir flag?", | |
| }, | |
| { | |
| name: "Valid IR binary file (contains null byte)", | |
| content: []byte{0x0a, 0x08, 0x00, 0x12, 0x04, 0x01, 0x02, 0x03}, | |
| mode: FuzzModeIr, | |
| wantErr: false, | |
| }, | |
| { | |
| name: "Invalid IR - text file", | |
| content: []byte("@fragment\nfn main() {}\n"), | |
| mode: FuzzModeIr, | |
| wantErr: true, | |
| errMatch: "wrong file type detected: expected a binary file for IR mode, but '/test' appears to be text. Did you mean to remove the -ir flag?", | |
| }, | |
| { | |
| name: "Empty file in WGSL mode (Success)", | |
| content: []byte{}, | |
| mode: FuzzModeWgsl, | |
| wantErr: false, | |
| }, | |
| { | |
| name: "Empty file in IR mode (Success)", | |
| content: []byte{}, | |
| mode: FuzzModeIr, | |
| wantErr: false, | |
| }, | |
| } | |
| for _, tc := range tests { | |
| t.Run(tc.name, func(t *testing.T) { | |
| fs := oswrapper.CreateFSTestOSWrapper() | |
| err := fs.WriteFile("/test", tc.content, 0666) | |
| require.NoError(t, err) | |
| err = checkInputFileType("/test", tc.mode, fs) | |
| if tc.wantErr { | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), tc.errMatch) | |
| } else { | |
| require.NoError(t, err) | |
| } | |
| }) | |
| } | |
| } | |
| func TestTaskModeAndFuzzModeString(t *testing.T) { | |
| taskTests := []struct { | |
| mode TaskMode | |
| expected string | |
| }{ | |
| {TaskModeRun, "run"}, | |
| {TaskModeCheck, "check"}, | |
| {TaskModeGenerate, "generate"}, | |
| {TaskModeTriage, "triage"}, | |
| {TaskModeBisect, "bisect"}, | |
| {TaskModeBisectStep, "bisect-step"}, | |
| {TaskModeExperiment, "experiment"}, | |
| {TaskMode(-1), "<unknown> (-1)"}, | |
| {TaskMode(99), "<unknown> (99)"}, | |
| } | |
| for _, tc := range taskTests { | |
| require.Equal(t, tc.expected, tc.mode.String()) | |
| } | |
| fuzzTests := []struct { | |
| mode FuzzMode | |
| expected string | |
| }{ | |
| {FuzzModeWgsl, "wgsl"}, | |
| {FuzzModeIr, "ir"}, | |
| {FuzzMode(-1), "<unknown> (-1)"}, | |
| {FuzzMode(99), "<unknown> (99)"}, | |
| } | |
| for _, tc := range fuzzTests { | |
| require.Equal(t, tc.expected, tc.mode.String()) | |
| } | |
| } | |
| func TestParseFlagsHelp(t *testing.T) { | |
| // Empty args outputs help to stderr and returns ErrHelp | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{}, &cfg, &stdout, &stderr) | |
| require.ErrorIs(t, err, flag.ErrHelp) | |
| require.Contains(t, stderr.String(), "error: subcommand required") | |
| require.Contains(t, stderr.String(), "Usage:\n fuzz <subcommand> [flags...] [args...]") | |
| } | |
| // Top-level help | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"help"}, &cfg, &stdout, &stderr) | |
| require.ErrorIs(t, err, flag.ErrHelp) | |
| outStr := stdout.String() | |
| require.Contains(t, outStr, "Usage:\n fuzz <subcommand> [flags...] [args...]") | |
| require.Contains(t, outStr, "run") | |
| require.Contains(t, outStr, "check") | |
| require.Contains(t, outStr, "generate") | |
| require.Contains(t, outStr, "triage") | |
| require.Contains(t, outStr, "bisect") | |
| require.Contains(t, outStr, "bisect-step") | |
| require.Contains(t, outStr, "experiment") | |
| require.Contains(t, outStr, "analyze") | |
| require.Contains(t, outStr, "Common Flags") | |
| require.Contains(t, outStr, "-build") | |
| require.Contains(t, outStr, "-verbose") | |
| } | |
| // Subcommand help via 'fuzz help <subcommand>' | |
| subcmds := []string{"run", "check", "generate", "triage", "bisect", "bisect-step", "experiment", "analyze"} | |
| for _, sc := range subcmds { | |
| t.Run("help_"+sc, func(t *testing.T) { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"help", sc}, &cfg, &stdout, &stderr) | |
| require.ErrorIs(t, err, flag.ErrHelp) | |
| require.Contains(t, stdout.String(), fmt.Sprintf("Usage: fuzz %s", sc)) | |
| require.Contains(t, stdout.String(), "-build") | |
| require.Contains(t, stdout.String(), "-verbose") | |
| }) | |
| } | |
| // Subcommand help via 'fuzz <subcommand> -h' | |
| for _, sc := range subcmds { | |
| t.Run("dash_h_"+sc, func(t *testing.T) { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{sc, "-h"}, &cfg, &stdout, &stderr) | |
| require.ErrorIs(t, err, flag.ErrHelp) | |
| require.Contains(t, stderr.String(), fmt.Sprintf("Usage: fuzz %s", sc)) | |
| }) | |
| } | |
| // Unknown subcommand help | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"help", "nonexistent"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.False(t, errors.Is(err, flag.ErrHelp)) | |
| require.Contains(t, err.Error(), `unknown subcommand "nonexistent" to get help for`) | |
| } | |
| } | |
| func TestParseFlagsSubcommands(t *testing.T) { | |
| // Unknown subcommand | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"foobar"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.False(t, errors.Is(err, flag.ErrHelp)) | |
| require.Contains(t, err.Error(), `unknown subcommand "foobar"`) | |
| } | |
| // Run | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"run", "-dump", "-verbose"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeRun, cfg.cmdMode) | |
| require.Equal(t, FuzzModeWgsl, cfg.fuzzMode) | |
| require.True(t, cfg.dump) | |
| require.True(t, cfg.verbose) | |
| require.Equal(t, runtime.NumCPU(), cfg.numProcesses) | |
| // With IR and filter | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"run", "-ir", "-filter", "writer"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, FuzzModeIr, cfg.fuzzMode) | |
| require.Equal(t, "writer", cfg.filter) | |
| // Mesa with filter (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"run", "-mesa", "-filter", "writer"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "cannot set -mesa and -filter flags at the same time") | |
| // Unexpected positional args (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"run", "unexpected"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "unexpected arguments for run") | |
| } | |
| // Check | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"check", "-ir", "-j", "3"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeCheck, cfg.cmdMode) | |
| require.Equal(t, FuzzModeIr, cfg.fuzzMode) | |
| require.Equal(t, 3, cfg.numProcesses) | |
| // Unexpected positional args (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"check", "unexpected"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "unexpected arguments for check") | |
| } | |
| // Generate | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"generate", "-out", "/my/corpus", "-ir", "-mesa"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeGenerate, cfg.cmdMode) | |
| require.Equal(t, FuzzModeIr, cfg.fuzzMode) | |
| require.True(t, cfg.mesaMode) | |
| require.Equal(t, "/my/corpus", cfg.out) | |
| // Missing -out (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"generate"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "need to specify -out when using generate") | |
| // Unexpected positional args (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"generate", "-out", "/my/corpus", "unexpected"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "unexpected arguments for generate") | |
| } | |
| // Triage | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"triage", "-timeout", "30", "-ir", "crash.tirb"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeTriage, cfg.cmdMode) | |
| require.Equal(t, FuzzModeIr, cfg.fuzzMode) | |
| require.Equal(t, "crash.tirb", cfg.triageFile) | |
| require.Equal(t, 30, cfg.timeout) | |
| // Missing file (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"triage"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "triage requires a test case file path") | |
| // Multiple files (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"triage", "file1", "file2"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "triage accepts only one test case file path") | |
| } | |
| // Bisect | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"bisect", "-known-failing", "abc1234", "crash.wgsl"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeBisect, cfg.cmdMode) | |
| require.Equal(t, "crash.wgsl", cfg.bisectFile) | |
| require.Equal(t, "abc1234", cfg.knownFailing) | |
| // Missing -known-failing (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"bisect", "crash.wgsl"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "-known-failing flag is required when bisecting") | |
| // Missing file (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"bisect", "-known-failing", "abc"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "bisect requires a test case file path") | |
| // Multiple files (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"bisect", "-known-failing", "abc", "f1", "f2"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "bisect accepts only one test case file path") | |
| } | |
| // Bisect-step | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"bisect-step", "-is-fix", "crash.wgsl"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeBisectStep, cfg.cmdMode) | |
| require.True(t, cfg.bisectStep) | |
| require.True(t, cfg.isFix) | |
| require.Equal(t, "crash.wgsl", cfg.bisectFile) | |
| // With -bisect flag instead of positional | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"bisect-step", "-bisect", "crash.wgsl"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, "crash.wgsl", cfg.bisectFile) | |
| // Missing file (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"bisect-step"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "bisect-step requires a test case file path") | |
| } | |
| // Experiment | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"experiment", "-machine", "box1", "exp_dir"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeExperiment, cfg.cmdMode) | |
| require.Equal(t, "exp_dir", cfg.experimentPath) | |
| require.Equal(t, "box1", cfg.machineName) | |
| require.Equal(t, 1, cfg.numProcesses) | |
| // Missing dir (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"experiment"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "experiment requires an experiment root directory") | |
| // Multiple dirs (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"experiment", "d1", "d2"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "experiment accepts only one experiment root directory") | |
| } | |
| // Analyze | |
| { | |
| cfg := newDefaultMainConfig() | |
| var stdout, stderr bytes.Buffer | |
| err := parseFlags([]string{"analyze", "-j", "2", "exp_dir"}, &cfg, &stdout, &stderr) | |
| require.NoError(t, err) | |
| require.Equal(t, TaskModeAnalyze, cfg.cmdMode) | |
| require.Equal(t, "exp_dir", cfg.analyzePath) | |
| require.Equal(t, 2, cfg.numProcesses) | |
| // Missing dir (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"analyze"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "analyze requires an experiment root directory") | |
| // Multiple dirs (error) | |
| cfg = newDefaultMainConfig() | |
| err = parseFlags([]string{"analyze", "d1", "d2"}, &cfg, &stdout, &stderr) | |
| require.Error(t, err) | |
| require.Contains(t, err.Error(), "analyze accepts only one experiment root directory") | |
| } | |
| } |