func TestLoadsV1DataCorrectly(t *testing.T) {
cfg := &config.Config{DataDir: filepath.Join(".", "testdata", "v1", "1")}
taskEngine := engine.NewTaskEngine(&config.Config{}, nil, nil, nil)
var containerInstanceArn, cluster, savedInstanceID string
var sequenceNumber int64
stateManager, err := statemanager.NewStateManager(cfg,
statemanager.AddSaveable("TaskEngine", taskEngine),
statemanager.AddSaveable("ContainerInstanceArn", &containerInstanceArn),
statemanager.AddSaveable("Cluster", &cluster),
statemanager.AddSaveable("EC2InstanceID", &savedInstanceID),
statemanager.AddSaveable("SeqNum", &sequenceNumber),
)
if err != nil {
t.Fatal(err)
}
err = stateManager.Load()
if err != nil {
t.Fatal("Error loading state", err)
}
if cluster != "test" {
t.Fatal("Wrong cluster: " + cluster)
}
if sequenceNumber != 0 {
t.Fatal("v1 should give a sequence number of 0")
}
tasks, err := taskEngine.ListTasks()
if err != nil {
t.Fatal(err)
}
var deadTask *api.Task
for _, task := range tasks {
if task.Arn == "arn:aws:ecs:us-west-2:1234567890:task/f44b4fc9-adb0-4f4f-9dff-871512310588" {
deadTask = task
}
}
if deadTask == nil {
t.Fatal("Could not find task expected to be in state")
}
if deadTask.SentStatus != api.TaskStopped {
t.Fatal("task dead should be stopped now")
}
if deadTask.Containers[0].SentStatus != api.ContainerStopped {
t.Fatal("container Dead should go to stopped")
}
if deadTask.Containers[0].DesiredStatus != api.ContainerStopped {
t.Fatal("container Dead should go to stopped")
}
if deadTask.Containers[0].KnownStatus != api.ContainerStopped {
t.Fatal("container Dead should go to stopped")
}
expected, _ := time.Parse(time.RFC3339, "2015-04-28T17:29:48.129140193Z")
if deadTask.KnownStatusTime != expected {
t.Fatal("Time was not correct")
}
}
func initializeStateManager(cfg *config.Config, taskEngine engine.TaskEngine, cluster, containerInstanceArn, savedInstanceID *string, sequenceNumber *utilatomic.IncreasingInt64) (statemanager.StateManager, error) {
if !cfg.Checkpoint {
return statemanager.NewNoopStateManager(), nil
}
stateManager, err := statemanager.NewStateManager(cfg,
statemanager.AddSaveable("TaskEngine", taskEngine),
statemanager.AddSaveable("ContainerInstanceArn", containerInstanceArn),
statemanager.AddSaveable("Cluster", cluster),
statemanager.AddSaveable("EC2InstanceID", savedInstanceID),
statemanager.AddSaveable("ACSSeqNum", sequenceNumber),
)
if err != nil {
return nil, err
}
return stateManager, nil
}
func initializeStateManager(cfg *config.Config, taskEngine engine.TaskEngine, cluster, containerInstanceArn, savedInstanceID *string) (statemanager.StateManager, error) {
if !cfg.Checkpoint {
return statemanager.NewNoopStateManager(), nil
}
stateManager, err := statemanager.NewStateManager(cfg,
statemanager.AddSaveable("TaskEngine", taskEngine),
statemanager.AddSaveable("ContainerInstanceArn", containerInstanceArn),
statemanager.AddSaveable("Cluster", cluster),
statemanager.AddSaveable("EC2InstanceID", savedInstanceID),
//The ACSSeqNum field is retained for compatibility with statemanager.EcsDataVersion 4 and
//can be removed in the future with a version bump.
statemanager.AddSaveable("ACSSeqNum", 1),
)
if err != nil {
return nil, err
}
return stateManager, nil
}
func TestStateManager(t *testing.T) {
tmpDir, err := ioutil.TempDir("/tmp", "ecs_statemanager_test")
if err != nil {
t.Fatal(err)
}
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(tmpDir)
cfg := &config.Config{DataDir: tmpDir}
manager, err := statemanager.NewStateManager(cfg)
if err != nil {
t.Fatal("Error loading manager", err)
}
err = manager.Load()
if err != nil {
t.Error("Expected loading a non-existant file to not be an error")
}
// Now let's make some state to save
containerInstanceArn := ""
taskEngine := engine.NewTaskEngine(&config.Config{}, nil, nil, nil)
manager, err = statemanager.NewStateManager(cfg, statemanager.AddSaveable("TaskEngine", taskEngine), statemanager.AddSaveable("ContainerInstanceArn", &containerInstanceArn))
if err != nil {
t.Fatal(err)
}
containerInstanceArn = "containerInstanceArn"
testTask := &api.Task{Arn: "test-arn"}
taskEngine.(*engine.DockerTaskEngine).State().AddTask(testTask)
err = manager.Save()
if err != nil {
t.Fatal("Error saving state", err)
}
// Now make sure we can load that state sanely
loadedTaskEngine := engine.NewTaskEngine(&config.Config{}, nil, nil, nil)
var loadedContainerInstanceArn string
manager, err = statemanager.NewStateManager(cfg, statemanager.AddSaveable("TaskEngine", &loadedTaskEngine), statemanager.AddSaveable("ContainerInstanceArn", &loadedContainerInstanceArn))
if err != nil {
t.Fatal(err)
}
err = manager.Load()
if err != nil {
t.Fatal("Error loading state", err)
}
if loadedContainerInstanceArn != containerInstanceArn {
t.Error("Did not load containerInstanceArn correctly; got ", loadedContainerInstanceArn, " instead of ", containerInstanceArn)
}
if !engine_testutils.DockerTaskEnginesEqual(loadedTaskEngine.(*engine.DockerTaskEngine), (taskEngine.(*engine.DockerTaskEngine))) {
t.Error("Did not load taskEngine correctly")
}
// I'd rather double check .Equal there; let's make sure ListTasks agrees.
tasks, err := loadedTaskEngine.ListTasks()
if err != nil {
t.Error("Error listing tasks", err)
}
if len(tasks) != 1 {
t.Error("Should have a task!")
} else {
if tasks[0].Arn != "test-arn" {
t.Error("Wrong arn, expected test-arn but got ", tasks[0].Arn)
}
}
}