Function TestReportAndApply

engine/engine_test.go:249–313

(t *testing.T)

}

func TestReportAndApply(t *testing.T) {
	nod, _, sto := prepare(t)
	mockCtl := gomock.NewController(t)
	defer mockCtl.Finish()
	mockAmi := mock.NewMockAMI(mockCtl)
	mockSync := mock.NewMockSync(mockCtl)
	eng := engineImpl{
		ami: mockAmi,
		cfg: config.Config{},
		sto: sto,
		syn: mockSync,
		nod: nod,
		log: log.With(log.Any("engine", "test")),
	}
	assert.NotNil(t, eng)
	infos := map[string]interface{}{}
	stats := map[string]interface{}{}

	t.Setenv(kube.KubeNodeName, "knn")
	mockAmi.EXPECT().CollectNodeInfo().Return(infos, nil)
	mockAmi.EXPECT().CollectNodeStats().Return(stats, nil)
	appStats := []specv1.AppStats{{AppInfo: specv1.AppInfo{Name: "app1", Version: "v1"}}, {AppInfo: specv1.AppInfo{Name: "app2", Version: "v2"}}}
	mockAmi.EXPECT().StatsApps(gomock.Any()).Return(appStats, nil)
	mockAmi.EXPECT().GetModeInfo().Return("modeinfo", nil)

	reApp := specv1.Report{
		"apps": []specv1.AppInfo{{Name: "app1", Version: "v1"}, {Name: "app2", Version: "v2"}},
	}
	deApp := specv1.Desire{
		"apps": []specv1.AppInfo{{Name: "app2", Version: "v2"}, {Name: "app3", Version: "v3"}},
	}
	_, err := nod.Report(reApp, false)
	assert.NoError(t, err)
	_, err = nod.Desire(deApp, false)
	assert.NoError(t, err)

	app1 := specv1.Application{Name: "app1", Version: "v1"}
	err = sto.Upsert(makeKey(specv1.KindApplication, "app1", "v1"), app1)
	assert.NoError(t, err)
	app3 := specv1.Application{Name: "app3", Version: "v3"}
	err = sto.Upsert(makeKey(specv1.KindApplication, "app3", "v3"), app3)
	mockSync.EXPECT().SyncResource(gomock.Any()).Return(nil)
	mockSync.EXPECT().SyncApps(gomock.Any()).Return(nil, nil)
	mockAmi.EXPECT().ApplyApp(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
	mockAmi.EXPECT().DeleteApp(gomock.Any(), gomock.Any()).Return(nil)
	err = eng.reportAndApply(false, true, nil)
	assert.NoError(t, err)

	// desire app is nil
	mockAmi.EXPECT().CollectNodeInfo().Return(nil, nil)
	mockAmi.EXPECT().CollectNodeStats().Return(nil, nil)
	appStats = []specv1.AppStats{{AppInfo: specv1.AppInfo{Name: "app1", Version: "v1"}}}
	mockAmi.EXPECT().StatsApps(gomock.Any()).Return(appStats, nil)
	mockAmi.EXPECT().GetModeInfo().Return("modeinfo", nil)
	reApp = specv1.Report{
		"apps": []specv1.AppInfo{{Name: "app1", Version: "v1"}},
	}
	deApp = specv1.Desire{"apps": nil}