Added UT for creating a template with same name after deleting one

Signed-off-by: parauliya <[email protected]>

db/mock/template.go

@@ -9,8 +9,9 @@ import (
 )
 
 type Template struct {
-	ID   uuid.UUID
-	Data string
+	ID      uuid.UUID
+	Data    string
+	Deleted bool
 }
 
 // CreateTemplate creates a new workflow template
@@ -20,12 +21,20 @@ func (d DB) CreateTemplate(ctx context.Context, name string, data string, id uui
 	}
 
 	if _, ok := d.TemplateDB[name]; ok {
-		return errors.New("Template name already exist in the database")
+		tmpl := d.TemplateDB[name]
+		switch stmpl := tmpl.(type) {
+		case Template:
+			if !stmpl.Deleted {
+				return errors.New("Template name already exist in the database")
+			}
+		default:
+			return errors.New("Not a Template type in the database")
+		}
 	}
-
 	d.TemplateDB[name] = Template{
-		ID:   id,
-		Data: data,
+		ID:      id,
+		Data:    data,
+		Deleted: false,
 	}
 
 	return nil

grpc-server/template_test.go

@@ -83,7 +83,10 @@ func TestCreateTemplate(t *testing.T) {
 			args: args{
 				db: mock.DB{
 					TemplateDB: map[string]interface{}{
-						"template_1": template1,
+						"template_1": mock.Template{
+							Data:    template1,
+							Deleted: false,
+						},
 					},
 				},
 				name:     "template_2",
@@ -98,7 +101,10 @@ func TestCreateTemplate(t *testing.T) {
 			args: args{
 				db: mock.DB{
 					TemplateDB: map[string]interface{}{
-						"template_1": template1,
+						"template_1": mock.Template{
+							Data:    template1,
+							Deleted: false,
+						},
 					},
 				},
 				name:     "template_1",
@@ -108,6 +114,24 @@ func TestCreateTemplate(t *testing.T) {
 				expectedError: true,
 			},
 		},
+
+		"SuccessfulTemplateCreationAfterDeletingWithSameName": {
+			args: args{
+				db: mock.DB{
+					TemplateDB: map[string]interface{}{
+						"template_1": mock.Template{
+							Data:    template1,
+							Deleted: true,
+						},
+					},
+				},
+				name:     "template_1",
+				template: template2,
+			},
+			want: want{
+				expectedError: false,
+			},
+		},
 	}
 
 	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
@@ -124,6 +148,7 @@ func TestCreateTemplate(t *testing.T) {
 				assert.Nil(t, err)
 				assert.NotEmpty(t, res)
 			}
+			tc.args.db.ClearTemplateDB()
 		})
 	}
 }