1212// See the License for the specific language governing permissions and
1313// limitations under the License.
1414
15- package tango
15+ package targetoverlap
1616
1717import (
1818 "context"
@@ -24,6 +24,7 @@ import (
2424
2525 "github.com/uber/submitqueue/submitqueue/entity"
2626 "github.com/uber/submitqueue/submitqueue/extension/conflict"
27+ "github.com/uber/submitqueue/submitqueue/extension/dependency/resolver"
2728)
2829
2930// fakeResolver is an in-test TargetResolver that returns pre-configured target
@@ -47,11 +48,16 @@ func (f *fakeResolver) failWith(err error) *fakeResolver {
4748 return f
4849}
4950
50- func (f * fakeResolver ) ChangedTargets (_ context.Context , batch entity.Batch ) ([]string , error ) {
51+ func (f * fakeResolver ) ChangedTargets (_ context.Context , batch entity.Batch ) ([]resolver. Target , error ) {
5152 if f .err != nil {
5253 return nil , f .err
5354 }
54- return f .targets [batch .ID ], nil
55+ names := f .targets [batch .ID ]
56+ targets := make ([]resolver.Target , len (names ))
57+ for i , n := range names {
58+ targets [i ] = resolver.Target {Name : n }
59+ }
60+ return targets , nil
5561}
5662
5763func cfg () conflict.Config {
@@ -123,14 +129,14 @@ func TestAnalyze(t *testing.T) {
123129
124130 for _ , tt := range tests {
125131 t .Run (tt .name , func (t * testing.T ) {
126- resolver := newFakeResolver ().set (tt .candidate , tt .candTargets ... )
132+ r := newFakeResolver ().set (tt .candidate , tt .candTargets ... )
127133 inFlight := make ([]entity.Batch , 0 , len (tt .inFlight ))
128134 for _ , f := range tt .inFlight {
129- resolver .set (f .id , f .targets ... )
135+ r .set (f .id , f .targets ... )
130136 inFlight = append (inFlight , entity.Batch {ID : f .id })
131137 }
132138
133- got , err := New (cfg (), resolver ).Analyze (context .Background (), entity.Batch {ID : tt .candidate }, inFlight )
139+ got , err := New (cfg (), r ).Analyze (context .Background (), entity.Batch {ID : tt .candidate }, inFlight )
134140 require .NoError (t , err )
135141
136142 var ids []string
@@ -150,17 +156,17 @@ func TestAnalyze_EmptyInFlight(t *testing.T) {
150156}
151157
152158func TestAnalyze_ResolverError (t * testing.T ) {
153- sentinel := errors .New ("tango unavailable" )
159+ sentinel := errors .New ("resolver unavailable" )
154160
155161 t .Run ("candidate resolution fails" , func (t * testing.T ) {
156- resolver := newFakeResolver ().failWith (sentinel )
157- _ , err := New (cfg (), resolver ).Analyze (context .Background (), entity.Batch {ID : "cand" }, []entity.Batch {{ID : "x" }})
162+ r := newFakeResolver ().failWith (sentinel )
163+ _ , err := New (cfg (), r ).Analyze (context .Background (), entity.Batch {ID : "cand" }, []entity.Batch {{ID : "x" }})
158164 require .ErrorIs (t , err , sentinel )
159165 })
160166
161167 t .Run ("in-flight resolution fails" , func (t * testing.T ) {
162- resolver := newFakeResolver ().set ("cand" , "//foo:lib" ).failWith (sentinel )
163- _ , err := New (cfg (), resolver ).Analyze (context .Background (), entity.Batch {ID : "cand" }, []entity.Batch {{ID : "x" }})
168+ r := newFakeResolver ().set ("cand" , "//foo:lib" ).failWith (sentinel )
169+ _ , err := New (cfg (), r ).Analyze (context .Background (), entity.Batch {ID : "cand" }, []entity.Batch {{ID : "x" }})
164170 require .ErrorIs (t , err , sentinel )
165171 })
166172}
0 commit comments