Merge 610f974 into e9f03c3

Author: khanhx

persistent/memory/memory.go

@@ -174,6 +174,70 @@ func (m *Memory) PatchRoute(ctx context.Context, mockID string, routeID string,
 	return nil
 }
 
+func (m *Memory) DeleteRoute(ctx context.Context, mockID string, routeID string) error {
+	mok, err := m.GetMock(ctx, mockID)
+	if err != nil {
+		return err
+	}
+
+	if mok == nil {
+		return errors.New("mock not found")
+	}
+
+	_, idx, ok := lo.FindIndexOf[*mock.Route](mok.Routes, func(route *mock.Route) bool {
+		return route.ID == routeID
+	})
+
+	if !ok {
+		return errors.New("route not found")
+	}
+
+	mok.Routes = append(mok.Routes[:idx], mok.Routes[idx+1:]...)
+
+	if err := m.SetMock(ctx, mok); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (m *Memory) CreateRoute(ctx context.Context, mockID string, data string) error {
+	mok, err := m.GetMock(ctx, mockID)
+	if err != nil {
+		return err
+	}
+
+	if mok == nil {
+		return errors.New("mock not found")
+	}
+
+	var values map[string]*json.RawMessage
+	if err := json.Unmarshal([]byte(data), &values); err != nil {
+		return err
+	}
+
+	var newRoute = &mock.Route{}
+	if err := patchStruct(newRoute, values); err != nil {
+		return err
+	}
+
+	_, _, ok := lo.FindIndexOf[*mock.Route](mok.Routes, func(route *mock.Route) bool {
+		return route.ID == newRoute.ID
+	})
+
+	if ok {
+		return errors.New("route already created")
+	}
+
+	mok.Routes = append(mok.Routes, newRoute)
+
+	if err := m.SetMock(ctx, mok); err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func (m *Memory) PatchResponse(ctx context.Context, mockID, routeID, responseID, data string) error {
 	mok, err := m.GetMock(ctx, mockID)
 	if err != nil {

persistent/memory/memory_test.go

@@ -122,6 +122,81 @@ func TestMemory_PatchRoute(t *testing.T) {
 	})
 }
 
+func TestMemory_DeleteRoute(t *testing.T) {
+	m := New()
+	mok := &mock.Mock{
+		ID: "mockid",
+		Routes: []*mock.Route{
+			{
+				ID:     "routeid",
+				Method: "GET",
+			},
+			{
+				ID:     "routeid1",
+				Method: "PUT",
+			},
+		},
+	}
+	_ = m.SetMock(context.Background(), mok)
+
+	t.Run("success", func(t *testing.T) {
+		err := m.DeleteRoute(context.Background(), "mockid", "routeid")
+		require.NoError(t, err)
+
+		configs, err := m.GetMock(context.Background(), "mockid")
+		require.NoError(t, err)
+		assert.Equal(t, 1, len(configs.Routes))
+	})
+
+	t.Run("mock not found", func(t *testing.T) {
+		err := m.DeleteRoute(context.Background(), "random", "")
+		require.Error(t, err)
+	})
+
+	t.Run("route not found", func(t *testing.T) {
+		err := m.DeleteRoute(context.Background(), "mockid", "random")
+		require.Error(t, err)
+	})
+}
+
+func TestMemory_CreateRoute(t *testing.T) {
+	m := New()
+	mok := &mock.Mock{
+		ID: "mockid",
+		Routes: []*mock.Route{
+			{
+				ID:     "routeid",
+				Method: "GET",
+			},
+		},
+	}
+	_ = m.SetMock(context.Background(), mok)
+
+	t.Run("success", func(t *testing.T) {
+		err := m.CreateRoute(context.Background(), "mockid", `{"id":"routeid1","method":"PUT"}`)
+		require.NoError(t, err)
+		configs, err := m.GetMock(context.Background(), "mockid")
+		require.NoError(t, err)
+		assert.Equal(t, 2, len(configs.Routes))
+	})
+
+	t.Run("mock not found", func(t *testing.T) {
+		err := m.CreateRoute(context.Background(), "random", `{}`)
+		require.Error(t, err)
+	})
+
+	t.Run("route already created", func(t *testing.T) {
+		err := m.CreateRoute(context.Background(), "mockid", `{"id":"routeid1","method":"PUT"}`)
+		require.Error(t, err)
+	})
+
+	t.Run("invalid json", func(t *testing.T) {
+		err := m.CreateRoute(context.Background(), "mockid", `{"method": "}`)
+		require.Error(t, err)
+	})
+
+}
+
 func TestMemory_PatchResponse(t *testing.T) {
 	m := New()
 	mok := &mock.Mock{