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 })