Refactor Repository: split into PackagePool and PublishedStorage.

2026-06-10 06:14:22 +00:00 · 2014-02-19 12:03:01 +04:00
parent 7864ce241b
commit d1e16a0ef0
20 changed files with 616 additions and 494 deletions
@@ -0,0 +1,86 @@
+package files
+
+import (
+	"io/ioutil"
+	. "launchpad.net/gocheck"
+	"os"
+	"path/filepath"
+)
+
+type PackagePoolSuite struct {
+	pool *PackagePool
+}
+
+var _ = Suite(&PackagePoolSuite{})
+
+func (s *PackagePoolSuite) SetUpTest(c *C) {
+	s.pool = NewPackagePool(c.MkDir())
+
+}
+
+func (s *PackagePoolSuite) TestRelativePath(c *C) {
+	path, err := s.pool.RelativePath("a/b/package.deb", "91b1a1480b90b9e269ca44d897b12575")
+	c.Assert(err, IsNil)
+	c.Assert(path, Equals, "91/b1/package.deb")
+
+	_, err = s.pool.RelativePath("/", "91b1a1480b90b9e269ca44d897b12575")
+	c.Assert(err, ErrorMatches, ".*is invalid")
+	_, err = s.pool.RelativePath("", "91b1a1480b90b9e269ca44d897b12575")
+	c.Assert(err, ErrorMatches, ".*is invalid")
+	_, err = s.pool.RelativePath("a/b/package.deb", "9")
+	c.Assert(err, ErrorMatches, ".*MD5 is missing")
+}
+
+func (s *PackagePoolSuite) TestPath(c *C) {
+	path, err := s.pool.Path("a/b/package.deb", "91b1a1480b90b9e269ca44d897b12575")
+	c.Assert(err, IsNil)
+	c.Assert(path, Equals, filepath.Join(s.pool.rootPath, "91/b1/package.deb"))
+
+	_, err = s.pool.Path("/", "91b1a1480b90b9e269ca44d897b12575")
+	c.Assert(err, ErrorMatches, ".*is invalid")
+}
+
+func (s *PackagePoolSuite) TestFilepathList(c *C) {
+	list, err := s.pool.FilepathList(nil)
+	c.Check(err, IsNil)
+	c.Check(list, IsNil)
+
+	os.MkdirAll(filepath.Join(s.pool.rootPath, "bd", "0b"), 0755)
+	os.MkdirAll(filepath.Join(s.pool.rootPath, "bd", "0a"), 0755)
+	os.MkdirAll(filepath.Join(s.pool.rootPath, "ae", "0c"), 0755)
+
+	list, err = s.pool.FilepathList(nil)
+	c.Check(err, IsNil)
+	c.Check(list, DeepEquals, []string{})
+
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "ae", "0c", "1.deb"), nil, 0644)
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "ae", "0c", "2.deb"), nil, 0644)
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "bd", "0a", "3.deb"), nil, 0644)
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "bd", "0b", "4.deb"), nil, 0644)
+
+	list, err = s.pool.FilepathList(nil)
+	c.Check(err, IsNil)
+	c.Check(list, DeepEquals, []string{"ae/0c/1.deb", "ae/0c/2.deb", "bd/0a/3.deb", "bd/0b/4.deb"})
+}
+
+func (s *PackagePoolSuite) TestRemove(c *C) {
+	os.MkdirAll(filepath.Join(s.pool.rootPath, "bd", "0b"), 0755)
+	os.MkdirAll(filepath.Join(s.pool.rootPath, "bd", "0a"), 0755)
+	os.MkdirAll(filepath.Join(s.pool.rootPath, "ae", "0c"), 0755)
+
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "ae", "0c", "1.deb"), []byte("1"), 0644)
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "ae", "0c", "2.deb"), []byte("22"), 0644)
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "bd", "0a", "3.deb"), []byte("333"), 0644)
+	ioutil.WriteFile(filepath.Join(s.pool.rootPath, "bd", "0b", "4.deb"), []byte("4444"), 0644)
+
+	size, err := s.pool.Remove("ae/0c/2.deb")
+	c.Check(err, IsNil)
+	c.Check(size, Equals, int64(2))
+
+	_, err = s.pool.Remove("ae/0c/2.deb")
+	c.Check(err, ErrorMatches, ".*no such file or directory")
+
+	list, err := s.pool.FilepathList(nil)
+	c.Check(err, IsNil)
+	c.Check(list, DeepEquals, []string{"ae/0c/1.deb", "bd/0a/3.deb", "bd/0b/4.deb"})
+}