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)
}