github.com/cloudfoundry-attic/cli-with-i18n@v6.32.1-0.20171002233121-7401370d3b85+incompatible/command/v3/add_network_policy_command_test.go (about)

     1  package v3_test
     2  
     3  import (
     4  	"code.cloudfoundry.org/cli/actor/cfnetworkingaction"
     5  	"code.cloudfoundry.org/cli/actor/sharedaction"
     6  	"code.cloudfoundry.org/cli/actor/v3action"
     7  	"code.cloudfoundry.org/cli/command/commandfakes"
     8  	"code.cloudfoundry.org/cli/command/flag"
     9  	"code.cloudfoundry.org/cli/command/translatableerror"
    10  	. "code.cloudfoundry.org/cli/command/v3"
    11  	"code.cloudfoundry.org/cli/command/v3/v3fakes"
    12  	"code.cloudfoundry.org/cli/util/configv3"
    13  	"code.cloudfoundry.org/cli/util/ui"
    14  	. "github.com/onsi/ginkgo"
    15  	. "github.com/onsi/gomega"
    16  	. "github.com/onsi/gomega/gbytes"
    17  )
    18  
    19  var _ = Describe("add-network-policy Command", func() {
    20  	var (
    21  		cmd             AddNetworkPolicyCommand
    22  		testUI          *ui.UI
    23  		fakeConfig      *commandfakes.FakeConfig
    24  		fakeSharedActor *commandfakes.FakeSharedActor
    25  		fakeActor       *v3fakes.FakeAddNetworkPolicyActor
    26  		binaryName      string
    27  		executeErr      error
    28  		srcApp          string
    29  		destApp         string
    30  		protocol        string
    31  	)
    32  
    33  	BeforeEach(func() {
    34  		testUI = ui.NewTestUI(nil, NewBuffer(), NewBuffer())
    35  		fakeConfig = new(commandfakes.FakeConfig)
    36  		fakeSharedActor = new(commandfakes.FakeSharedActor)
    37  		fakeActor = new(v3fakes.FakeAddNetworkPolicyActor)
    38  
    39  		srcApp = "some-app"
    40  		destApp = "some-other-app"
    41  		protocol = "tcp"
    42  
    43  		cmd = AddNetworkPolicyCommand{
    44  			UI:             testUI,
    45  			Config:         fakeConfig,
    46  			SharedActor:    fakeSharedActor,
    47  			Actor:          fakeActor,
    48  			RequiredArgs:   flag.AddNetworkPolicyArgs{SourceApp: srcApp},
    49  			DestinationApp: destApp,
    50  		}
    51  
    52  		binaryName = "faceman"
    53  		fakeConfig.BinaryNameReturns(binaryName)
    54  	})
    55  
    56  	JustBeforeEach(func() {
    57  		executeErr = cmd.Execute(nil)
    58  	})
    59  
    60  	Context("when checking target fails", func() {
    61  		BeforeEach(func() {
    62  			fakeSharedActor.CheckTargetReturns(sharedaction.NotLoggedInError{BinaryName: binaryName})
    63  		})
    64  
    65  		It("returns an error", func() {
    66  			Expect(executeErr).To(MatchError(translatableerror.NotLoggedInError{BinaryName: binaryName}))
    67  
    68  			Expect(fakeSharedActor.CheckTargetCallCount()).To(Equal(1))
    69  			passedConfig, checkTargetedOrg, checkTargetedSpace := fakeSharedActor.CheckTargetArgsForCall(0)
    70  			Expect(passedConfig).To(Equal(fakeConfig))
    71  			Expect(checkTargetedOrg).To(BeTrue())
    72  			Expect(checkTargetedSpace).To(BeTrue())
    73  		})
    74  	})
    75  
    76  	Context("when the user is logged in, an org is targeted, and a space is targeted", func() {
    77  		BeforeEach(func() {
    78  			fakeConfig.CurrentUserReturns(configv3.User{Name: "some-user"}, nil)
    79  			fakeConfig.TargetedSpaceReturns(configv3.Space{Name: "some-space", GUID: "some-space-guid"})
    80  			fakeConfig.TargetedOrganizationReturns(configv3.Organization{Name: "some-org"})
    81  		})
    82  
    83  		Context("when protocol is specified but port is not", func() {
    84  			BeforeEach(func() {
    85  				cmd.Protocol = flag.NetworkProtocol{Protocol: protocol}
    86  			})
    87  
    88  			It("returns an error", func() {
    89  				Expect(executeErr).To(MatchError(translatableerror.NetworkPolicyProtocolOrPortNotProvidedError{}))
    90  				Expect(testUI.Out).NotTo(Say(`Adding network policy`))
    91  			})
    92  		})
    93  
    94  		Context("when port is specified but protocol is not", func() {
    95  			BeforeEach(func() {
    96  				cmd.Port = flag.NetworkPort{StartPort: 8080, EndPort: 8081}
    97  			})
    98  
    99  			It("returns an error", func() {
   100  				Expect(executeErr).To(MatchError(translatableerror.NetworkPolicyProtocolOrPortNotProvidedError{}))
   101  				Expect(testUI.Out).NotTo(Say(`Adding network policy`))
   102  			})
   103  		})
   104  
   105  		Context("when both protocol and port are specificed", func() {
   106  			BeforeEach(func() {
   107  				cmd.Protocol = flag.NetworkProtocol{Protocol: protocol}
   108  				cmd.Port = flag.NetworkPort{StartPort: 8080, EndPort: 8081}
   109  			})
   110  
   111  			Context("when the policy creation is successful", func() {
   112  				BeforeEach(func() {
   113  					fakeActor.AddNetworkPolicyReturns(cfnetworkingaction.Warnings{"some-warning-1", "some-warning-2"}, nil)
   114  				})
   115  
   116  				It("displays OK when no error occurs", func() {
   117  					Expect(executeErr).ToNot(HaveOccurred())
   118  					Expect(fakeActor.AddNetworkPolicyCallCount()).To(Equal(1))
   119  					passedSpaceGuid, passedSrcAppName, passedDestAppName, passedProtocol, passedStartPort, passedEndPort := fakeActor.AddNetworkPolicyArgsForCall(0)
   120  					Expect(passedSpaceGuid).To(Equal("some-space-guid"))
   121  					Expect(passedSrcAppName).To(Equal("some-app"))
   122  					Expect(passedDestAppName).To(Equal("some-other-app"))
   123  					Expect(passedProtocol).To(Equal("tcp"))
   124  					Expect(passedStartPort).To(Equal(8080))
   125  					Expect(passedEndPort).To(Equal(8081))
   126  
   127  					Expect(testUI.Out).To(Say(`Adding network policy to app %s in org some-org / space some-space as some-user\.\.\.`, srcApp))
   128  					Expect(testUI.Err).To(Say("some-warning-1"))
   129  					Expect(testUI.Err).To(Say("some-warning-2"))
   130  					Expect(testUI.Out).To(Say("OK"))
   131  				})
   132  			})
   133  
   134  			Context("when the policy creation is not successful", func() {
   135  				BeforeEach(func() {
   136  					fakeActor.AddNetworkPolicyReturns(cfnetworkingaction.Warnings{"some-warning-1", "some-warning-2"}, v3action.ApplicationNotFoundError{Name: srcApp})
   137  				})
   138  
   139  				It("does not display OK when an error occurs", func() {
   140  					Expect(executeErr).To(MatchError(translatableerror.ApplicationNotFoundError{Name: srcApp}))
   141  
   142  					Expect(testUI.Out).To(Say(`Adding network policy to app %s in org some-org / space some-space as some-user\.\.\.`, srcApp))
   143  					Expect(testUI.Err).To(Say("some-warning-1"))
   144  					Expect(testUI.Err).To(Say("some-warning-2"))
   145  					Expect(testUI.Out).ToNot(Say("OK"))
   146  				})
   147  			})
   148  		})
   149  
   150  		Context("when both protocol and port are not specified", func() {
   151  			It("defaults protocol to 'tcp' and port to '8080'", func() {
   152  				Expect(executeErr).ToNot(HaveOccurred())
   153  
   154  				Expect(fakeActor.AddNetworkPolicyCallCount()).To(Equal(1))
   155  				_, _, _, passedProtocol, passedStartPort, passedEndPort := fakeActor.AddNetworkPolicyArgsForCall(0)
   156  				Expect(passedProtocol).To(Equal("tcp"))
   157  				Expect(passedStartPort).To(Equal(8080))
   158  				Expect(passedEndPort).To(Equal(8080))
   159  			})
   160  		})
   161  	})
   162  })