github.com/kyma-incubator/compass/components/director@v0.0.0-20230623144113-d764f56ff805/internal/domain/operation/fixtures_test.go (about)

     1  package operation_test
     2  
     3  import (
     4  	"database/sql/driver"
     5  	"encoding/json"
     6  
     7  	"github.com/kyma-incubator/compass/components/director/internal/domain/operation"
     8  
     9  	"time"
    10  
    11  	"github.com/kyma-incubator/compass/components/director/internal/model"
    12  	"github.com/kyma-incubator/compass/components/director/internal/repo"
    13  )
    14  
    15  const (
    16  	ordOpType   = "ORD_AGGREGATION"
    17  	operationID = "aaaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
    18  )
    19  
    20  func fixOperationInput(opType string, opStatus model.OperationStatus) *model.OperationInput {
    21  	return &model.OperationInput{
    22  		OpType:     opType,
    23  		Status:     opStatus,
    24  		Data:       json.RawMessage("[]"),
    25  		Error:      json.RawMessage("[]"),
    26  		Priority:   1,
    27  		CreatedAt:  &time.Time{},
    28  		FinishedAt: &time.Time{},
    29  	}
    30  }
    31  
    32  func fixOperationModel(opType string, status model.OperationStatus) *model.Operation {
    33  	return fixOperationModelWithID(operationID, opType, status)
    34  }
    35  
    36  func fixOperationModelWithID(id, opType string, opStatus model.OperationStatus) *model.Operation {
    37  	return &model.Operation{
    38  		ID:         id,
    39  		OpType:     opType,
    40  		Status:     opStatus,
    41  		Data:       json.RawMessage("[]"),
    42  		Error:      json.RawMessage("[]"),
    43  		Priority:   1,
    44  		CreatedAt:  &time.Time{},
    45  		FinishedAt: &time.Time{},
    46  	}
    47  }
    48  
    49  func fixEntityOperation(id, opType string, opStatus model.OperationStatus) *operation.Entity {
    50  	return &operation.Entity{
    51  		ID:         id,
    52  		Type:       opType,
    53  		Status:     string(opStatus),
    54  		Data:       repo.NewValidNullableString("[]"),
    55  		Error:      repo.NewValidNullableString("[]"),
    56  		Priority:   1,
    57  		CreatedAt:  &time.Time{},
    58  		FinishedAt: &time.Time{},
    59  	}
    60  }
    61  
    62  func fixOperationCreateArgs(op *model.Operation) []driver.Value {
    63  	return []driver.Value{op.ID, op.OpType, op.Status, repo.NewNullableStringFromJSONRawMessage(op.Data), repo.NewNullableStringFromJSONRawMessage(op.Error), op.Priority, op.CreatedAt, op.FinishedAt}
    64  }