mirror of
https://github.com/mudler/luet.git
synced 2025-06-26 07:22:59 +00:00
f6a4b634c1
With this change the solver during install now considers only the part of the tree which is required to calculate the solution, it doesn't consider anymore World() as the search space. The search space now is narrowed down to the packages that related to the one which we are considering. In this subset of changes we are also optimizing the Parallel solver avoiding an useless loop. This change boost overall performance on large datasets which don't necessarly have relations touching the whole tree.
189 lines
7.4 KiB
Go
189 lines
7.4 KiB
Go
// Copyright © 2019 Ettore Di Giacinto <mudler@gentoo.org>
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 2 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
|
|
// Recipe is a builder imeplementation.
|
|
|
|
// It reads a Tree and spit it in human readable form (YAML), called recipe,
|
|
// It also loads a tree (recipe) from a YAML (to a db, e.g. BoltDB), allowing to query it
|
|
// with the solver, using the package object.
|
|
package tree_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"regexp"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
|
|
pkg "github.com/mudler/luet/pkg/package"
|
|
"github.com/mudler/luet/pkg/solver"
|
|
. "github.com/mudler/luet/pkg/tree"
|
|
)
|
|
|
|
var _ = Describe("Tree", func() {
|
|
|
|
Context("Simple solving with the fixture tree", func() {
|
|
It("writes and reads back the same tree", func() {
|
|
for index := 0; index < 300; index++ { // Just to make sure we don't have false positives
|
|
db := pkg.NewInMemoryDatabase(false)
|
|
generalRecipe := NewCompilerRecipe(db)
|
|
tmpdir, err := ioutil.TempDir("", "package")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
defer os.RemoveAll(tmpdir) // clean up
|
|
|
|
err = generalRecipe.Load("../../tests/fixtures/buildableseed")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(generalRecipe.GetDatabase().World())).To(Equal(4))
|
|
|
|
D, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(D.GetRequires()[0].GetName()).To(Equal("c"))
|
|
CfromD, err := generalRecipe.GetDatabase().FindPackage(D.GetRequires()[0])
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(CfromD.GetRequires()) != 0).To(BeTrue())
|
|
Expect(CfromD.GetRequires()[0].GetName()).To(Equal("b"))
|
|
|
|
s := solver.NewSolver(solver.Options{Type: solver.SingleCoreSimple}, pkg.NewInMemoryDatabase(false), generalRecipe.GetDatabase(), db)
|
|
pack, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
solution, err := s.Install([]pkg.Package{pack})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
solution, err = solution.Order(generalRecipe.GetDatabase(), pack.GetFingerPrint())
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(solution[0].Package.GetName()).To(Equal("b"))
|
|
Expect(solution[0].Value).To(BeTrue())
|
|
|
|
Expect(solution[1].Package.GetName()).To(Equal("c"))
|
|
Expect(solution[1].Value).To(BeTrue())
|
|
|
|
Expect(solution[2].Package.GetName()).To(Equal("d"))
|
|
Expect(solution[2].Value).To(BeTrue())
|
|
Expect(len(solution)).To(Equal(3))
|
|
|
|
newsolution := solution.Drop(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
|
|
Expect(len(newsolution)).To(Equal(2))
|
|
|
|
Expect(newsolution[0].Package.GetName()).To(Equal("b"))
|
|
Expect(newsolution[0].Value).To(BeTrue())
|
|
|
|
Expect(newsolution[1].Package.GetName()).To(Equal("c"))
|
|
Expect(newsolution[1].Value).To(BeTrue())
|
|
|
|
}
|
|
})
|
|
})
|
|
|
|
Context("Multiple trees", func() {
|
|
It("Merges", func() {
|
|
for index := 0; index < 300; index++ { // Just to make sure we don't have false positives
|
|
db := pkg.NewInMemoryDatabase(false)
|
|
generalRecipe := NewCompilerRecipe(db)
|
|
tmpdir, err := ioutil.TempDir("", "package")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
defer os.RemoveAll(tmpdir) // clean up
|
|
|
|
err = generalRecipe.Load("../../tests/fixtures/buildableseed")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(generalRecipe.GetDatabase().World())).To(Equal(4))
|
|
|
|
err = generalRecipe.Load("../../tests/fixtures/layers")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(generalRecipe.GetDatabase().World())).To(Equal(6))
|
|
|
|
extra, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "extra", Category: "layer", Version: "1.0"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(extra).ToNot(BeNil())
|
|
|
|
D, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(D.GetRequires()[0].GetName()).To(Equal("c"))
|
|
CfromD, err := generalRecipe.GetDatabase().FindPackage(D.GetRequires()[0])
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(CfromD.GetRequires()) != 0).To(BeTrue())
|
|
Expect(CfromD.GetRequires()[0].GetName()).To(Equal("b"))
|
|
|
|
s := solver.NewSolver(solver.Options{Type: solver.SingleCoreSimple}, pkg.NewInMemoryDatabase(false), generalRecipe.GetDatabase(), db)
|
|
|
|
Dd, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
solution, err := s.Install([]pkg.Package{Dd})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
solution, err = solution.Order(generalRecipe.GetDatabase(), Dd.GetFingerPrint())
|
|
Expect(err).ToNot(HaveOccurred())
|
|
pack, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "a", Category: "test", Version: "1.0"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
base, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "base", Category: "layer", Version: "0.2"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(solution).ToNot(ContainElement(solver.PackageAssert{Package: pack.(*pkg.DefaultPackage), Value: true}))
|
|
Expect(solution).To(ContainElement(solver.PackageAssert{Package: D.(*pkg.DefaultPackage), Value: true}))
|
|
Expect(solution).ToNot(ContainElement(solver.PackageAssert{Package: extra.(*pkg.DefaultPackage), Value: true}))
|
|
Expect(solution).ToNot(ContainElement(solver.PackageAssert{Package: base.(*pkg.DefaultPackage), Value: true}))
|
|
Expect(len(solution)).To(Equal(3))
|
|
}
|
|
})
|
|
})
|
|
|
|
Context("Simple tree with labels", func() {
|
|
It("Read tree with labels", func() {
|
|
db := pkg.NewInMemoryDatabase(false)
|
|
generalRecipe := NewCompilerRecipe(db)
|
|
|
|
err := generalRecipe.Load("../../tests/fixtures/labels")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(generalRecipe.GetDatabase().World())).To(Equal(1))
|
|
pack, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "pkgA", Category: "test", Version: "0.1"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(pack.HasLabel("label1")).To(Equal(true))
|
|
Expect(pack.HasLabel("label3")).To(Equal(false))
|
|
})
|
|
})
|
|
|
|
Context("Simple tree with annotations", func() {
|
|
It("Read tree with annotations", func() {
|
|
db := pkg.NewInMemoryDatabase(false)
|
|
generalRecipe := NewCompilerRecipe(db)
|
|
r := regexp.MustCompile("^label")
|
|
|
|
err := generalRecipe.Load("../../tests/fixtures/annotations")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(len(generalRecipe.GetDatabase().World())).To(Equal(1))
|
|
pack, err := generalRecipe.GetDatabase().FindPackage(&pkg.DefaultPackage{Name: "pkgA", Category: "test", Version: "0.1"})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(pack.HasAnnotation("label1")).To(Equal(true))
|
|
Expect(pack.HasAnnotation("label3")).To(Equal(false))
|
|
Expect(pack.MatchAnnotation(r)).To(Equal(true))
|
|
})
|
|
})
|
|
|
|
})
|