Thorough test for compression.

utils/checksum.go

@@ -10,6 +10,7 @@ import (
 	"os"
 )
 
+// ChecksumInfo represents checksums for a single file
 type ChecksumInfo struct {
 	Size   int64
 	MD5    string

utils/checksum_test.go

@@ -14,7 +14,7 @@ var _ = Suite(&ChecksumSuite{})
 
 func (s *ChecksumSuite) SetUpTest(c *C) {
 	s.tempfile, _ = ioutil.TempFile(c.MkDir(), "aptly-test")
-	s.tempfile.WriteString("Quick brown fox jumps over black dog and runs away... Really far away... who knows?")
+	s.tempfile.WriteString(testString)
 }
 
 func (s *ChecksumSuite) TearDownTest(c *C) {

utils/compress_test.go

@@ -1,6 +1,8 @@
 package utils
 
 import (
+	"compress/bzip2"
+	"compress/gzip"
 	"io/ioutil"
 	. "launchpad.net/gocheck"
 	"os"
@@ -12,9 +14,11 @@ type CompressSuite struct {
 
 var _ = Suite(&CompressSuite{})
 
+const testString = "Quick brown fox jumps over black dog and runs away... Really far away... who knows?"
+
 func (s *CompressSuite) SetUpTest(c *C) {
 	s.tempfile, _ = ioutil.TempFile(c.MkDir(), "aptly-test")
-	s.tempfile.WriteString("Quick brown fox jumps over black dog and runs away... Really far away... who knows?")
+	s.tempfile.WriteString(testString)
 }
 
 func (s *CompressSuite) TearDownTest(c *C) {
@@ -25,11 +29,31 @@ func (s *CompressSuite) TestCompress(c *C) {
 	err := CompressFile(s.tempfile)
 	c.Assert(err, IsNil)
 
-	st, err := os.Stat(s.tempfile.Name() + ".gz")
-	c.Assert(err, IsNil)
-	c.Assert(st.Size() < 100, Equals, true)
+	buf := make([]byte, len(testString))
 
-	st, err = os.Stat(s.tempfile.Name() + ".bz2")
+	file, err := os.Open(s.tempfile.Name() + ".gz")
 	c.Assert(err, IsNil)
-	c.Assert(st.Size() < 120, Equals, true)
+
+	gzReader, err := gzip.NewReader(file)
+	c.Assert(err, IsNil)
+
+	_, err = gzReader.Read(buf)
+	c.Assert(err, IsNil)
+
+	gzReader.Close()
+	file.Close()
+
+	c.Check(string(buf), Equals, testString)
+
+	file, err = os.Open(s.tempfile.Name() + ".bz2")
+	c.Assert(err, IsNil)
+
+	bzReader := bzip2.NewReader(file)
+
+	_, err = bzReader.Read(buf)
+	c.Assert(err, IsNil)
+
+	file.Close()
+
+	c.Check(string(buf), Equals, testString)
 }