diff --git a/checks/raw/binary_artifact_test.go b/checks/raw/binary_artifact_test.go new file mode 100644 index 00000000000..72f50f1bcb0 --- /dev/null +++ b/checks/raw/binary_artifact_test.go @@ -0,0 +1,97 @@ +// Copyright 2021 Security Scorecard 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. + +package raw + +import ( + "fmt" + "log" + "os" + "testing" + + "github.com/golang/mock/gomock" + + mockrepo "github.com/ossf/scorecard/v4/clients/mockclients" +) + +// TestBinaryArtifact tests the BinaryArtifact checker. +func TestBinaryArtifacts(t *testing.T) { + t.Parallel() + tests := []struct { + name string + inputFile string + err error + files []string + expect int + }{ + { + name: "Jar file", + inputFile: "../testdata/binaryartifacts/jars/aws-java-sdk-core-1.11.571.jar", + err: nil, + files: []string{ + "../testdata/binaryartifacts/jars/aws-java-sdk-core-1.11.571.jar", + }, + expect: 1, + }, + { + name: "non binary file", + inputFile: "../testdata/licensedir/withlicense/LICENSE", + err: nil, + files: []string{ + "../testdata/licensedir/withlicense/LICENSE", + }, + }, + { + name: "non binary file", + inputFile: "../doesnotexist", + err: nil, + files: []string{ + "../doesnotexist", + }, + }, + } + for _, tt := range tests { + tt := tt // Re-initializing variable so it is not changed while executing the closure below + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + + ctrl := gomock.NewController(t) + mockRepoClient := mockrepo.NewMockRepoClient(ctrl) + mockRepoClient.EXPECT().ListFiles(gomock.Any()).Return(tt.files, nil) + mockRepoClient.EXPECT().GetFileContent(gomock.Any()).DoAndReturn(func(file string) ([]byte, error) { + // This will read the file and return the content + content, err := os.ReadFile(file) + log.Println(os.Getwd()) + if err != nil { + return content, fmt.Errorf("%w", err) + } + return content, nil + }) + + f, err := BinaryArtifacts(mockRepoClient) + + if tt.err != nil { + // If we expect an error, make sure it is the same + if err == nil { + t.Errorf("expected error, got nil") + } + } else { + // Check that the expected number of files are returned + if len(f.Files) != tt.expect { + t.Errorf("expected %d files, got %d test %v", tt.expect, len(f.Files), tt.name) + } + } + }) + } +}