Repository mirroring: working first version.

debian/list_test.go

@@ -0,0 +1,57 @@
+package debian
+
+import (
+	debc "github.com/smira/godebiancontrol"
+	. "launchpad.net/gocheck"
+)
+
+type PackageListSuite struct {
+	list           *PackageList
+	p1, p2, p3, p4 *Package
+}
+
+var _ = Suite(&PackageListSuite{})
+
+func (s *PackageListSuite) SetUpTest(c *C) {
+	s.list = NewPackageList()
+
+	paraGen := func() debc.Paragraph {
+		para := make(debc.Paragraph)
+		for k, v := range packagePara {
+			para[k] = v
+		}
+		return para
+	}
+
+	s.p1 = NewPackageFromControlFile(paraGen())
+	s.p2 = NewPackageFromControlFile(paraGen())
+	para := paraGen()
+	para["Package"] = "mars-invaders"
+	s.p3 = NewPackageFromControlFile(para)
+	para = paraGen()
+	para["Size"] = "42"
+	s.p4 = NewPackageFromControlFile(para)
+}
+
+func (s *PackageListSuite) TestAddLength(c *C) {
+	c.Check(s.list.Length(), Equals, 0)
+	c.Check(s.list.Add(s.p1), IsNil)
+	c.Check(s.list.Length(), Equals, 1)
+	c.Check(s.list.Add(s.p2), IsNil)
+	c.Check(s.list.Length(), Equals, 1)
+	c.Check(s.list.Add(s.p3), IsNil)
+	c.Check(s.list.Length(), Equals, 2)
+	c.Check(s.list.Add(s.p4), ErrorMatches, "conflict in package.*")
+}
+
+func (s *PackageListSuite) TestForeach(c *C) {
+	s.list.Add(s.p1)
+	s.list.Add(s.p3)
+
+	length := 0
+	s.list.ForEach(func(*Package) {
+		length++
+	})
+
+	c.Check(length, Equals, 2)
+}