Skip to content

Commit

Permalink
Merge pull request #443 from glours/remove-dependencies-disabled-serv…
Browse files Browse the repository at this point in the history
…ices

remove potential dependencies to disabled services in ForServices
  • Loading branch information
glours committed Aug 10, 2023
2 parents acec34a + 2037d3f commit 45494a8
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 7 deletions.
13 changes: 12 additions & 1 deletion types/project.go
Original file line number Diff line number Diff line change
Expand Up @@ -423,13 +423,24 @@ func (p *Project) ForServices(names []string, options ...DependencyOption) error
}
enabled = append(enabled, s)
} else {
p.DisabledServices = append(p.DisabledServices, s)
p.DisableService(s)
}
}
p.Services = enabled
return nil
}

func (p *Project) DisableService(service ServiceConfig) {
// We should remove all dependencies which reference the disabled service
for i, s := range p.Services {
if _, ok := s.DependsOn[service.Name]; ok {
delete(s.DependsOn, service.Name)
p.Services[i] = s
}
}
p.DisabledServices = append(p.DisabledServices, service)
}

// ResolveImages updates services images to include digest computed by a resolver function
func (p *Project) ResolveImages(resolver func(named reference.Named) (godigest.Digest, error)) error {
eg := errgroup.Group{}
Expand Down
30 changes: 24 additions & 6 deletions types/project_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,10 @@ import (
func Test_ApplyProfiles(t *testing.T) {
p := makeProject()
p.ApplyProfiles([]string{"foo"})
assert.Equal(t, len(p.Services), 2)
assert.Equal(t, len(p.Services), 3)
assert.Equal(t, p.Services[0].Name, "service_1")
assert.Equal(t, p.Services[1].Name, "service_2")
assert.Equal(t, p.Services[2].Name, "service_6")
assert.Equal(t, len(p.DisabledServices), 3)
assert.Equal(t, p.DisabledServices[0].Name, "service_3")
assert.Equal(t, p.DisabledServices[1].Name, "service_4")
Expand All @@ -40,11 +41,12 @@ func Test_ApplyProfiles(t *testing.T) {
err := p.EnableServices("service_4")
assert.NilError(t, err)

assert.Equal(t, len(p.Services), 4)
assert.Equal(t, len(p.Services), 5)
assert.Equal(t, p.Services[0].Name, "service_1")
assert.Equal(t, p.Services[1].Name, "service_2")
assert.Equal(t, p.Services[2].Name, "service_4")
assert.Equal(t, p.Services[3].Name, "service_5")
assert.Equal(t, p.Services[2].Name, "service_6")
assert.Equal(t, p.Services[3].Name, "service_4")
assert.Equal(t, p.Services[4].Name, "service_5")
assert.Equal(t, len(p.DisabledServices), 1)
assert.Equal(t, p.DisabledServices[0].Name, "service_3")

Expand Down Expand Up @@ -74,9 +76,10 @@ func Test_WithoutUnnecessaryResources(t *testing.T) {
func Test_NoProfiles(t *testing.T) {
p := makeProject()
p.ApplyProfiles(nil)
assert.Equal(t, len(p.Services), 1)
assert.Equal(t, len(p.Services), 2)
assert.Equal(t, len(p.DisabledServices), 4)
assert.Equal(t, p.Services[0].Name, "service_1")
assert.Equal(t, p.Services[1].Name, "service_6")
}

func Test_ServiceProfiles(t *testing.T) {
Expand All @@ -94,10 +97,22 @@ func Test_ForServices(t *testing.T) {
err := p.ForServices([]string{"service_2"})
assert.NilError(t, err)

assert.Equal(t, len(p.DisabledServices), 3)
assert.Equal(t, len(p.DisabledServices), 4)
assert.Equal(t, p.DisabledServices[0].Name, "service_3")
assert.Equal(t, p.DisabledServices[1].Name, "service_4")
assert.Equal(t, p.DisabledServices[2].Name, "service_5")
assert.Equal(t, p.DisabledServices[3].Name, "service_6")

// Should not load the dependency service_1 when explicitly loading service_6
p = makeProject()
err = p.ForServices([]string{"service_6"})
assert.NilError(t, err)
assert.Equal(t, len(p.DisabledServices), 5)
assert.Equal(t, p.DisabledServices[0].Name, "service_1")
assert.Equal(t, p.DisabledServices[1].Name, "service_2")
assert.Equal(t, p.DisabledServices[2].Name, "service_3")
assert.Equal(t, p.DisabledServices[3].Name, "service_4")
assert.Equal(t, p.DisabledServices[4].Name, "service_5")
}

func Test_ForServicesCycle(t *testing.T) {
Expand Down Expand Up @@ -127,6 +142,9 @@ func makeProject() Project {
}, ServiceConfig{
Name: "service_5",
Profiles: []string{"zot"},
}, ServiceConfig{
Name: "service_6",
Links: []string{"service_1"},
}),
Networks: Networks{},
Volumes: Volumes{},
Expand Down

0 comments on commit 45494a8

Please sign in to comment.