github.com/khulnasoft-lab/tunnel-db@v0.0.0-20231117205118-74e1113bd007/pkg/vulnsrc/bucket/bucket_test.go (about)

     1  package bucket_test
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  
     8  	"github.com/khulnasoft-lab/tunnel-db/pkg/types"
     9  	"github.com/khulnasoft-lab/tunnel-db/pkg/vulnsrc/bucket"
    10  	"github.com/khulnasoft-lab/tunnel-db/pkg/vulnsrc/vulnerability"
    11  )
    12  
    13  func TestBucketName(t *testing.T) {
    14  	testCases := []struct {
    15  		name       string
    16  		ecosystem  types.Ecosystem
    17  		dataSource string
    18  		want       string
    19  		wantErr    string
    20  	}{
    21  		{
    22  			name:       "go",
    23  			ecosystem:  vulnerability.Go,
    24  			dataSource: "GitLab Advisory Database",
    25  			want:       "go::GitLab Advisory Database",
    26  		},
    27  		{
    28  			name:       "rubygems",
    29  			ecosystem:  vulnerability.RubyGems,
    30  			dataSource: "GitHub Advisory Database",
    31  			want:       "rubygems::GitHub Advisory Database",
    32  		},
    33  	}
    34  
    35  	for _, tc := range testCases {
    36  		t.Run(tc.name, func(t *testing.T) {
    37  			got := bucket.Name(tc.ecosystem, tc.dataSource)
    38  			assert.Equal(t, tc.want, got)
    39  		})
    40  	}
    41  }