How to use NewController method of gomock Package

...23	"github.com/ZTE/Knitter/pkg/iaas-accessor"24	"github.com/golang/gostub"25)26func TestCreateRouterErrNotCfgIaaS(t *testing.T) {27	cfgMock := gomock.NewController(t)28	defer cfgMock.Finish()29	mockDB := NewMockDbAccessor(cfgMock)30	common.SetDataBase(mockDB)31	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)32	defer stubs.Reset()33	MockPaasAdminCheck(mockDB)34	cfg := string(`{"router":{"name":"test-create-router"}}`)35	resp := CreateRouter(cfg)36	Convey("TestCreateRouterErrNotCfgIaaS\n", t, func() {37		So(resp.Code, ShouldEqual, 401)38	})39}40func TestCreateRouterErr403(t *testing.T) {41	cfgMock := gomock.NewController(t)42	defer cfgMock.Finish()43	mockDB := NewMockDbAccessor(cfgMock)44	mockIaaS := NewMockIaaS(cfgMock)45	common.SetDataBase(mockDB)46	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)47	defer stubs.Reset()48	MockPaasAdminCheck(mockDB)49	cfg := string(`{"router":{"name":"test-create-router"}`)50	resp := CreateRouter(cfg)51	Convey("TestCreateNetworkErrNotCfgIaaS\n", t, func() {52		So(resp.Code, ShouldEqual, 403)53	})54}55func TestCreateRouterErr406A(t *testing.T) {56	cfgMock := gomock.NewController(t)57	defer cfgMock.Finish()58	mockDB := NewMockDbAccessor(cfgMock)59	mockIaaS := NewMockIaaS(cfgMock)60	common.SetDataBase(mockDB)61	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)62	defer stubs.Reset()63	MockPaasAdminCheck(mockDB)64	mockIaaS.EXPECT().CreateRouter(gomock.Any(),65		gomock.Any()).Return("", errors.New("create-router-error"))66	cfg := string(`{"router":{"name":"test-create-router"}}`)67	resp := CreateRouter(cfg)68	Convey("TestCreateRouterErr406A\n", t, func() {69		So(resp.Code, ShouldEqual, 406)70	})71}72func TestCreateRouterErr406B(t *testing.T) {73	cfgMock := gomock.NewController(t)74	defer cfgMock.Finish()75	mockDB := NewMockDbAccessor(cfgMock)76	mockIaaS := NewMockIaaS(cfgMock)77	common.SetDataBase(mockDB)78	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)79	defer stubs.Reset()80	MockPaasAdminCheck(mockDB)81	mockIaaS.EXPECT().CreateRouter(gomock.Any(),82		gomock.Any()).Return("router-uuid", nil)83	mockIaaS.EXPECT().GetRouter(gomock.Any()).Return(84		nil, errors.New("get-router-error"))85	cfg := string(`{"router":{"name":"test-create-router"}}`)86	resp := CreateRouter(cfg)87	Convey("TestCreateRouterErr406B\n", t, func() {88		So(resp.Code, ShouldEqual, 406)89	})90}91func TestCreateRouterErr406C(t *testing.T) {92	cfgMock := gomock.NewController(t)93	defer cfgMock.Finish()94	mockDB := NewMockDbAccessor(cfgMock)95	mockIaaS := NewMockIaaS(cfgMock)96	common.SetDataBase(mockDB)97	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)98	defer stubs.Reset()99	MockPaasAdminCheck(mockDB)100	mockIaaS.EXPECT().CreateRouter(gomock.Any(),101		gomock.Any()).Return("router-uuid", nil)102	router := iaasaccessor.Router{Id: "router-uuid"}103	mockIaaS.EXPECT().GetRouter(gomock.Any()).Return(104		&router, nil)105	mockDB.EXPECT().SaveLeaf(gomock.Any(),106		gomock.Any()).Return(errors.New("error"))107	mockIaaS.EXPECT().DeleteRouter(gomock.Any()).Return(nil)108	cfg := string(`{"router":{"name":"test-create-router"}}`)109	resp := CreateRouter(cfg)110	Convey("TestCreateRouterErr406C\n", t, func() {111		So(resp.Code, ShouldEqual, 406)112	})113}114func TestCreateRouterOK(t *testing.T) {115	cfgMock := gomock.NewController(t)116	defer cfgMock.Finish()117	mockDB := NewMockDbAccessor(cfgMock)118	mockIaaS := NewMockIaaS(cfgMock)119	common.SetDataBase(mockDB)120	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)121	defer stubs.Reset()122	MockPaasAdminCheck(mockDB)123	mockIaaS.EXPECT().CreateRouter(gomock.Any(),124		gomock.Any()).Return("router-uuid", nil)125	router := iaasaccessor.Router{Id: "router-uuid"}126	mockIaaS.EXPECT().GetRouter(gomock.Any()).Return(127		&router, nil)128	mockDB.EXPECT().SaveLeaf(gomock.Any(),129		gomock.Any()).Return(nil)130	cfg := string(`{"router":{"name":"test-create-router"}}`)131	resp := CreateRouter(cfg)132	Convey("TestCreateRouterOK\n", t, func() {133		So(resp.Code, ShouldEqual, 200)134	})135}136func TestDeleteRouterERR401(t *testing.T) {137	cfgMock := gomock.NewController(t)138	defer cfgMock.Finish()139	mockDB := NewMockDbAccessor(cfgMock)140	common.SetDataBase(mockDB)141	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)142	defer stubs.Reset()143	MockPaasAdminCheck(mockDB)144	resp := DeleteRouter("the-uuid-of-router")145	Convey("TestDeleteRouterERR401\n", t, func() {146		So(resp.Code, ShouldEqual, 401)147	})148}149func TestDeleteRouterERR404(t *testing.T) {150	cfgMock := gomock.NewController(t)151	defer cfgMock.Finish()152	mockDB := NewMockDbAccessor(cfgMock)153	mockIaaS := NewMockIaaS(cfgMock)154	common.SetDataBase(mockDB)155	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)156	defer stubs.Reset()157	MockPaasAdminCheck(mockDB)158	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(159		"", errors.New("read-router-return-error"))160	resp := DeleteRouter("the-uuid-of-router")161	Convey("TestDeleteRouterERR404\n", t, func() {162		So(resp.Code, ShouldEqual, 404)163	})164}165func TestDeleteRouter404A(t *testing.T) {166	cfgMock := gomock.NewController(t)167	defer cfgMock.Finish()168	mockDB := NewMockDbAccessor(cfgMock)169	mockIaaS := NewMockIaaS(cfgMock)170	common.SetDataBase(mockDB)171	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)172	defer stubs.Reset()173	MockPaasAdminCheck(mockDB)174	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(175		"router-uuid", nil)176	mockIaaS.EXPECT().DeleteRouter(gomock.Any()).Return(177		errors.New("error"))178	resp := DeleteRouter("the-uuid-of-router")179	Convey("TestDeleteRouter404A\n", t, func() {180		So(resp.Code, ShouldEqual, 404)181	})182}183func TestDeleteRouter404B(t *testing.T) {184	cfgMock := gomock.NewController(t)185	defer cfgMock.Finish()186	mockDB := NewMockDbAccessor(cfgMock)187	mockIaaS := NewMockIaaS(cfgMock)188	common.SetDataBase(mockDB)189	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)190	defer stubs.Reset()191	MockPaasAdminCheck(mockDB)192	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(193		"router-uuid", nil)194	mockIaaS.EXPECT().DeleteRouter(gomock.Any()).Return(nil)195	mockDB.EXPECT().DeleteLeaf(gomock.Any()).Return(196		errors.New("error"))197	resp := DeleteRouter("the-uuid-of-router")198	Convey("TestDeleteRouter404B\n", t, func() {199		So(resp.Code, ShouldEqual, 404)200	})201}202func TestDeleteRouter404C(t *testing.T) {203	cfgMock := gomock.NewController(t)204	defer cfgMock.Finish()205	mockDB := NewMockDbAccessor(cfgMock)206	mockIaaS := NewMockIaaS(cfgMock)207	common.SetDataBase(mockDB)208	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)209	defer stubs.Reset()210	MockPaasAdminCheck(mockDB)211	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(212		"router-uuid", nil)213	mockIaaS.EXPECT().DeleteRouter(gomock.Any()).Return(nil)214	mockDB.EXPECT().DeleteLeaf(gomock.Any()).Return(nil)215	mockDB.EXPECT().DeleteDir(gomock.Any()).Return(216		errors.New("error"))217	resp := DeleteRouter("the-uuid-of-router")218	Convey("TestDeleteRouter404C\n", t, func() {219		So(resp.Code, ShouldEqual, 404)220	})221}222func TestDeleteRouterOK(t *testing.T) {223	cfgMock := gomock.NewController(t)224	defer cfgMock.Finish()225	mockDB := NewMockDbAccessor(cfgMock)226	mockIaaS := NewMockIaaS(cfgMock)227	common.SetDataBase(mockDB)228	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)229	defer stubs.Reset()230	MockPaasAdminCheck(mockDB)231	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(232		"router-uuid", nil)233	mockIaaS.EXPECT().DeleteRouter(gomock.Any()).Return(nil)234	mockDB.EXPECT().DeleteLeaf(gomock.Any()).Return(nil)235	mockDB.EXPECT().DeleteDir(gomock.Any()).Return(nil)236	resp := DeleteRouter("the-uuid-of-router")237	Convey("TestDeleteRouterOK\n", t, func() {238		So(resp.Code, ShouldEqual, 200)239	})240}241func TestGetRouterERR401(t *testing.T) {242	cfgMock := gomock.NewController(t)243	defer cfgMock.Finish()244	mockDB := NewMockDbAccessor(cfgMock)245	common.SetDataBase(mockDB)246	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)247	defer stubs.Reset()248	MockPaasAdminCheck(mockDB)249	resp := GetRouter("the-uuid-of-router")250	Convey("TestGetRouterERR401\n", t, func() {251		So(resp.Code, ShouldEqual, 401)252	})253}254func TestGetRouterERR404B(t *testing.T) {255	cfgMock := gomock.NewController(t)256	defer cfgMock.Finish()257	mockDB := NewMockDbAccessor(cfgMock)258	mockIaaS := NewMockIaaS(cfgMock)259	common.SetDataBase(mockDB)260	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)261	defer stubs.Reset()262	MockPaasAdminCheck(mockDB)263	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(264		"", errors.New("read-router-from-etcd-error"))265	resp := GetRouter("the-uuid-of-router")266	Convey("TestGetRouterERR404B\n", t, func() {267		So(resp.Code, ShouldEqual, 404)268	})269}270func TestGetRouterOK(t *testing.T) {271	cfgMock := gomock.NewController(t)272	defer cfgMock.Finish()273	mockDB := NewMockDbAccessor(cfgMock)274	mockIaaS := NewMockIaaS(cfgMock)275	common.SetDataBase(mockDB)276	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)277	defer stubs.Reset()278	MockPaasAdminCheck(mockDB)279	router := string(`{280    "router": {281        "name": "router_show",282        "id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"283    }}`)284	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(285		router, nil)286	resp := GetRouter("the-uuid-of-router")287	Convey("TestGetRouterOK\n", t, func() {288		So(resp.Code, ShouldEqual, 200)289	})290}291var Config string = `{292		"router":{293			"name":"auto-test-create-router-should-be-delete"294			}295		}`296var errCfg string = `{297		"router":{298			"name":"auto-test-create-router-should-be-delete"299		}`300func TestUpdateRouterERR401(t *testing.T) {301	cfgMock := gomock.NewController(t)302	defer cfgMock.Finish()303	mockDB := NewMockDbAccessor(cfgMock)304	common.SetDataBase(mockDB)305	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)306	defer stubs.Reset()307	MockPaasAdminCheck(mockDB)308	resp := UpdateRouter("the-uuid-of-router", Config)309	Convey("TestGetRouterERR401\n", t, func() {310		So(resp.Code, ShouldEqual, 401)311	})312}313func TestUpdateRouterERR403(t *testing.T) {314	cfgMock := gomock.NewController(t)315	defer cfgMock.Finish()316	mockDB := NewMockDbAccessor(cfgMock)317	mockIaaS := NewMockIaaS(cfgMock)318	common.SetDataBase(mockDB)319	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)320	defer stubs.Reset()321	MockPaasAdminCheck(mockDB)322	resp := UpdateRouter("the-uuid-of-router", errCfg)323	Convey("TestUpdateRouterERR403\n", t, func() {324		So(resp.Code, ShouldEqual, 403)325	})326}327func TestUpdateRouterERR406A(t *testing.T) {328	cfgMock := gomock.NewController(t)329	defer cfgMock.Finish()330	mockDB := NewMockDbAccessor(cfgMock)331	mockIaaS := NewMockIaaS(cfgMock)332	common.SetDataBase(mockDB)333	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)334	defer stubs.Reset()335	MockPaasAdminCheck(mockDB)336	mockIaaS.EXPECT().UpdateRouter(gomock.Any(),337		gomock.Any(), gomock.Any()).Return(errors.New("error"))338	resp := UpdateRouter("the-uuid-of-router", Config)339	Convey("TestUpdateRouterERR406A\n", t, func() {340		So(resp.Code, ShouldEqual, 406)341	})342}343func TestUpdateRouterERR406B(t *testing.T) {344	cfgMock := gomock.NewController(t)345	defer cfgMock.Finish()346	mockDB := NewMockDbAccessor(cfgMock)347	mockIaaS := NewMockIaaS(cfgMock)348	common.SetDataBase(mockDB)349	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)350	defer stubs.Reset()351	MockPaasAdminCheck(mockDB)352	mockIaaS.EXPECT().UpdateRouter(gomock.Any(),353		gomock.Any(), gomock.Any()).Return(nil)354	mockIaaS.EXPECT().GetRouter(gomock.Any()).Return(355		nil, errors.New("error"))356	resp := UpdateRouter("the-uuid-of-router", Config)357	Convey("TestUpdateRouterERR406B\n", t, func() {358		So(resp.Code, ShouldEqual, 406)359	})360}361func TestUpdateRouterERR406C(t *testing.T) {362	cfgMock := gomock.NewController(t)363	defer cfgMock.Finish()364	mockDB := NewMockDbAccessor(cfgMock)365	mockIaaS := NewMockIaaS(cfgMock)366	common.SetDataBase(mockDB)367	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)368	defer stubs.Reset()369	MockPaasAdminCheck(mockDB)370	createRouter := iaasaccessor.Router{Id: "router-uuid", Name: "name-router"}371	mockIaaS.EXPECT().UpdateRouter(gomock.Any(),372		gomock.Any(), gomock.Any()).Return(nil)373	mockIaaS.EXPECT().GetRouter(gomock.Any()).Return(374		&createRouter, nil)375	mockDB.EXPECT().SaveLeaf(gomock.Any(),376		gomock.Any()).Return(errors.New("error"))377	resp := UpdateRouter("the-uuid-of-router", Config)378	Convey("TestUpdateRouterERR406B\n", t, func() {379		So(resp.Code, ShouldEqual, 406)380	})381}382func TestUpdateRouterOK(t *testing.T) {383	cfgMock := gomock.NewController(t)384	defer cfgMock.Finish()385	mockDB := NewMockDbAccessor(cfgMock)386	mockIaaS := NewMockIaaS(cfgMock)387	common.SetDataBase(mockDB)388	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)389	defer stubs.Reset()390	MockPaasAdminCheck(mockDB)391	createRouter := iaasaccessor.Router{Id: "router-uuid", Name: "name-router"}392	mockIaaS.EXPECT().UpdateRouter(gomock.Any(),393		gomock.Any(), gomock.Any()).Return(nil)394	mockIaaS.EXPECT().GetRouter(gomock.Any()).Return(395		&createRouter, nil)396	mockDB.EXPECT().SaveLeaf(gomock.Any(),397		gomock.Any()).Return(nil)398	resp := UpdateRouter("the-uuid-of-router", Config)399	Convey("TestUpdateRouterOK\n", t, func() {400		So(resp.Code, ShouldEqual, 200)401	})402}403func TestGetAllRouterERR401(t *testing.T) {404	cfgMock := gomock.NewController(t)405	defer cfgMock.Finish()406	mockDB := NewMockDbAccessor(cfgMock)407	common.SetDataBase(mockDB)408	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)409	defer stubs.Reset()410	MockPaasAdminCheck(mockDB)411	resp := GetAllRouter()412	Convey("TestGetAllRouterERR401\n", t, func() {413		So(resp.Code, ShouldEqual, 401)414	})415}416func TestGetAllRouterOK1(t *testing.T) {417	cfgMock := gomock.NewController(t)418	defer cfgMock.Finish()419	mockDB := NewMockDbAccessor(cfgMock)420	mockIaaS := NewMockIaaS(cfgMock)421	common.SetDataBase(mockDB)422	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)423	defer stubs.Reset()424	MockPaasAdminCheck(mockDB)425	mockDB.EXPECT().ReadDir(gomock.Any()).Return(nil,426		errors.New("read-dir-error"))427	resp := GetAllRouter()428	Convey("TestGetAllRouterOK1\n", t, func() {429		So(resp.Code, ShouldEqual, 200)430	})431}432func TestGetAllRouterOK2(t *testing.T) {433	cfgMock := gomock.NewController(t)434	defer cfgMock.Finish()435	mockDB := NewMockDbAccessor(cfgMock)436	mockIaaS := NewMockIaaS(cfgMock)437	common.SetDataBase(mockDB)438	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)439	defer stubs.Reset()440	MockPaasAdminCheck(mockDB)441	mockDB.EXPECT().ReadDir(gomock.Any()).Return(nil, nil)442	resp := GetAllRouter()443	Convey("TestGetAllRouterOK2\n", t, func() {444		So(resp.Code, ShouldEqual, 200)445	})446}447func TestGetAllRouterOK3(t *testing.T) {448	cfgMock := gomock.NewController(t)449	defer cfgMock.Finish()450	mockDB := NewMockDbAccessor(cfgMock)451	mockIaaS := NewMockIaaS(cfgMock)452	common.SetDataBase(mockDB)453	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)454	defer stubs.Reset()455	MockPaasAdminCheck(mockDB)456	router := string(`{457    "router": {458        "name": "network_show",459        "id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"460    }}`)461	var list1 []*client.Node462	node := client.Node{Key: "network-uuid", Value: "network-info"}463	list1 = append(list1, &node)464	list1 = append(list1, &node)465	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list1, nil)466	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(router, nil)467	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return("",468		errors.New("read-leaf-error"))469	resp := GetAllRouter()470	Convey("TestGetAllRouterOK3\n", t, func() {471		So(resp.Code, ShouldEqual, 200)472	})473}474func TestAttachERR401(t *testing.T) {475	cfgMock := gomock.NewController(t)476	defer cfgMock.Finish()477	mockDB := NewMockDbAccessor(cfgMock)478	common.SetDataBase(mockDB)479	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)480	defer stubs.Reset()481	MockPaasAdminCheck(mockDB)482	resp := AttachRouter("uuid-for-routet", "uuid-for-network")483	Convey("TestAttachERR401\n", t, func() {484		So(resp.Code, ShouldEqual, 401)485	})486}487func TestAttachERR403(t *testing.T) {488	cfgMock := gomock.NewController(t)489	defer cfgMock.Finish()490	mockDB := NewMockDbAccessor(cfgMock)491	mockIaaS := NewMockIaaS(cfgMock)492	common.SetDataBase(mockDB)493	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)494	defer stubs.Reset()495	MockPaasAdminCheck(mockDB)496	resp := AttachRouter403("uuid-for-routet", "uuid-for-network")497	Convey("TestAttachERR403\n", t, func() {498		So(resp.Code, ShouldEqual, 403)499	})500}501func TestAttachERR406A(t *testing.T) {502	cfgMock := gomock.NewController(t)503	defer cfgMock.Finish()504	mockDB := NewMockDbAccessor(cfgMock)505	mockIaaS := NewMockIaaS(cfgMock)506	common.SetDataBase(mockDB)507	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)508	defer stubs.Reset()509	MockPaasAdminCheck(mockDB)510	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(511		"", errors.New("error"))512	resp := AttachRouter("uuid-for-routet", "uuid-for-network")513	Convey("TestAttachERR401\n", t, func() {514		So(resp.Code, ShouldEqual, 406)515	})516}517func TestAttachERR406B(t *testing.T) {518	cfgMock := gomock.NewController(t)519	defer cfgMock.Finish()520	mockDB := NewMockDbAccessor(cfgMock)521	mockIaaS := NewMockIaaS(cfgMock)522	common.SetDataBase(mockDB)523	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)524	defer stubs.Reset()525	MockPaasAdminCheck(mockDB)526	netInfo := string(`{527    "network": {528        "cidr": "123.124.125.0/24",529        "gateway": "123.124.125.1",530        "name": "network_show",531        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"532    }}`)533	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(534		netInfo, nil)535	mockIaaS.EXPECT().AttachNetToRouter(536		gomock.Any(), gomock.Any()).Return("", errors.New("error"))537	resp := AttachRouter("uuid-for-routet", "uuid-for-network")538	Convey("TestAttachERR401\n", t, func() {539		So(resp.Code, ShouldEqual, 406)540	})541}542func TestAttachERR406C(t *testing.T) {543	cfgMock := gomock.NewController(t)544	defer cfgMock.Finish()545	mockDB := NewMockDbAccessor(cfgMock)546	mockIaaS := NewMockIaaS(cfgMock)547	common.SetDataBase(mockDB)548	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)549	defer stubs.Reset()550	MockPaasAdminCheck(mockDB)551	netInfo := string(`{552    "network": {553        "cidr": "123.124.125.0/24",554        "gateway": "123.124.125.1",555        "name": "network_show",556        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"557    }}`)558	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(559		netInfo, nil)560	mockIaaS.EXPECT().AttachNetToRouter(gomock.Any(),561		gomock.Any()).Return("uuid-for-port", nil)562	mockDB.EXPECT().SaveLeaf(gomock.Any(),563		gomock.Any()).Return(nil)564	mockDB.EXPECT().SaveLeaf(gomock.Any(),565		gomock.Any()).Return(errors.New("error"))566	resp := AttachRouter("uuid-for-routet", "uuid-for-network")567	Convey("TestAttachERR401\n", t, func() {568		So(resp.Code, ShouldEqual, 406)569	})570}571/*572func TestAttachERR406D(t *testing.T) {573	cfgMock := gomock.NewController(t)574	defer cfgMock.Finish()575	mockDB := NewMockDbAccessor(cfgMock)576	mockIaaS := NewMockIaaS(cfgMock)577	common.SetDataBase(mockDB)578	common.SetIaaS(mockIaaS)579	netInfo := string(`{580    "network": {581        "cidr": "123.124.125.0/24",582        "gateway": "123.124.125.1",583        "name": "network_show",584        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"585    }}`)586	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(587		netInfo, nil)588	mockIaaS.EXPECT().AttachNetToRouter(gomock.Any(),589		gomock.Any()).Return("uuid-for-port",nil)590	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(nil)591	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(nil)592	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(593		errors.New("error"))594	resp := AttachRouter("uuid-for-routet","uuid-for-network")595	Convey("TestAttachERR406C\n", t, func() {596		So(resp.Code, ShouldEqual, 406)597	})598}599func TestAttachERR406E(t *testing.T) {600	cfgMock := gomock.NewController(t)601	defer cfgMock.Finish()602	mockDB := NewMockDbAccessor(cfgMock)603	mockIaaS := NewMockIaaS(cfgMock)604	common.SetDataBase(mockDB)605	common.SetIaaS(mockIaaS)606	netInfo := string(`{607    "network": {608        "cidr": "123.124.125.0/24",609        "gateway": "123.124.125.1",610        "name": "network_show",611        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"612    }}`)613	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(614		netInfo, nil)615	mockIaaS.EXPECT().AttachNetToRouter(gomock.Any(),616		gomock.Any()).Return("uuid-for-port",nil)617	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(nil)618	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(nil)619	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(nil)620	mockDB.EXPECT().SaveLeaf(gomock.Any(),gomock.Any()).Return(nil)621	mockDB.EXPECT().SaveLeaf(gomock.Any(),622		gomock.Any()).Return(errors.New("error"))623	resp := AttachRouter("uuid-for-routet","uuid-for-network")624	Convey("TestAttachERR406C\n", t, func() {625		So(resp.Code, ShouldEqual, 406)626	})627}628*/629func TestAttachOK(t *testing.T) {630	cfgMock := gomock.NewController(t)631	defer cfgMock.Finish()632	mockDB := NewMockDbAccessor(cfgMock)633	mockIaaS := NewMockIaaS(cfgMock)634	common.SetDataBase(mockDB)635	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)636	defer stubs.Reset()637	MockPaasAdminCheck(mockDB)638	netInfo := string(`{639    "network": {640        "cidr": "123.124.125.0/24",641        "gateway": "123.124.125.1",642        "name": "network_show",643        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"644    }}`)645	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(646		netInfo, nil)647	mockIaaS.EXPECT().AttachNetToRouter(gomock.Any(),648		gomock.Any()).Return("uuid-for-port", nil)649	mockDB.EXPECT().SaveLeaf(gomock.Any(),650		gomock.Any()).Return(nil)651	mockDB.EXPECT().SaveLeaf(gomock.Any(),652		gomock.Any()).Return(nil)653	mockDB.EXPECT().SaveLeaf(gomock.Any(),654		gomock.Any()).Return(nil)655	mockDB.EXPECT().SaveLeaf(gomock.Any(),656		gomock.Any()).Return(nil)657	mockDB.EXPECT().SaveLeaf(gomock.Any(),658		gomock.Any()).Return(nil)659	mockDB.EXPECT().SaveLeaf(gomock.Any(),660		gomock.Any()).Return(nil)661	resp := AttachRouter("uuid-for-routet", "uuid-for-network")662	Convey("TestAttachERR406D\n", t, func() {663		So(resp.Code, ShouldEqual, 200)664	})665}666func TestDetachERR401(t *testing.T) {667	cfgMock := gomock.NewController(t)668	defer cfgMock.Finish()669	mockDB := NewMockDbAccessor(cfgMock)670	common.SetDataBase(mockDB)671	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)672	defer stubs.Reset()673	MockPaasAdminCheck(mockDB)674	resp := DetachRouter("uuid-for-routet", "uuid-for-network")675	Convey("TestDetachERR401\n", t, func() {676		So(resp.Code, ShouldEqual, 401)677	})678}679func TestDetachERR403(t *testing.T) {680	cfgMock := gomock.NewController(t)681	defer cfgMock.Finish()682	mockDB := NewMockDbAccessor(cfgMock)683	mockIaaS := NewMockIaaS(cfgMock)684	common.SetDataBase(mockDB)685	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)686	defer stubs.Reset()687	MockPaasAdminCheck(mockDB)688	resp := DetachRouter403("uuid-for-routet", "uuid-for-network")689	Convey("TestDetachERR403\n", t, func() {690		So(resp.Code, ShouldEqual, 403)691	})692}693func TestDetachERR406A(t *testing.T) {694	cfgMock := gomock.NewController(t)695	defer cfgMock.Finish()696	mockDB := NewMockDbAccessor(cfgMock)697	mockIaaS := NewMockIaaS(cfgMock)698	common.SetDataBase(mockDB)699	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)700	defer stubs.Reset()701	MockPaasAdminCheck(mockDB)702	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(703		"", errors.New("error"))704	resp := DetachRouter("uuid-for-routet", "uuid-for-network")705	Convey("TestAttachERR401\n", t, func() {706		So(resp.Code, ShouldEqual, 406)707	})708}709func TestDetachERR406B(t *testing.T) {710	cfgMock := gomock.NewController(t)711	defer cfgMock.Finish()712	mockDB := NewMockDbAccessor(cfgMock)713	mockIaaS := NewMockIaaS(cfgMock)714	common.SetDataBase(mockDB)715	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)716	defer stubs.Reset()717	MockPaasAdminCheck(mockDB)718	netInfo := string(`{719    "network": {720        "cidr": "123.124.125.0/24",721        "gateway": "123.124.125.1",722        "name": "network_show",723        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"724    }}`)725	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(726		netInfo, nil)727	mockIaaS.EXPECT().DetachNetFromRouter(gomock.Any(),728		gomock.Any()).Return("", errors.New("error"))729	resp := DetachRouter("uuid-for-routet", "uuid-for-network")730	Convey("TestDetachERR406B\n", t, func() {731		So(resp.Code, ShouldEqual, 406)732	})733}734func TestDetachERR406C(t *testing.T) {735	cfgMock := gomock.NewController(t)736	defer cfgMock.Finish()737	mockDB := NewMockDbAccessor(cfgMock)738	mockIaaS := NewMockIaaS(cfgMock)739	common.SetDataBase(mockDB)740	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)741	defer stubs.Reset()742	netInfo := string(`{743    "network": {744        "cidr": "123.124.125.0/24",745        "gateway": "123.124.125.1",746        "name": "network_show",747        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"748    }}`)749	MockPaasAdminCheck(mockDB)750	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(751		netInfo, nil)752	mockIaaS.EXPECT().DetachNetFromRouter(gomock.Any(),753		gomock.Any()).Return("uuid-for-port", nil)754	mockDB.EXPECT().DeleteLeaf(755		gomock.Any()).Return(errors.New("error"))756	resp := DetachRouter("uuid-for-routet", "uuid-for-network")757	Convey("TestDetachERR406C\n", t, func() {758		So(resp.Code, ShouldEqual, 406)759	})760}761func TestDetachERR406D(t *testing.T) {762	cfgMock := gomock.NewController(t)763	defer cfgMock.Finish()764	mockDB := NewMockDbAccessor(cfgMock)765	mockIaaS := NewMockIaaS(cfgMock)766	common.SetDataBase(mockDB)767	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)768	defer stubs.Reset()769	netInfo := string(`{770    "network": {771        "cidr": "123.124.125.0/24",772        "gateway": "123.124.125.1",773        "name": "network_show",774        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"775    }}`)776	MockPaasAdminCheck(mockDB)777	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(778		netInfo, nil)779	mockIaaS.EXPECT().DetachNetFromRouter(gomock.Any(),780		gomock.Any()).Return("uuid-for-port", nil)781	mockDB.EXPECT().DeleteLeaf(gomock.Any()).Return(nil)782	mockDB.EXPECT().DeleteLeaf(783		gomock.Any()).Return(errors.New("error"))784	resp := DetachRouter("uuid-for-routet", "uuid-for-network")785	Convey("TestDetachERR406D\n", t, func() {786		So(resp.Code, ShouldEqual, 406)787	})788}789func TestDetachERR406E(t *testing.T) {790	cfgMock := gomock.NewController(t)791	defer cfgMock.Finish()792	mockDB := NewMockDbAccessor(cfgMock)793	mockIaaS := NewMockIaaS(cfgMock)794	common.SetDataBase(mockDB)795	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)796	defer stubs.Reset()797	netInfo := string(`{798    "network": {799        "cidr": "123.124.125.0/24",800        "gateway": "123.124.125.1",801        "name": "network_show",802        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"803    }}`)804	MockPaasAdminCheck(mockDB)805	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(806		netInfo, nil)807	mockIaaS.EXPECT().DetachNetFromRouter(gomock.Any(),808		gomock.Any()).Return("uuid-for-port", nil)809	mockDB.EXPECT().DeleteLeaf(gomock.Any()).Return(nil)810	mockDB.EXPECT().DeleteLeaf(gomock.Any()).Return(nil)811	mockDB.EXPECT().DeleteDir(gomock.Any()).Return(errors.New("error"))812	resp := DetachRouter("uuid-for-routet", "uuid-for-network")813	Convey("TestDetachERR406D\n", t, func() {814		So(resp.Code, ShouldEqual, 406)815	})816}817func TestDetachOK(t *testing.T) {818	cfgMock := gomock.NewController(t)819	defer cfgMock.Finish()820	mockDB := NewMockDbAccessor(cfgMock)821	mockIaaS := NewMockIaaS(cfgMock)822	common.SetDataBase(mockDB)823	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)824	defer stubs.Reset()825	netInfo := string(`{826    "network": {827        "cidr": "123.124.125.0/24",828        "gateway": "123.124.125.1",829        "name": "network_show",830        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"831    }}`)832	MockPaasAdminCheck(mockDB)...

...23	"github.com/ZTE/Knitter/pkg/iaas-accessor"24	"github.com/golang/gostub"25)26func TestApiCreatPortERR401(t *testing.T) {27	//cfgMock := gomock.NewController(t)28	//defer cfgMock.Finish()29	//mockDB := NewMockDbAccessor(cfgMock)30	//common.SetDataBase(mockDB)31	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)32	defer stubs.Reset()33	resp := APICreatePort("the-uuid-of-port")34	Convey("TestApiCreatPortERR401\n", t, func() {35		So(resp.Code, ShouldEqual, 401)36	})37}38func TestApiCreatPortERR403A(t *testing.T) {39	cfgMock := gomock.NewController(t)40	defer cfgMock.Finish()41	//mockDB := NewMockDbAccessor(cfgMock)42	mockIaaS := NewMockIaaS(cfgMock)43	//common.SetDataBase(mockDB)44	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)45	defer stubs.Reset()46	resp := APICreatePort("844d0d23-2d53-454a-93cf-73c8253f94d6")47	Convey("TestApiCreatPortERR403A\n", t, func() {48		So(resp.Code, ShouldEqual, 403)49	})50}51func TestApiCreatPortERR406B(t *testing.T) {52	cfgMock := gomock.NewController(t)53	defer cfgMock.Finish()54	//mockDB := NewMockDbAccessor(cfgMock)55	mockIaaS := NewMockIaaS(cfgMock)56	//common.SetDataBase(mockDB)57	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)58	defer stubs.Reset()59	cfg := string(`{"network_name":"want-to-attach-network",60		"vnic_type":"normal"}`)61	resp := APICreatePort(cfg)62	Convey("TestApiCreatPortERR406B\n", t, func() {63		So(resp.Code, ShouldEqual, 406)64	})65}66func listAllNetwork(mockDB *MockDbAccessor) {67	var list []*client.Node68	node := client.Node{Key: "network-uuid", Value: "network-info"}69	list = append(list, &node)70	list = append(list, &node)71	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)72	netInfo0 := string(`{73    "network": {74        "cidr": "123.124.125.0/24",75        "gateway": "123.124.125.1",76        "name": "want-to-attach-network",77        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"78    }}`)79	netInfo1 := string(`{80    "network": {81        "cidr": "123.124.125.0/24",82        "gateway": "123.124.125.1",83        "name": "network_not_show",84        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"85    }}`)86	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)87	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)88	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)89	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)90	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)91}92func listAllNetwork2(mockDB *MockDbAccessor) {93	var list []*client.Node94	node := client.Node{Key: "network-uuid", Value: "network-info"}95	list = append(list, &node)96	list = append(list, &node)97	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)98	netInfo0 := string(`{99    "network": {100        "cidr": "123.124.125.0/24",101        "gateway": "123.124.125.1",102        "name": "want-to-attach-network",103        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"104    }}`)105	netInfo1 := string(`{106    "network": {107        "cidr": "123.124.125.0/24",108        "gateway": "123.124.125.1",109        "name": "network_not_show",110        "network_id": "9a4bf247-0cdd-4649-a5aa-4255856d3da2"111    }}`)112	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)113	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)114	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)115	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)116	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)117	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)118	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)119	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)120	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)121	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)122	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)123	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)124	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)125	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)126	mockDB.EXPECT().ReadDir(gomock.Any()).Return(list, nil)127	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo0, nil)128	mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(netInfo1, nil)129}130func TestApiDeletePortERR401(t *testing.T) {131	cfgMock := gomock.NewController(t)132	defer cfgMock.Finish()133	//mockDB := NewMockDbAccessor(cfgMock)134	//common.SetDataBase(mockDB)135	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)136	defer stubs.Reset()137	resp := APIDeletePort("the-name-of-network")138	Convey("TestApiDeletePortERR401\n", t, func() {139		So(resp.Code, ShouldEqual, 401)140	})141}142func TestApiAttachERR401(t *testing.T) {143	cfgMock := gomock.NewController(t)144	defer cfgMock.Finish()145	//mockDB := NewMockDbAccessor(cfgMock)146	//common.SetDataBase(mockDB)147	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)148	defer stubs.Reset()149	resp := APIAttach("port-uuid", "vm-uuid")150	Convey("TestApiDeletePortERR401\n", t, func() {151		So(resp.Code, ShouldEqual, 401)152	})153}154func TestApiAttachERR404A(t *testing.T) {155	cfgMock := gomock.NewController(t)156	defer cfgMock.Finish()157	mockDB := NewMockDbAccessor(cfgMock)158	mockIaaS := NewMockIaaS(cfgMock)159	common.SetDataBase(mockDB)160	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)161	defer stubs.Reset()162	mockIaaS.EXPECT().AttachPortToVM(163		gomock.Any(), gomock.Any()).Return(nil, errors.New("error"))164	mockDB.EXPECT().SaveLeaf(gomock.Any(), gomock.Any())165	resp := APIAttach("port-uuid", "vm-uuid")166	Convey("TestApiAttachERR404A\n", t, func() {167		So(resp.Code, ShouldEqual, 406)168	})169}170func TestApiAttachERR404B(t *testing.T) {171	cfgMock := gomock.NewController(t)172	defer cfgMock.Finish()173	mockDB := NewMockDbAccessor(cfgMock)174	mockIaaS := NewMockIaaS(cfgMock)175	common.SetDataBase(mockDB)176	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)177	defer stubs.Reset()178	mockIaaS.EXPECT().AttachPortToVM(179		gomock.Any(), gomock.Any()).Return(nil, errors.New("error"))180	mockDB.EXPECT().SaveLeaf(gomock.Any(), gomock.Any()).Return(nil)181	resp := APIAttach("port-uuid", "vm-uuid")182	Convey("TestApiAttachERR404A\n", t, func() {183		So(resp.Code, ShouldEqual, 406)184	})185}186func TestApiAttachERR404C(t *testing.T) {187	cfgMock := gomock.NewController(t)188	defer cfgMock.Finish()189	mockDB := NewMockDbAccessor(cfgMock)190	mockIaaS := NewMockIaaS(cfgMock)191	common.SetDataBase(mockDB)192	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)193	defer stubs.Reset()194	mockIaaS.EXPECT().AttachPortToVM(gomock.Any(), gomock.Any()).Return(nil, nil)195	mockIaaS.EXPECT().GetType().Return("TECS")196	mockIaaS.EXPECT().GetPort(gomock.Any()).Return(nil, errors.New("error"))197	mockIaaS.EXPECT().DetachPortFromVM(gomock.Any(), gomock.Any()).Return(nil)198	resp := APIAttach("port-uuid", "vm-uuid")199	Convey("TestApiAttachERR404A\n", t, func() {200		So(resp.Code, ShouldEqual, 406)201	})202}203func TestApiAttachERR404D(t *testing.T) {204	cfgMock := gomock.NewController(t)205	defer cfgMock.Finish()206	mockDB := NewMockDbAccessor(cfgMock)207	mockIaaS := NewMockIaaS(cfgMock)208	common.SetDataBase(mockDB)209	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)210	defer stubs.Reset()211	mockIaaS.EXPECT().AttachPortToVM(gomock.Any(), gomock.Any()).Return(nil, nil)212	mockIaaS.EXPECT().GetType().Return("TECS")213	mockIaaS.EXPECT().GetPort(gomock.Any()).Return(nil, errors.New("error"))214	mockIaaS.EXPECT().DetachPortFromVM(gomock.Any(), gomock.Any()).Return(errors.New("error"))215	mockDB.EXPECT().SaveLeaf(gomock.Any(), gomock.Any()).Return(nil)216	resp := APIAttach("port-uuid", "vm-uuid")217	Convey("TestApiAttachERR404A\n", t, func() {218		So(resp.Code, ShouldEqual, 406)219	})220}221func TestApiAttachOK(t *testing.T) {222	cfgMock := gomock.NewController(t)223	defer cfgMock.Finish()224	mockDB := NewMockDbAccessor(cfgMock)225	mockIaaS := NewMockIaaS(cfgMock)226	common.SetDataBase(mockDB)227	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)228	defer stubs.Reset()229	mockIaaS.EXPECT().AttachPortToVM(gomock.Any(), gomock.Any()).Return(nil, nil)230	mockIaaS.EXPECT().GetType().Return("TECS")231	port := iaasaccessor.Interface{Status: "ACTIVE"}232	mockIaaS.EXPECT().GetPort(gomock.Any()).Return(&port, nil)233	resp := APIAttach("port-uuid", "vm-uuid")234	Convey("TestApiAttachERR404A\n", t, func() {235		So(resp.Code, ShouldEqual, 200)236	})237}238func TestApiAttachVNFMOK(t *testing.T) {239	cfgMock := gomock.NewController(t)240	defer cfgMock.Finish()241	mockDB := NewMockDbAccessor(cfgMock)242	mockIaaS := NewMockIaaS(cfgMock)243	common.SetDataBase(mockDB)244	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)245	defer stubs.Reset()246	mockIaaS.EXPECT().AttachPortToVM(gomock.Any(), gomock.Any()).Return(nil, nil)247	mockIaaS.EXPECT().GetType().Return("VNFM")248	resp := APIAttach("port-uuid", "vm-uuid")249	Convey("TestApiAttachVNFMOK\n", t, func() {250		So(resp.Code, ShouldEqual, 200)251	})252}253func TestApiDetachERR401(t *testing.T) {254	cfgMock := gomock.NewController(t)255	defer cfgMock.Finish()256	//mockDB := NewMockDbAccessor(cfgMock)257	//common.SetDataBase(mockDB)258	stubs := gostub.StubFunc(&iaas.GetIaaS, nil)259	defer stubs.Reset()260	resp := APIDetach("port-uuid", "vm-uuid")261	Convey("TestApiDetachERR401\n", t, func() {262		So(resp.Code, ShouldEqual, 401)263	})264}265func TestApiDetachERR404A(t *testing.T) {266	cfgMock := gomock.NewController(t)267	defer cfgMock.Finish()268	mockDB := NewMockDbAccessor(cfgMock)269	mockIaaS := NewMockIaaS(cfgMock)270	common.SetDataBase(mockDB)271	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)272	defer stubs.Reset()273	mockIaaS.EXPECT().DetachPortFromVM(274		gomock.Any(), gomock.Any()).Return(errors.New("error")).Times(3)275	mockDB.EXPECT().SaveLeaf(gomock.Any(), gomock.Any()).Return(nil)276	resp := APIDetach("port-uuid", "vm-uuid")277	Convey("TestApiAttachERR404A\n", t, func() {278		So(resp.Code, ShouldEqual, 406)279	})280}281func TestApiDetachERR404B(t *testing.T) {282	cfgMock := gomock.NewController(t)283	defer cfgMock.Finish()284	mockDB := NewMockDbAccessor(cfgMock)285	mockIaaS := NewMockIaaS(cfgMock)286	common.SetDataBase(mockDB)287	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)288	defer stubs.Reset()289	mockIaaS.EXPECT().DetachPortFromVM(gomock.Any(), gomock.Any()).Return(nil)290	mockIaaS.EXPECT().GetType().Return("TECS")291	mockIaaS.EXPECT().GetPort(gomock.Any()).Return(nil, errors.New("error"))292	mockDB.EXPECT().SaveLeaf(gomock.Any(), gomock.Any()).Return(nil)293	resp := APIDetach("port-uuid", "vm-uuid")294	Convey("TestApiAttachERR404A\n", t, func() {295		So(resp.Code, ShouldEqual, 406)296	})297}298func TestApiDetachOK(t *testing.T) {299	cfgMock := gomock.NewController(t)300	defer cfgMock.Finish()301	mockDB := NewMockDbAccessor(cfgMock)302	mockIaaS := NewMockIaaS(cfgMock)303	common.SetDataBase(mockDB)304	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)305	defer stubs.Reset()306	mockIaaS.EXPECT().DetachPortFromVM(gomock.Any(), gomock.Any()).Return(nil)307	mockIaaS.EXPECT().GetType().Return("TECS")308	port2 := iaasaccessor.Interface{Status: "DOWN"}309	mockIaaS.EXPECT().GetPort(gomock.Any()).Return(&port2, nil)310	resp := APIDetach("port-uuid", "vm-uuid")311	Convey("TestApiAttachERR404A\n", t, func() {312		So(resp.Code, ShouldEqual, 200)313	})314}315func TestApiDetachVNFMOK(t *testing.T) {316	cfgMock := gomock.NewController(t)317	defer cfgMock.Finish()318	mockDB := NewMockDbAccessor(cfgMock)319	mockIaaS := NewMockIaaS(cfgMock)320	common.SetDataBase(mockDB)321	stubs := gostub.StubFunc(&iaas.GetIaaS, mockIaaS)322	defer stubs.Reset()323	mockIaaS.EXPECT().DetachPortFromVM(gomock.Any(), gomock.Any()).Return(nil)324	mockIaaS.EXPECT().GetType().Return("VNFM")325	resp := APIDetach("port-uuid", "vm-uuid")326	Convey("TestApiAttachERR404A\n", t, func() {327		So(resp.Code, ShouldEqual, 200)328	})329}...

...21	"testing"22)23func TestCase1GetPaasUUID(t *testing.T) {24	var wantUUID string = "UUID-PAAS-KNITTER"25	mockCtl := gomock.NewController(t)26	defer mockCtl.Finish()27	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)28	SetDataBase(mockDB)29	gomock.InOrder(30		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfPaasUUID()).Return(wantUUID, nil),31	)32	convey.Convey("TestCase1GetPaasUUID\n", t, func() {33		id := GetPaasUUID()34		convey.So(id, convey.ShouldEqual, wantUUID)35	})36}37func TestCase2GetPaasUUID(t *testing.T) {38	mockCtl := gomock.NewController(t)39	defer mockCtl.Finish()40	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)41	SetDataBase(mockDB)42	gomock.InOrder(43		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfPaasUUID()).Return("",44			errors.New("unknow errors")),45	)46	convey.Convey("TestCase2GetPaasUUID\n", t, func() {47		id := GetPaasUUID()48		convey.So(id, convey.ShouldEqual, uuid.NIL.String())49	})50}51func TestCase3GetPaasUUID(t *testing.T) {52	mockCtl := gomock.NewController(t)53	defer mockCtl.Finish()54	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)55	SetDataBase(mockDB)56	gomock.InOrder(57		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfPaasUUID()).Return("",58			errors.New(ErrorKeyNotFound)),59		mockDB.EXPECT().SaveLeaf(dbaccessor.GetKeyOfPaasUUID(),60			gomock.Any()).Return(nil),61	)62	convey.Convey("TestCase3GetPaasUUID\n", t, func() {63		id := GetPaasUUID()64		convey.So(id, convey.ShouldNotEqual, uuid.NIL.String())65	})66}67func TestCase4GetPaasUUID(t *testing.T) {68	mockCtl := gomock.NewController(t)69	defer mockCtl.Finish()70	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)71	SetDataBase(mockDB)72	gomock.InOrder(73		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfPaasUUID()).Return("",74			errors.New(ErrorKeyNotFound)),75		mockDB.EXPECT().SaveLeaf(dbaccessor.GetKeyOfPaasUUID(),76			gomock.Any()).Return(errors.New("unknow error happen")),77	)78	convey.Convey("TestCase4GetPaasUUID\n", t, func() {79		id := GetPaasUUID()80		convey.So(id, convey.ShouldEqual, uuid.NIL.String())81	})82}83func TestCase1GetOpenstackCfg(t *testing.T) {84	mockCtl := gomock.NewController(t)85	defer mockCtl.Finish()86	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)87	SetDataBase(mockDB)88	gomock.InOrder(89		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfOpenstack()).Return("",90			errors.New(ErrorKeyNotFound)),91	)92	convey.Convey("TestCase1GetOpenstackCfg\n", t, func() {93		cfg := GetOpenstackCfg()94		convey.So(cfg, convey.ShouldEqual, "")95	})96}97func TestCase2GetOpenstackCfg(t *testing.T) {98	const OpenstackConfig string = "right openstack configuration"99	mockCtl := gomock.NewController(t)100	defer mockCtl.Finish()101	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)102	SetDataBase(mockDB)103	gomock.InOrder(104		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfOpenstack()).Return(105			OpenstackConfig, nil),106	)107	convey.Convey("TestCase2GetOpenstackCfg\n", t, func() {108		cfg := GetOpenstackCfg()109		convey.So(cfg, convey.ShouldEqual, OpenstackConfig)110	})111}112func TestCase1GetVnfmCfg(t *testing.T) {113	mockCtl := gomock.NewController(t)114	defer mockCtl.Finish()115	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)116	SetDataBase(mockDB)117	gomock.InOrder(118		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfVnfm()).Return(119			"", errors.New(ErrorKeyNotFound)),120	)121	convey.Convey("TestCase1GetVnfmCfg\n", t, func() {122		cfg := GetVnfmCfg()123		convey.So(cfg, convey.ShouldEqual, "")124	})125}126func TestCase2GetVnfmCfg(t *testing.T) {127	const OpenstackConfig string = "right openstack configuration"128	mockCtl := gomock.NewController(t)129	defer mockCtl.Finish()130	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)131	SetDataBase(mockDB)132	gomock.InOrder(133		mockDB.EXPECT().ReadLeaf(dbaccessor.GetKeyOfVnfm()).Return(134			OpenstackConfig, nil),135	)136	convey.Convey("TestCase2GetVnfmCfg\n", t, func() {137		cfg := GetVnfmCfg()138		convey.So(cfg, convey.ShouldEqual, OpenstackConfig)139	})140}141func TestCase1RegisterSelfToDb(t *testing.T) {142	const ManagerURL string = "https://paas.zte.com.cn:9527"143	mockCtl := gomock.NewController(t)144	defer mockCtl.Finish()145	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)146	SetDataBase(mockDB)147	gomock.InOrder(148		mockDB.EXPECT().SaveLeaf(dbaccessor.GetKeyOfKnitterManagerUrl(),149			ManagerURL).Return(nil),150	)151	convey.Convey("TestCase1RegisterSelfToDb\n", t, func() {152		e := RegisterSelfToDb(ManagerURL)153		convey.So(e, convey.ShouldEqual, nil)154	})155}156func TestCase2RegisterSelfToDb(t *testing.T) {157	const ManagerURL string = "https://paas.zte.com.cn:9527"158	mockCtl := gomock.NewController(t)159	defer mockCtl.Finish()160	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)161	SetDataBase(mockDB)162	gomock.InOrder(163		mockDB.EXPECT().SaveLeaf(dbaccessor.GetKeyOfKnitterManagerUrl(),164			ManagerURL).Return(errors.New(ErrorKeyNotFound)),165	)166	convey.Convey("TestCase1RegisterSelfToDb\n", t, func() {167		e := RegisterSelfToDb(ManagerURL)168		convey.So(e, convey.ShouldNotEqual, nil)169	})170}171func TestCase1SetDataBase(t *testing.T) {172	convey.Convey("TestCase1SetDataBase\n", t, func() {173		e := SetDataBase(nil)174		convey.So(e, convey.ShouldEqual, nil)175	})176}177func TestCase1GetDataBase(t *testing.T) {178	convey.Convey("TestCase1GetDataBase\n", t, func() {179		e := SetDataBase(nil)180		convey.So(e, convey.ShouldEqual, nil)181		p := GetDataBase()182		convey.So(p, convey.ShouldEqual, nil)183	})184}185//todo fix ut error186/*func TestCase1CheckDB(t *testing.T) {187	mockCtl := gomock.NewController(t)188	defer mockCtl.Finish()189	mockDB := mockdbaccessor.NewMockDbAccessor(mockCtl)190	SetDataBase(mockDB)191	gomock.InOrder(192		//case 1193		mockDB.EXPECT().SaveLeaf(gomock.Any(),194			gomock.Any()).Return(errors.New(ErrorKeyNotFound)),195		//case 2196		mockDB.EXPECT().SaveLeaf(gomock.Any(),197			gomock.Any()).Return(nil),198		mockDB.EXPECT().ReadLeaf(gomock.Any()).Return(199			"", errors.New(ErrorKeyNotFound)),200		//case 3201		mockDB.EXPECT().SaveLeaf(gomock.Any(),...

1mockCtrl := gomock.NewController(t)2defer mockCtrl.Finish()3mockCtrl := gomock.NewController(t)4defer mockCtrl.Finish()5mockCtrl := gomock.NewController(t)6defer mockCtrl.Finish()7--- FAIL: TestNewController (0.00s)8panic: gomock: out of order call: *mocks.MockController.Finish() [recovered]9    panic: gomock: out of order call: *mocks.MockController.Finish()10testing.tRunner.func1(0xc4200d40f0)11panic(0x11c9e60, 0xc42000a4e0)12github.com/golang/mock/gomock.(*Controller).Finish(0xc4200b2000)13github.com/golang/mock/gomock.NewController(0x0, 0x0, 0x0, 0x0)14github.com/golang/mock/gomock.NewController-fm(0x0, 0x0, 0x0, 0x0)15github.com/golang/mock/gomock.(*Controller).Finish(0xc4200b2000)16github.com/golang/mock/gomock.NewController(0x

1func NewController(t TestingT) *gomock.Controller {2	return gomock.NewController(t)3}4func NewController(t TestingT) *gomock.Controller {5	return gomock.NewController(t)6}7So, I have to import the 1.go file in my test file. I have to import the 1.go file in my test file. I have to import the 1.go file in my test file. I have t

1import (2func TestController(t *testing.T) {3    ctrl := gomock.NewController(t)4    defer ctrl.Finish()5    fmt.Println("Test Controller")6}7import (8func TestController(t *testing.T) {9    ctrl := gomock.NewController(t)10    defer ctrl.Finish()11    fmt.Println("Test Controller")12}

1import (2func TestNewController(t *testing.T) {3	ctrl := gomock.NewController(t)4	defer ctrl.Finish()5	fmt.Println(ctrl)6	assert.NotNil(t, ctrl)7}8import (9func TestNewController(t *testing.T) {10	ctrl := gomock.NewController(t)11	defer ctrl.Finish()12	fmt.Println(ctrl)13	assert.NotNil(t, ctrl)14}15import (16func TestNewController(t *testing.T) {17	ctrl := gomock.NewController(t)18	defer ctrl.Finish()19	fmt.Println(ctrl)20	assert.NotNil(t, ctrl)21}22import (23func TestNewController(t *testing.T) {24	ctrl := gomock.NewController(t)25	defer ctrl.Finish()26	fmt.Println(ctrl)27	assert.NotNil(t, ctrl)28}29import (30func TestNewController(t *testing.T) {31	ctrl := gomock.NewController(t)32	defer ctrl.Finish()33	fmt.Println(ctrl)34	assert.NotNil(t, ctrl)35}36import (37func TestNewController(t *testing.T) {38	ctrl := gomock.NewController(t)39	defer ctrl.Finish()40	fmt.Println(ctrl)41	assert.NotNil(t, ctrl)42}

1func TestController(t *testing.T) {2	ctrl := gomock.NewController(t)3	defer ctrl.Finish()4}5func TestController(t *testing.T) {6	ctrl := gomock.NewController(t)7	defer ctrl.Finish()8}9func TestController(t *testing.T) {10	ctrl := gomock.NewController(t)11	defer ctrl.Finish()12}13func TestController(t *testing.T) {14	ctrl := gomock.NewController(t)15	defer ctrl.Finish()16}17func TestController(t *testing.T) {18	ctrl := gomock.NewController(t)19	defer ctrl.Finish()20}21func TestController(t *testing.T) {22	ctrl := gomock.NewController(t)23	defer ctrl.Finish()24}25func TestController(t *testing.T) {26	ctrl := gomock.NewController(t)27	defer ctrl.Finish()28}29func TestController(t *testing.T) {30	ctrl := gomock.NewController(t)31	defer ctrl.Finish()32}33func TestController(t *testing.T) {34	ctrl := gomock.NewController(t)35	defer ctrl.Finish()36}

1func TestNewController(t *testing.T) {2    mock := NewMockMyInterface(ctrl)3    c := NewController(t)4}5func TestNewController(t *testing.T) {6    mock := NewMockMyInterface(ctrl)7    c := NewController(t)8}9func TestNewController(t *testing.T) {10    mock := NewMockMyInterface(ctrl)11    c := NewController(t)12}13func TestNewController(t *testing.T) {14    mock := NewMockMyInterface(ctrl)15    c := NewController(t)16}17func TestNewController(t *testing.T) {18    mock := NewMockMyInterface(ctrl)19    c := NewController(t)20}21func TestNewController(t *testing.T) {22    mock := NewMockMyInterface(ctrl)23    c := NewController(t)24}25func TestNewController(t *testing.T) {26    mock := NewMockMyInterface(ctrl)27    c := NewController(t)28}29func TestNewController(t *testing.T) {30    mock := NewMockMyInterface(ctrl)31    c := NewController(t