@@ -24,6 +24,8 @@ var _ = Describe("Socksify", func() {
2424 )
2525
2626 BeforeEach (func () {
27+ os .Unsetenv ("CREDHUB_PROXY" )
28+ os .Unsetenv ("https_proxy" )
2729 proxyDialer = & FakeProxyDialer {}
2830 origDial = credhub .DialFunc (func (x , y string ) (net.Conn , error ) {
2931 return nil , errors .New ("original dialer" )
@@ -50,7 +52,6 @@ var _ = Describe("Socksify", func() {
5052 err = ioutil .WriteFile (privateKeyPath , []byte ("some-key" ), 0600 )
5153 Expect (err ).NotTo (HaveOccurred ())
5254 os .Setenv ("CREDHUB_PROXY" , fmt .Sprintf ("ssh+socks5://localhost:12345?private-key=%s" , privateKeyPath ))
53-
5455 dialFunc = credhub .SOCKS5DialFuncFromEnvironment (origDial , proxyDialer )
5556 })
5657
@@ -60,6 +61,7 @@ var _ = Describe("Socksify", func() {
6061 Expect (proxyDialer .DialerCall .CallCount ).To (Equal (1 ))
6162 Expect (proxyDialer .DialerCall .Receives .Key ).To (Equal ("some-key" ))
6263 Expect (proxyDialer .DialerCall .Receives .URL ).To (Equal ("localhost:12345" ))
64+ os .Unsetenv ("CREDHUB_PROXY" )
6365 })
6466
6567 It ("Can be called multiple times and only create the dialer once" , func () {
@@ -70,6 +72,7 @@ var _ = Describe("Socksify", func() {
7072 Expect (proxyDialer .DialerCall .CallCount ).To (Equal (1 ))
7173 Expect (proxyDialer .DialerCall .Receives .Key ).To (Equal ("some-key" ))
7274 Expect (proxyDialer .DialerCall .Receives .URL ).To (Equal ("localhost:12345" ))
75+ os .Unsetenv ("CREDHUB_PROXY" )
7376 })
7477
7578 It ("Can be concurrently (run ginkgo with -race flag)" , func () {
@@ -87,6 +90,7 @@ var _ = Describe("Socksify", func() {
8790 Expect (proxyDialer .DialerCall .CallCount ).To (Equal (1 ))
8891 Expect (proxyDialer .DialerCall .Receives .Key ).To (Equal ("some-key" ))
8992 Expect (proxyDialer .DialerCall .Receives .URL ).To (Equal ("localhost:12345" ))
93+ os .Unsetenv ("CREDHUB_PROXY" )
9094 })
9195
9296 Context ("when the URL after the ssh+ prefix cannot be parsed" , func () {
@@ -97,6 +101,7 @@ var _ = Describe("Socksify", func() {
97101 It ("returns the dialer that was passed in" , func () {
98102 _ , err := dialFunc ("" , "" )
99103 Expect (err ).To (MatchError ("original dialer" ))
104+ os .Unsetenv ("CREDHUB_PROXY" )
100105 })
101106 })
102107
@@ -108,6 +113,7 @@ var _ = Describe("Socksify", func() {
108113 It ("returns the dialer that was passed in" , func () {
109114 _ , err := dialFunc ("" , "" )
110115 Expect (err ).To (MatchError ("original dialer" ))
116+ os .Unsetenv ("CREDHUB_PROXY" )
111117 })
112118 })
113119
@@ -119,6 +125,7 @@ var _ = Describe("Socksify", func() {
119125 It ("returns the dialer that was passed in" , func () {
120126 _ , err := dialFunc ("" , "" )
121127 Expect (err ).To (MatchError ("original dialer" ))
128+ os .Unsetenv ("CREDHUB_PROXY" )
122129 })
123130 })
124131
@@ -130,6 +137,7 @@ var _ = Describe("Socksify", func() {
130137 It ("returns the dialer that was passed in" , func () {
131138 _ , err := dialFunc ("" , "" )
132139 Expect (err ).To (MatchError ("original dialer" ))
140+ os .Unsetenv ("CREDHUB_PROXY" )
133141 })
134142 })
135143 })
@@ -144,6 +152,7 @@ var _ = Describe("Socksify", func() {
144152 It ("returns the dialer that was passed in" , func () {
145153 _ , err := dialFunc ("" , "" )
146154 Expect (err ).To (MatchError ("original dialer" ))
155+ os .Unsetenv ("CREDHUB_PROXY" )
147156 })
148157 })
149158
@@ -155,6 +164,7 @@ var _ = Describe("Socksify", func() {
155164 It ("returns the dialer that was passed in" , func () {
156165 _ , err := dialFunc ("" , "" )
157166 Expect (err ).To (MatchError ("original dialer" ))
167+ os .Unsetenv ("CREDHUB_PROXY" )
158168 })
159169 })
160170 })
0 commit comments