src/pkg/container/list/list_test.go

// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package list

import (
	"container/list";
	"testing";
)

func checkListPointers(t *testing.T, l *List, es []*Element) {
	if len(es) == 0 {
		if l.front != nil || l.back != nil {
			t.Errorf("l.front/l.back = %v/%v should be nil/nil", l.front, l.back);
		}
		return
	}

	if l.front != es[0] {
		t.Errorf("l.front = %v, want %v", l.front, es[0]);
	}
	if last := es[len(es)-1]; l.back != last {
		t.Errorf("l.back = %v, want %v", l.back, last);
	}

	for i := 0; i < len(es); i++ {
		e := es[i];
		var e_prev, e_next *Element = nil, nil;
		if i > 0 {
			e_prev = es[i-1];
		}
		if i < len(es) - 1 {
			e_next = es[i+1];
		}
		if e.prev != e_prev {
			t.Errorf("elt #%d (%v) has prev=%v, want %v", i, e, e.prev, e_prev);
		}
		if e.next != e_next {
			t.Errorf("elt #%d (%v) has next=%v, want %v", i, e, e.next, e_next);
		}
	}
}

func TestList(t *testing.T) {
	l := New();
	checkListPointers(t, l, []*Element{});

	// Single element list
	e := l.PushFront("a");
	checkListPointers(t, l, []*Element{ e });
	l.MoveToFront(e);
	checkListPointers(t, l, []*Element{ e });
	l.MoveToBack(e);
	checkListPointers(t, l, []*Element{ e });
	l.Remove(e);
	checkListPointers(t, l, []*Element{});

	// Bigger list
	e2 := l.PushFront(2);
	e1 := l.PushFront(1);
	e3 := l.PushBack(3);
	e4 := l.PushBack("banana");
	checkListPointers(t, l, []*Element{ e1, e2, e3, e4 });

	l.Remove(e2);
	checkListPointers(t, l, []*Element{ e1, e3, e4 });

	l.MoveToFront(e3);  // move from middle
	checkListPointers(t, l, []*Element{ e3, e1, e4 });

	l.MoveToFront(e1);
	l.MoveToBack(e3);  // move from middle
	checkListPointers(t, l, []*Element{ e1, e4, e3 });

	l.MoveToFront(e3);  // move from back
	checkListPointers(t, l, []*Element{ e3, e1, e4 });
	l.MoveToFront(e3);  // should be no-op
	checkListPointers(t, l, []*Element{ e3, e1, e4 });

	l.MoveToBack(e3);  // move from front
	checkListPointers(t, l, []*Element{ e1, e4, e3 });
	l.MoveToBack(e3);  // should be no-op
	checkListPointers(t, l, []*Element{ e1, e4, e3 });

	// Clear all elements by iterating
	for e := range l.Iter() {
		l.Remove(e);
	}
	checkListPointers(t, l, []*Element{});
}