package container // import "github.com/docker/docker/container"

 
 import (
 	"testing"
 	"time"
 )
 
 func TestNewMemoryStore(t *testing.T) {
 	s := NewMemoryStore()
 	m, ok := s.(*memoryStore)
 	if !ok {
 		t.Fatalf("store is not a memory store %v", s)
 	}
 	if m.s == nil {
 		t.Fatal("expected store map to not be nil")
 	}
 }
 
 func TestAddContainers(t *testing.T) {
 	s := NewMemoryStore()
 	s.Add("id", NewBaseContainer("id", "root"))
 	if s.Size() != 1 {
 		t.Fatalf("expected store size 1, got %v", s.Size())
 	}
 }
 
 func TestGetContainer(t *testing.T) {
 	s := NewMemoryStore()
 	s.Add("id", NewBaseContainer("id", "root"))
 	c := s.Get("id")
 	if c == nil {
 		t.Fatal("expected container to not be nil")
 	}
 }
 
 func TestDeleteContainer(t *testing.T) {
 	s := NewMemoryStore()
 	s.Add("id", NewBaseContainer("id", "root"))
 	s.Delete("id")
 	if c := s.Get("id"); c != nil {
 		t.Fatalf("expected container to be nil after removal, got %v", c)
 	}
 
 	if s.Size() != 0 {
 		t.Fatalf("expected store size to be 0, got %v", s.Size())
 	}
 }
 
 func TestListContainers(t *testing.T) {
 	s := NewMemoryStore()
 
 	cont := NewBaseContainer("id", "root")
 	cont.Created = time.Now()
 	cont2 := NewBaseContainer("id2", "root")
 	cont2.Created = time.Now().Add(24 * time.Hour)
 
 	s.Add("id", cont)
 	s.Add("id2", cont2)
 
 	list := s.List()
 	if len(list) != 2 {
 		t.Fatalf("expected list size 2, got %v", len(list))
 	}
 	if list[0].ID != "id2" {

 		t.Fatalf("expected id2, got %v", list[0].ID)

 	}
 }
 
 func TestFirstContainer(t *testing.T) {
 	s := NewMemoryStore()
 
 	s.Add("id", NewBaseContainer("id", "root"))
 	s.Add("id2", NewBaseContainer("id2", "root"))
 
 	first := s.First(func(cont *Container) bool {
 		return cont.ID == "id2"
 	})
 
 	if first == nil {
 		t.Fatal("expected container to not be nil")
 	}
 	if first.ID != "id2" {
 		t.Fatalf("expected id2, got %v", first)
 	}
 }
 
 func TestApplyAllContainer(t *testing.T) {
 	s := NewMemoryStore()
 
 	s.Add("id", NewBaseContainer("id", "root"))
 	s.Add("id2", NewBaseContainer("id2", "root"))
 
 	s.ApplyAll(func(cont *Container) {
 		if cont.ID == "id2" {
 			cont.ID = "newID"
 		}
 	})
 
 	cont := s.Get("id2")
 	if cont == nil {
 		t.Fatal("expected container to not be nil")
 	}
 	if cont.ID != "newID" {

 		t.Fatalf("expected newID, got %v", cont.ID)

 	}
 }