Merge pull request gusaul#36 from beruangcoklat/master

handle error when remove and create directory

Author: gusaul

core/resource.go

@@ -285,8 +285,14 @@ func (r *Resource) AddProtos(protos []Proto) error {
 	protoPath := filepath.Join(BasePath, r.clientConn.Target())
 	err := os.MkdirAll(protoPath, 0777)
 	if os.IsExist(err) {
-		os.RemoveAll(protoPath)
+		err = os.RemoveAll(protoPath)
+		if err != nil {
+			return err
+		}
 		err = os.MkdirAll(protoPath, 0777)
+		if err != nil {
+			return err
+		}
 	} else if err != nil {
 		return err
 	}

core/resource_test.go

@@ -60,7 +60,7 @@ func Test_prepareImport(t *testing.T) {
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			if got := prepareImport(tt.args.proto, ""); !reflect.DeepEqual(got, tt.want) {
+			if got := prepareImport(tt.args.proto); !reflect.DeepEqual(got, tt.want) {
 				t.Errorf("prepareImport() = %v, want %v",
 					string(got),
 					string(tt.want))