func testStorageStoreRemoveContainer(storage cluster.Storage, t *testing.T) {
err := storage.StoreContainer("container-1", "host-9")
assertIsNil(err, t)
err = storage.RemoveContainer("container-1")
assertIsNil(err, t)
_, err = storage.RetrieveContainer("container-1")
if err != cstorage.ErrNoSuchContainer {
t.Errorf("Error should be cstorage.ErrNoSuchContainer, received: %s", err)
}
}
func testStorageStoreRetrieveContainer(storage cluster.Storage, t *testing.T) {
defer storage.RemoveContainer("container-1")
defer storage.RemoveContainer("container-2")
defer storage.RemoveContainer("container-3")
err := storage.StoreContainer("container-1", "host-1")
assertIsNil(err, t)
err = storage.StoreContainer("container-2", "host-1")
assertIsNil(err, t)
err = storage.StoreContainer("container-3", "host-2")
assertIsNil(err, t)
host, err := storage.RetrieveContainer("container-1")
assertIsNil(err, t)
if host != "host-1" {
t.Errorf("Unexpected hostname %s - expected %s", host, "host-1")
}
host, err = storage.RetrieveContainer("container-2")
assertIsNil(err, t)
if host != "host-1" {
t.Errorf("Unexpected hostname %s - expected %s", host, "host-1")
}
host, err = storage.RetrieveContainer("container-3")
assertIsNil(err, t)
if host != "host-2" {
t.Errorf("Unexpected hostname %s - expected %s", host, "host-2")
}
}
func testRetrieveContainers(storage cluster.Storage, t *testing.T) {
defer storage.RemoveContainer("container-1")
defer storage.RemoveContainer("container-2")
defer storage.RemoveContainer("container-3")
err := storage.StoreContainer("container-1", "host-1")
assertIsNil(err, t)
err = storage.StoreContainer("container-2", "host-1")
assertIsNil(err, t)
err = storage.StoreContainer("container-3", "host-2")
assertIsNil(err, t)
containers, err := storage.RetrieveContainers()
assertIsNil(err, t)
if len(containers) != 3 {
t.Errorf("Unexpected len %d - expected %d", len(containers), 3)
}
}