github.com/demonoid81/moby@v0.0.0-20200517203328-62dd8e17c460/container/memory_store_test.go (about)

     1  package container // import "github.com/demonoid81/moby/container"
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  )
     7  
     8  func TestNewMemoryStore(t *testing.T) {
     9  	s := NewMemoryStore()
    10  	m, ok := s.(*memoryStore)
    11  	if !ok {
    12  		t.Fatalf("store is not a memory store %v", s)
    13  	}
    14  	if m.s == nil {
    15  		t.Fatal("expected store map to not be nil")
    16  	}
    17  }
    18  
    19  func TestAddContainers(t *testing.T) {
    20  	s := NewMemoryStore()
    21  	s.Add("id", NewBaseContainer("id", "root"))
    22  	if s.Size() != 1 {
    23  		t.Fatalf("expected store size 1, got %v", s.Size())
    24  	}
    25  }
    26  
    27  func TestGetContainer(t *testing.T) {
    28  	s := NewMemoryStore()
    29  	s.Add("id", NewBaseContainer("id", "root"))
    30  	c := s.Get("id")
    31  	if c == nil {
    32  		t.Fatal("expected container to not be nil")
    33  	}
    34  }
    35  
    36  func TestDeleteContainer(t *testing.T) {
    37  	s := NewMemoryStore()
    38  	s.Add("id", NewBaseContainer("id", "root"))
    39  	s.Delete("id")
    40  	if c := s.Get("id"); c != nil {
    41  		t.Fatalf("expected container to be nil after removal, got %v", c)
    42  	}
    43  
    44  	if s.Size() != 0 {
    45  		t.Fatalf("expected store size to be 0, got %v", s.Size())
    46  	}
    47  }
    48  
    49  func TestListContainers(t *testing.T) {
    50  	s := NewMemoryStore()
    51  
    52  	cont := NewBaseContainer("id", "root")
    53  	cont.Created = time.Now()
    54  	cont2 := NewBaseContainer("id2", "root")
    55  	cont2.Created = time.Now().Add(24 * time.Hour)
    56  
    57  	s.Add("id", cont)
    58  	s.Add("id2", cont2)
    59  
    60  	list := s.List()
    61  	if len(list) != 2 {
    62  		t.Fatalf("expected list size 2, got %v", len(list))
    63  	}
    64  	if list[0].ID != "id2" {
    65  		t.Fatalf("expected id2, got %v", list[0].ID)
    66  	}
    67  }
    68  
    69  func TestFirstContainer(t *testing.T) {
    70  	s := NewMemoryStore()
    71  
    72  	s.Add("id", NewBaseContainer("id", "root"))
    73  	s.Add("id2", NewBaseContainer("id2", "root"))
    74  
    75  	first := s.First(func(cont *Container) bool {
    76  		return cont.ID == "id2"
    77  	})
    78  
    79  	if first == nil {
    80  		t.Fatal("expected container to not be nil")
    81  	}
    82  	if first.ID != "id2" {
    83  		t.Fatalf("expected id2, got %v", first)
    84  	}
    85  }
    86  
    87  func TestApplyAllContainer(t *testing.T) {
    88  	s := NewMemoryStore()
    89  
    90  	s.Add("id", NewBaseContainer("id", "root"))
    91  	s.Add("id2", NewBaseContainer("id2", "root"))
    92  
    93  	s.ApplyAll(func(cont *Container) {
    94  		if cont.ID == "id2" {
    95  			cont.ID = "newID"
    96  		}
    97  	})
    98  
    99  	cont := s.Get("id2")
   100  	if cont == nil {
   101  		t.Fatal("expected container to not be nil")
   102  	}
   103  	if cont.ID != "newID" {
   104  		t.Fatalf("expected newID, got %v", cont.ID)
   105  	}
   106  }