control_test.go

package main

import (
	"bytes"
	"strings"
	"testing"

	"code.google.com/p/go.crypto/openpgp"
)

func TestDebControlInvalid1(t *testing.T) {
	_, err := ParseDebianControl(strings.NewReader("Total rubbish"), nil)
	if err == nil {
		t.Errorf("Non-field value should not parse")
	}
}

func TestDebControlInvalid2(t *testing.T) {
	_, err := ParseDebianControl(strings.NewReader("  continuation before a field is declared"), nil)
	if err == nil {
		t.Errorf("Misplaced continuation field should not parse")
	}
}

var testDebControlEmptyStart = []string{
	"",
	"Package: first",
	"fieldb: bit of stuff",
	"fieldc: and other stuff",
	"thiny2: ",
	" val1",
	" val2",
	" val3",
}

var testDebControlEmptyStartResult = []string{
	"Package: first",
	"fieldb: bit of stuff",
	"fieldc: and other stuff",
	"thiny2: ",
	" val1",
	" val2",
	" val3",
}

// Check that we output unknown fields in a consistant order
func TestDebControlEmptyStart(t *testing.T) {
	inStr := strings.Join(testDebControlEmptyStart, "\n") + "\n"
	resStr := strings.Join(testDebControlEmptyStartResult, "\n") + "\n"

	paragraphs, err := ParseDebianControl(strings.NewReader(inStr), nil)
	if err != nil {
		t.Errorf("parsedebiancontrol failure: %s", inStr)
	}

	var buf bytes.Buffer
	debStartFields := []string{"Package", "Version", "Filename", "Directory", "Size"}
	debEndFields := []string{"MD5Sum", "MD5sum", "SHA1", "SHA256", "Description"}
	WriteDebianControl(&buf, paragraphs, debStartFields, debEndFields)
	outStr := string(buf.Bytes())
	if outStr != resStr {
		t.Errorf("\nExpected:\n%s\nGot:\n%s", inStr, outStr)
	}
}

var testDebControlPersist = []string{
	"Package: first",
	"fieldb: bit of stuff",
	"fieldc: and other stuff",
	"thiny2: ",
	" val1",
	" val2",
	" val3",
	"Description: thingy with",
	" multiple lines of",
	" stuff",
	"",
	"Package: second",
	"fielda: with a",
	"fieldb: bit of stuff",
	"fieldc: and other stuff",
	"thiny2: ",
	" val1",
	" val2",
	" val3",
	"Description: thingy with",
	" multiple lines of",
	" stuff",
}

// Check that we output unknown fields in a consistant order
func TestDebControlPersist(t *testing.T) {
	inStr := strings.Join(testDebControlPersist, "\n") + "\n"

	paragraphs, err := ParseDebianControl(strings.NewReader(inStr), nil)
	if err != nil {
		t.Errorf("parsedebiancontrol failure: %s", inStr)
	}

	var buf bytes.Buffer
	debStartFields := []string{"Package", "Version", "Filename", "Directory", "Size"}
	debEndFields := []string{"MD5Sum", "MD5sum", "SHA1", "SHA256", "Description"}
	WriteDebianControl(&buf, paragraphs, debStartFields, debEndFields)
	outStr := string(buf.Bytes())
	if outStr != inStr {
		t.Errorf("\nExpected:\n%s\nGot:\n%s", inStr, outStr)
	}
}

var testControlSignedValid = `-----BEGIN PGP SIGNED MESSAGE-----
Hash: SHA1

Checksums-Sha1: 
 7670839693da39075134c1a1f5faad6623df70af 2416 collectd_5.4.0-3.dsc
 8f06307bf1c17b83351fccdd7a93b4a822ecbdf4 76417 collectd_5.4.0-3.diff.gz
Checksums-Sha256: 
 c0679d60f28ceecd09b3c000361c691e373dba599e3878135bc36bede14e109d 2416 collectd_5.4.0-3.dsc
 e6d7f21737d2146a9bb30a46137fbd0f00be7971e8c3edc6e66a5981498a261e 76417 collectd_5.4.0-3.diff.gz
Files: 
 cd9aa41b337352fd160f326523a9c3d8 2416 utils optional collectd_5.4.0-3.dsc
 d1270867f1c9517dd92016ea9f2d5afe 76417 utils optional collectd_5.4.0-3.diff.gz
-----BEGIN PGP SIGNATURE-----
Version: GnuPG v1

iQEcBAEBAgAGBQJTiG5VAAoJEK9K5quiVC/K9IkH/3nBYgl89/DnjI6ESksYrtkT
Zv/7Cpfg7YsvUmMrghh1hWklK8zj2Tm3NgKshW/HF7orn5cmPUMVZZU8EFa5uR43
FEJ+r1zA40dcOUtKgRESke4b8MNebWq01Op7zrlU1w/4fZ2MFxhNiQ4Xr3ziEl61
kPPR+1ZG43h+wy1h6QXzcNdqcwUnCfX4Uqlhz1giJ1/1qEdW6HlLOiIomZLGhg6b
K5JmdVY4fiH1Fv0tSq7mVnN7LbXJBo8KyzbqJAGkWNu0zh/G/5whz6n2ohgWC/SJ
uiTJMgKpAOxBFeEzO1quFyWnQePIjQ2zWVaTwqDPiZNQ6+377gCrC4Fu+SYdmlQ=
=JixZ
-----END PGP SIGNATURE-----
`

var testKrStr = `-----BEGIN PGP PUBLIC KEY BLOCK-----
Version: GnuPG v1

mQENBFOIYWUBCACzPZm2qE7DrXc+feQ23ll24ZOQD4V7AjFPz1D20iXZqgZIJoQw
40CPu7/WRrJO4lf6Ak/VgeCnijzRuNkDUk2QT8fsNFVgSWxj6FAM1wan0oWqDyR3
yM9RYojQBpzt0uTolE4g4rs5ywWKaWmA6ra0fU6CCxETM+b1UaD+t2NNzOiQFzaZ
UNRs77Anm1/E6Yu3SRCJC5rUI4NZ3YqDUhI3B5OSdXiPQS1CHM6MoUI7fyVWsNJ9
hR/zTVAdB11nN6UAmGKyn36Xt1nIWlSJW/uPzT/sMu+xlfdHCqtzEsOYPzPNLVc4
YGUwEl++NogWZohJpH1FpWRkzpQaEAXodYDNABEBAAG0MVRyaXN0YW4gQ29sZ2F0
ZSAoYSB0ZXN0IGtleSkgPHRjb2xnYXRlQGdtYWlsLmNvbT6JATgEEwECACIFAlOI
YWUCGwMGCwkIBwMCBhUIAgkKCwQWAgMBAh4BAheAAAoJEK9K5quiVC/KtDYIAJ7b
I7jnzaBPjdzIpczdeOLT+AKA75NMXAqI0bWvjNjh2BvkPWcrmVuhYEkFYupiz2Vo
74AnSbd13iqrhtO9Ix9l5mer6Q03hN9JGWPuxWe31E46JWNUwj8pvy9M1WSjGu9I
HgLGef4vNDwMYCgKdTvJ8ckF2U/myUtt69AG3Kbo3xGucjhjGjqeKgSluqirDIl0
OfXKQUaHqKTZOa3MTKDYStycosFfA1ukOHD8wu25F7/m5SR9o8t/yDboHm2ClO67
1xIowZtFfN1inRsONWA9lC3fHeiffu4E6FzbkUq+vxGsS4XIWkSrvTDKDDYJelah
H/nuZh6Jugj+APniCIi5AQ0EU4hhZQEIAMHKn/mYgVJLSeJyIXnbcOqUXAyT6BPW
iF7p29g/y1j6qBWsOf25DyAg9fShBu1Ay3+UkFERgXePc68s0rZG2aRRY8t/kj0p
znx9ePW3wp6NjV4MSez8wzD7VDtFfWjI9HNV0vjre8sdfNbjHVKljg5zCMD8hFCQ
Tk5tMLAyBnaZTg7qWq9m2l+lO6ODlyQlfmN/4F9c8dvt+oRKvPRo2r6itg7CaX9Y
0+Ab9IvX8dIiqqJRZEQ6VHeYFUy1cdePx3ZUqgYkmVLwT7vXJlD7M4MtoYjWP9F9
/dmwzxgM/duKCdURI4hs3y2mAibRM5O9TaVvZhLqdj67W8bO94E6+JsAEQEAAYkB
HwQYAQIACQUCU4hhZQIbDAAKCRCvSuarolQvymmZB/9Mq/wmmy0DKHov2kJ9ZdLb
ULTghTMPHznVn5lrTu4DuBqocgw7fwLwTwYHp2gy4naJtOq7hhLhWmwsRW7C+51H
wl9Lz93KPPPI0g5tBy8tqq6wcUfmnsD71SKeBqd1v0TcuKNzj++pi/oGmfkS2d2b
Loj/6OElbUyzhAbXunJcZ/aBm+u5DPqYDr1XXg/8bp5UZ7h9vYuxPLGRyhSPdL7I
E6ZHucq5u1AleysWUUt4llmcJ69jtFK69PV4+Q1It+iFAWrS+H2kwMsZAUir2edO
7uk10RWxE1m3xhzU13Hke3osxYrN6BWViZzGUEeXRYk+DfSVegNT7dm92upzixMA
=vk0u
-----END PGP PUBLIC KEY BLOCK-----`

// Check that we output unknown fields in a consistant order
func TestDebControlVerify(t *testing.T) {
	var kr openpgp.EntityList
	kreader := strings.NewReader(testKrStr)
	kr, _ = openpgp.ReadArmoredKeyRing(kreader)

	c, err := ParseDebianControl(strings.NewReader(testControlSignedValid), kr)

	if err != nil {
		t.Errorf("ParseDebianControl failed, %s", err.Error())
		return
	}

	if !c.Signed {
		t.Errorf("ControlFile should have beenn signed")
		return
	}

	if !c.SignatureVerified {
		t.Errorf("ControlFile should have beenn verified")
		return
	}

	if len(c.SignedBy) != 1 {
		t.Errorf("Wrong signature count")
		return
	}

	signer := (c.SignedBy)[0]
	if signer != "Tristan Colgate (a test key) <tcolgate@gmail.com>" {
		t.Errorf("Wrong signer, %s", signer)
		return
	}
}

// Check that we output unknown fields in a consistant order
func TestDebControlUnknownSigner(t *testing.T) {
	kr := openpgp.EntityList{}

	c, err := ParseDebianControl(strings.NewReader(testControlSignedValid), kr)

	if err != nil {
		t.Errorf("ParseDebianControl failed, %s", err.Error())
		return
	}

	if !c.Signed {
		t.Errorf("ControlFile should have beenn signed")
		return
	}

	if c.SignatureVerified {
		t.Errorf("ControlFile should not have beenn verified")
		return
	}

	if len(c.SignedBy) != 0 {
		t.Errorf("Wrong signature count, %v", len(c.SignedBy))
		return
	}
}

var testControlSignedInvalid = `-----BEGIN PGP SIGNED MESSAGE-----
Hash: SHA1

Checksums-Sha1:  
 7670839693da39075134c1a1f5faad6623df70af 2416 collectd_5.4.0-3.dsc
 8f06307bf1c17b83351fccdd7a93b4a822ecbdf4 76417 collectd_5.4.0-3.diff.gz
 8f06307bf1c17b83351fccdd7a93b4a822ecbdf5 76418 EVILFILE.deb
Checksums-Sha256: 
 c0679d60f28ceecd09b3c000361c691e373dba599e3878135bc36bede14e109d 2416 collectd_5.4.0-3.dsc
 e6d7f21737d2146a9bb30a46137fbd0f00be7971e8c3edc6e66a5981498a261e 76417 collectd_5.4.0-3.diff.gz
Files: 
 cd9aa41b337352fd160f326523a9c3d8 2416 utils optional collectd_5.4.0-3.dsc
 d1270867f1c9517dd92016ea9f2d5afe 76417 utils optional collectd_5.4.0-3.diff.gz
-----BEGIN PGP SIGNATURE-----
Version: GnuPG v1

iQEcBAEBAgAGBQJTiG5VAAoJEK9K5quiVC/K9IkH/3nBYgl89/DnjI6ESksYrtkH
Zv/7Cpfg7YsvUmMrghh1hWklK8zj2Tm3NgKshW/HF7orn5cmPUMVZZU8EFa5uR43
FEJ+r1zA40dcOUtKgRESke4b8MNebWq01Op7zrlU1w/4fZ2MFxhNiQ4Xr3ziEl61
kPPR+1ZG43h+wy1h6QXzcNdqcwUnCfX4UqlhZ1giJ1/1qEdW6HlLOiIomZLGhg6B
K5JmdVY4fiH1Fv0tSq7mVnN7LbXJBo8KyzbqJAFkWNu0zh/G/5whz6n2ohgWC/SJ
uiTJMgKpAOxBFeEzO1quFyWnQePIjQ2zWVaTwqDPiZNQ6+377gCrC4Fu+SYdmlQ=
=JixZ
-----END PGP SIGNATURE-----
`

// Check that we output unknown fields in a consistant order
/*
func TestDebControlBadSignature(t *testing.T) {
	kr := openpgp.EntityList{}
	kreader := strings.NewReader(testKrStr)
	kr, _ = openpgp.ReadArmoredKeyRing(kreader)

	c, err := ParseDebianControl(strings.NewReader(testControlSignedInvalid), kr)
	if err != nil {
		t.Errorf("ParseDebianControl failed, %s", err.Error())
		return
	}

	if c.Signed {
		t.Errorf("ControlFile should not have beenn signed")
		return
	}

	if c.SignatureVerified {
		t.Errorf("ControlFile should not have beenn verified")
		return
	}

	if len(c.SignedBy) != 0 {
		t.Errorf("Wrong signature count")
		return
	}
}
*/