snowflake/common/packetpadding/container_test.go

package packetpadding_test

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"

	"gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake/v2/common/packetpadding"
)

func TestPacketPaddingContainer(t *testing.T) {
	Convey("Given a PacketPaddingContainer", t, func() {

		Convey("When packing data with padding", func() {
			data := []byte("testdata")
			paddingLength := 4
			packedData := packetpadding.Pack(data, paddingLength)

			Convey("The packed data should have the correct length", func() {
				expectedLength := len(data) + paddingLength + 2
				So(len(packedData), ShouldEqual, expectedLength)
			})

			Convey("When unpacking the packed data", func() {
				unpackedData, unpackedPaddingLength := packetpadding.Unpack(packedData)

				Convey("The unpacked data should match the original data", func() {
					So(string(unpackedData), ShouldEqual, string(data))
				})

				Convey("The unpacked padding length should match the original padding length", func() {
					So(unpackedPaddingLength, ShouldEqual, paddingLength)
				})
			})
		})

		Convey("When packing empty data with padding", func() {
			data := []byte("")
			paddingLength := 4
			packedData := packetpadding.Pack(data, paddingLength)

			Convey("The packed data should have the correct length", func() {
				expectedLength := len(data) + paddingLength + 2
				So(len(packedData), ShouldEqual, expectedLength)
			})

			Convey("When unpacking the packed data", func() {
				unpackedData, unpackedPaddingLength := packetpadding.Unpack(packedData)

				Convey("The unpacked data should match the original data", func() {
					So(string(unpackedData), ShouldEqual, string(data))
				})

				Convey("The unpacked padding length should match the original padding length", func() {
					So(unpackedPaddingLength, ShouldEqual, paddingLength)
				})
			})
		})

		Convey("When packing data with zero padding", func() {
			data := []byte("testdata")
			paddingLength := 0
			packedData := packetpadding.Pack(data, paddingLength)

			Convey("The packed data should have the correct length", func() {
				expectedLength := len(data) + paddingLength + 2
				So(len(packedData), ShouldEqual, expectedLength)
			})

			Convey("When unpacking the packed data", func() {
				unpackedData, unpackedPaddingLength := packetpadding.Unpack(packedData)

				Convey("The unpacked data should match the original data", func() {
					So(string(unpackedData), ShouldEqual, string(data))
				})

				Convey("The unpacked padding length should match the original padding length", func() {
					So(unpackedPaddingLength, ShouldEqual, paddingLength)
				})
			})
		})

		Convey("When padding data", func() {
			Convey("With a positive padding length", func() {
				padLength := 3
				padData := packetpadding.Pad(padLength)

				Convey("The padded data should have the correct length", func() {
					So(len(padData), ShouldEqual, padLength)
				})
			})

			Convey("With a zero padding length", func() {
				padLength := 0
				padData := packetpadding.Pad(padLength)

				Convey("The padded data should be empty", func() {
					So(len(padData), ShouldEqual, 0)
				})
			})

			Convey("With a negative padding length", func() {
				padLength := -1
				padData := packetpadding.Pad(padLength)

				Convey("The padded data should be nil", func() {
					So(padData, ShouldBeNil)
				})
			})
		})
	})
}