Had to change the tests aswell after changing the list function

Author: patrick.rissmann@l3montree.com

internal/core/attestation/attestation_controller.go

@@ -24,7 +24,7 @@ func (a *attestationController) List(ctx core.Context) error {
 	asset := core.GetAsset(ctx)
 	assetVersion := core.GetAssetVersion(ctx)
 
-	attestationList, err := a.attestationRepository.GetByAssetVersion(asset.GetID(), assetVersion.Name)
+	attestationList, err := a.attestationRepository.GetByAssetVersionAndAssetID(asset.GetID(), assetVersion.Name)
 	if err != nil {
 		return err
 	}

internal/core/attestation/attestation_test.go

@@ -21,10 +21,13 @@ func TestList(t *testing.T) {
 		ctx := e.NewContext(req, httptest.NewRecorder())
 
 		asset := models.Asset{}
+		assetVersion := models.AssetVersion{}
+
 		core.SetAsset(ctx, asset)
+		core.SetAssetVersion(ctx, assetVersion)
 
 		attestationRepository := mocks.NewCoreAttestationRepository(t)
-		attestationRepository.On("GetByAssetID", mock.Anything).Return([]models.Attestation{
+		attestationRepository.On("GetByAssetVersionAndAssetID", mock.Anything, mock.Anything).Return([]models.Attestation{
 			{AttestationName: "not ocol name"},
 		}, nil)
 		attestationController := attestation.NewAttestationController(attestationRepository)
@@ -40,10 +43,13 @@ func TestList(t *testing.T) {
 		ctx := e.NewContext(req, httptest.NewRecorder())
 
 		asset := models.Asset{}
+		assetVersion := models.AssetVersion{}
+
 		core.SetAsset(ctx, asset)
+		core.SetAssetVersion(ctx, assetVersion)
 
 		attestationRepository := mocks.NewCoreAttestationRepository(t)
-		attestationRepository.On("GetByAssetID", mock.Anything).Return([]models.Attestation{}, fmt.Errorf("Something went wrong"))
+		attestationRepository.On("GetByAssetVersionAndAssetID", mock.Anything, mock.Anything).Return([]models.Attestation{}, fmt.Errorf("Something went wrong"))
 		attestationController := attestation.NewAttestationController(attestationRepository)
 
 		result := attestationController.List(ctx)

internal/core/common_interfaces.go

@@ -55,7 +55,7 @@ type AssetRepository interface {
 type AttestationRepository interface {
 	common.Repository[string, models.Attestation, DB]
 	GetByAssetID(assetID uuid.UUID) ([]models.Attestation, error)
-	GetByAssetVersion(assetID uuid.UUID, assetVersion string) ([]models.Attestation, error)
+	GetByAssetVersionAndAssetID(assetID uuid.UUID, assetVersion string) ([]models.Attestation, error)
 }
 
 type CveRepository interface {

internal/database/repositories/attestation_repository.go

@@ -32,7 +32,7 @@ func (a *attestationRepository) GetByAssetID(assetID uuid.UUID) ([]models.Attest
 	return attestationList, nil
 }
 
-func (a *attestationRepository) GetByAssetVersion(assetID uuid.UUID, assetVersion string) ([]models.Attestation, error) {
+func (a *attestationRepository) GetByAssetVersionAndAssetID(assetID uuid.UUID, assetVersion string) ([]models.Attestation, error) {
 	var attestationList []models.Attestation
 	err := a.db.Where("asset_id = ? AND asset_version_name = ?", assetID, assetVersion).Find(&attestationList).Error
 	if err != nil {