mirror of
https://github.com/onepanelio/onepanel.git
synced 2025-10-04 21:32:39 +08:00
fix: unit tests
This commit is contained in:
@@ -1,6 +1,7 @@
|
|||||||
package request
|
package request
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"github.com/Masterminds/squirrel"
|
||||||
"github.com/onepanelio/core/pkg/util/request/pagination"
|
"github.com/onepanelio/core/pkg/util/request/pagination"
|
||||||
"github.com/onepanelio/core/pkg/util/request/sort"
|
"github.com/onepanelio/core/pkg/util/request/sort"
|
||||||
)
|
)
|
||||||
@@ -14,5 +15,22 @@ type Request struct {
|
|||||||
|
|
||||||
// HasSorting returns true if there are any sorting criteria in the request
|
// HasSorting returns true if there are any sorting criteria in the request
|
||||||
func (r *Request) HasSorting() bool {
|
func (r *Request) HasSorting() bool {
|
||||||
return r.Sort != nil && len(r.Sort.Properties) > 0
|
return r != nil &&
|
||||||
|
r.Sort != nil &&
|
||||||
|
len(r.Sort.Properties) > 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// HasFilter returns true if there is any filtering criteria in the request
|
||||||
|
func (r *Request) HasFilter() bool {
|
||||||
|
return r != nil &&
|
||||||
|
r.Filter != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ApplyPaginationToSelect applies the pagination to the selectBuilder, if there is a pagination.
|
||||||
|
func (r *Request) ApplyPaginationToSelect(sb *squirrel.SelectBuilder) *squirrel.SelectBuilder {
|
||||||
|
if r == nil || r.Pagination == nil {
|
||||||
|
return sb
|
||||||
|
}
|
||||||
|
|
||||||
|
return r.Pagination.ApplyToSelect(sb)
|
||||||
}
|
}
|
||||||
|
@@ -75,7 +75,7 @@ func (wf *WorkflowExecutionFilter) GetLabels() []*Label {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func applyWorkflowExecutionFilter(sb sq.SelectBuilder, request *request.Request) (sq.SelectBuilder, error) {
|
func applyWorkflowExecutionFilter(sb sq.SelectBuilder, request *request.Request) (sq.SelectBuilder, error) {
|
||||||
if request.Filter == nil {
|
if !request.HasFilter() {
|
||||||
return sb, nil
|
return sb, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -865,8 +865,7 @@ func (c *Client) ListWorkflowExecutions(namespace, workflowTemplateUID, workflow
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
sb = *request.Pagination.ApplyToSelect(&sb)
|
sb = *request.ApplyPaginationToSelect(&sb)
|
||||||
|
|
||||||
if err := c.DB.Selectx(&workflows, sb); err != nil {
|
if err := c.DB.Selectx(&workflows, sb); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@@ -171,7 +171,7 @@ func (c *Client) createWorkflowTemplate(namespace string, workflowTemplate *Work
|
|||||||
workflowTemplateVersion := &WorkflowTemplateVersion{
|
workflowTemplateVersion := &WorkflowTemplateVersion{
|
||||||
WorkflowTemplate: workflowTemplate,
|
WorkflowTemplate: workflowTemplate,
|
||||||
Manifest: workflowTemplate.Manifest,
|
Manifest: workflowTemplate.Manifest,
|
||||||
Labels: workflowTemplate.Labels,
|
Labels: workflowTemplate.Labels,
|
||||||
}
|
}
|
||||||
err = createWorkflowTemplateVersionDB(tx, workflowTemplateVersion, params)
|
err = createWorkflowTemplateVersionDB(tx, workflowTemplateVersion, params)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -422,7 +422,7 @@ func (c *Client) selectWorkflowTemplatesQuery(namespace string, request *request
|
|||||||
OrderBy("wt.created_at DESC")
|
OrderBy("wt.created_at DESC")
|
||||||
|
|
||||||
sb = applyLabelSelectQuery(sb, request)
|
sb = applyLabelSelectQuery(sb, request)
|
||||||
sb = *request.Pagination.ApplyToSelect(&sb)
|
sb = *request.ApplyPaginationToSelect(&sb)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
@@ -32,7 +32,7 @@ func (wf *WorkspaceFilter) GetLabels() []*Label {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func applyWorkspaceFilter(sb sq.SelectBuilder, request *request.Request) (sq.SelectBuilder, error) {
|
func applyWorkspaceFilter(sb sq.SelectBuilder, request *request.Request) (sq.SelectBuilder, error) {
|
||||||
if request.Filter == nil {
|
if !request.HasFilter() {
|
||||||
return sb, nil
|
return sb, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -711,7 +711,7 @@ func (c *Client) ListWorkspaces(namespace string, request *request.Request) (wor
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
sb = *request.Pagination.ApplyToSelect(&sb)
|
sb = *request.ApplyPaginationToSelect(&sb)
|
||||||
|
|
||||||
if err := c.DB.Selectx(&workspaces, sb); err != nil {
|
if err := c.DB.Selectx(&workspaces, sb); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@@ -89,7 +89,7 @@ func (wt *WorkspaceTemplateFilter) GetLabels() []*Label {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func applyWorkspaceTemplateFilter(sb sq.SelectBuilder, request *request.Request) (sq.SelectBuilder, error) {
|
func applyWorkspaceTemplateFilter(sb sq.SelectBuilder, request *request.Request) (sq.SelectBuilder, error) {
|
||||||
if request.Filter == nil {
|
if !request.HasFilter() {
|
||||||
return sb, nil
|
return sb, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1185,7 +1185,7 @@ func (c *Client) ListWorkspaceTemplates(namespace string, request *request.Reque
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
sb = *request.Pagination.ApplyToSelect(&sb)
|
sb = *request.ApplyPaginationToSelect(&sb)
|
||||||
|
|
||||||
err = c.DB.Selectx(&workspaceTemplates, sb)
|
err = c.DB.Selectx(&workspaceTemplates, sb)
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user