1- package oci
1+ package oci_test
22
33import (
44 "context"
@@ -13,6 +13,7 @@ import (
1313 "github.com/stretchr/testify/require"
1414
1515 "github.com/kyma-project/lifecycle-manager/internal/manifest/spec"
16+ "github.com/kyma-project/lifecycle-manager/internal/repository/oci"
1617)
1718
1819func TestNewRepository (t * testing.T ) {
@@ -62,7 +63,7 @@ func TestNewRepository(t *testing.T) {
6263
6364 for _ , tt := range tests {
6465 t .Run (tt .name , func (t * testing.T ) {
65- _ , err := NewRepository (tt .kcl , tt .hostPort , tt .insecure )
66+ _ , err := oci . NewRepository (tt .kcl , tt .hostPort , tt .insecure )
6667 if (err != nil ) != tt .expectErr {
6768 t .Errorf ("NewRepository() error = %v, expectErr %v" , err , tt .expectErr )
6869 }
@@ -76,9 +77,9 @@ func TestGetConfigFile(t *testing.T) {
7677 mcc := mockCraneClient {
7778 configResult : []byte ("mock config data" ),
7879 }
79- repo , err := NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
80+ repo , err := oci . NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
8081 require .NoError (t , err )
81- repo . craneClient = & mcc
82+ oci . SetCraneWrapper ( repo , & mcc )
8283 // when
8384 configData , err := repo .GetConfigFile (t .Context (), "test-name" , "test-version" )
8485 // then
@@ -92,7 +93,7 @@ func TestGetConfigFile(t *testing.T) {
9293
9394 t .Run ("should return an error from KeyChain Lookup" , func (t * testing.T ) {
9495 // given
95- repo , err := NewRepository (& errorKeyChainLookup {}, "dummy" , false )
96+ repo , err := oci . NewRepository (& errorKeyChainLookup {}, "dummy" , false )
9697 require .NoError (t , err )
9798 // when
9899 _ , err = repo .GetConfigFile (t .Context (), "test-name" , "test-version" )
@@ -108,9 +109,9 @@ func TestGetConfigFile(t *testing.T) {
108109 configResult : []byte ("mock config data" ),
109110 }
110111 mkcl := & mockKeyChainLookup {}
111- repo , err := NewRepository (mkcl , "europe-docker.pkg.dev/kyma-project/prod" , false )
112+ repo , err := oci . NewRepository (mkcl , "europe-docker.pkg.dev/kyma-project/prod" , false )
112113 require .NoError (t , err )
113- repo . craneClient = & mcc
114+ oci . SetCraneWrapper ( repo , & mcc )
114115 assert .Nil (t , mkcl .ctx )
115116 // when
116117 configData , err := repo .GetConfigFile (t .Context (), "test-name" , "test-version" )
@@ -122,9 +123,9 @@ func TestGetConfigFile(t *testing.T) {
122123
123124 t .Run ("should return an error from craneClient.Config" , func (t * testing.T ) {
124125 // given
125- repo , err := NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
126+ repo , err := oci . NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
126127 require .NoError (t , err )
127- repo . craneClient = & mockCraneClient {}
128+ oci . SetCraneWrapper ( repo , & mockCraneClient {})
128129 // when
129130 _ , err = repo .GetConfigFile (t .Context (), "test-name" , "test-version" )
130131 // then
@@ -142,9 +143,9 @@ func TestPullLayer(t *testing.T) {
142143 pullResult : mockLayer ,
143144 }
144145
145- repo , err := NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
146+ repo , err := oci . NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
146147 require .NoError (t , err )
147- repo . craneClient = & mcc
148+ oci . SetCraneWrapper ( repo , & mcc )
148149
149150 layer , err := repo .PullLayer (t .Context (), "test-name" , "test-version" , "sha256:abcdef1234567890" )
150151 require .NoError (t , err )
@@ -156,17 +157,17 @@ func TestPullLayer(t *testing.T) {
156157 })
157158
158159 t .Run ("should return an error from KeyChain Lookup" , func (t * testing.T ) {
159- repo , err := NewRepository (& errorKeyChainLookup {}, "europe-docker.pkg.dev/kyma-project/prod" , false )
160+ repo , err := oci . NewRepository (& errorKeyChainLookup {}, "europe-docker.pkg.dev/kyma-project/prod" , false )
160161 require .NoError (t , err )
161162 _ , err = repo .PullLayer (t .Context (), "test-name" , "test-version" , "sha256:abcdef1234567890" )
162163 require .Error (t , err )
163164 assert .ErrorIs (t , err , errKeyChain )
164165 })
165166
166167 t .Run ("should return an error from craneClient.PullLayer" , func (t * testing.T ) {
167- repo , err := NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
168+ repo , err := oci . NewRepository (nil , "europe-docker.pkg.dev/kyma-project/prod" , true )
168169 require .NoError (t , err )
169- repo . craneClient = & mockCraneClient {}
170+ oci . SetCraneWrapper ( repo , & mockCraneClient {})
170171 _ , err = repo .PullLayer (t .Context (), "test-name" , "test-version" , "sha256:abcdef1234567890" )
171172 require .Error (t , err )
172173 require .ErrorIs (t , err , errMockPull )
0 commit comments