...

Source file src/sigs.k8s.io/cli-utils/pkg/print/list/base_test.go

Documentation: sigs.k8s.io/cli-utils/pkg/print/list

     1  // Copyright 2022 The Kubernetes Authors.
     2  // SPDX-License-Identifier: Apache-2.0
     3  
     4  package list
     5  
     6  import (
     7  	"testing"
     8  
     9  	"sigs.k8s.io/cli-utils/pkg/apply/event"
    10  	"sigs.k8s.io/cli-utils/pkg/common"
    11  	"sigs.k8s.io/cli-utils/pkg/print/stats"
    12  	"sigs.k8s.io/cli-utils/pkg/printers/printer"
    13  	printertesting "sigs.k8s.io/cli-utils/pkg/printers/testutil"
    14  )
    15  
    16  func TestPrint(t *testing.T) {
    17  	printertesting.PrintResultErrorTest(t, func() printer.Printer {
    18  		return &BaseListPrinter{
    19  			FormatterFactory: func(previewStrategy common.DryRunStrategy) Formatter {
    20  				return newCountingFormatter()
    21  			},
    22  		}
    23  	})
    24  }
    25  
    26  func newCountingFormatter() *countingFormatter {
    27  	return &countingFormatter{}
    28  }
    29  
    30  type countingFormatter struct {
    31  	validationEvent  []event.ValidationEvent
    32  	applyEvents      []event.ApplyEvent
    33  	statusEvents     []event.StatusEvent
    34  	pruneEvents      []event.PruneEvent
    35  	deleteEvents     []event.DeleteEvent
    36  	waitEvents       []event.WaitEvent
    37  	errorEvent       event.ErrorEvent
    38  	actionGroupEvent []event.ActionGroupEvent
    39  }
    40  
    41  func (c *countingFormatter) FormatValidationEvent(e event.ValidationEvent) error {
    42  	c.validationEvent = append(c.validationEvent, e)
    43  	return nil
    44  }
    45  
    46  func (c *countingFormatter) FormatApplyEvent(e event.ApplyEvent) error {
    47  	c.applyEvents = append(c.applyEvents, e)
    48  	return nil
    49  }
    50  
    51  func (c *countingFormatter) FormatStatusEvent(e event.StatusEvent) error {
    52  	c.statusEvents = append(c.statusEvents, e)
    53  	return nil
    54  }
    55  
    56  func (c *countingFormatter) FormatPruneEvent(e event.PruneEvent) error {
    57  	c.pruneEvents = append(c.pruneEvents, e)
    58  	return nil
    59  }
    60  
    61  func (c *countingFormatter) FormatDeleteEvent(e event.DeleteEvent) error {
    62  	c.deleteEvents = append(c.deleteEvents, e)
    63  	return nil
    64  }
    65  
    66  func (c *countingFormatter) FormatWaitEvent(e event.WaitEvent) error {
    67  	c.waitEvents = append(c.waitEvents, e)
    68  	return nil
    69  }
    70  
    71  func (c *countingFormatter) FormatErrorEvent(e event.ErrorEvent) error {
    72  	c.errorEvent = e
    73  	return nil
    74  }
    75  
    76  func (c *countingFormatter) FormatActionGroupEvent(
    77  	e event.ActionGroupEvent,
    78  	_ []event.ActionGroup,
    79  	_ stats.Stats,
    80  	_ Collector,
    81  ) error {
    82  	c.actionGroupEvent = append(c.actionGroupEvent, e)
    83  	return nil
    84  }
    85  
    86  func (c *countingFormatter) FormatSummary(s stats.Stats) error {
    87  	return nil
    88  }
    89  

View as plain text