@@ -84,6 +84,7 @@ func TestServiceBodySetters(t *testing.T) {
8484 SetReferenceInstanceName ("refInstance" , "refEnv" ).
8585 SetIgnoreSpecBasedCreds (true ).
8686 SetStripOASExtensions (true ).
87+ SetStripOASServersBeforePublish ().
8788 SetInstanceLifecycle ("stage" , "active" , "" ).
8889 SetServiceEndpoints (ep )
8990
@@ -136,6 +137,7 @@ func TestServiceBodySetters(t *testing.T) {
136137 assert .Equal (t , "refEnv/refInstance" , sb .GetReferenceInstanceName ())
137138 assert .Equal (t , true , sb .ignoreSpecBasesCreds )
138139 assert .Equal (t , true , sb .stripOASExtensions )
140+ assert .Equal (t , true , sb .stripOASServersBeforePublish )
139141 instanceLifecycle := sb .GetInstanceLifeCycle ()
140142 assert .NotNil (t , instanceLifecycle )
141143 assert .Equal (t , "stage" , instanceLifecycle .Stage )
@@ -222,16 +224,17 @@ func TestServiceBodyBuilderWithLargeSpec(t *testing.T) {
222224 "status" : "active" ,
223225 }
224226
225- sb , err = serviceBuilder .
227+ serviceBuilder = serviceBuilder .
226228 SetDescription ("A sample Petstore server based on OpenAPI 3.0" ).
227229 SetAPIName ("petstore" ).
228230 SetAuthPolicy ("pass-through" ).
229231 SetStatus (PublishedStatus ).
230232 SetState (PublishedState ).
231233 SetServiceEndpoints (endpoints ).
232234 SetTags (tags ).
233- SetTeamName ("pet-team" ).
234- Build ()
235+ SetTeamName ("pet-team" )
236+
237+ sb , err = serviceBuilder .Build ()
235238
236239 // Assertions
237240 assert .Nil (t , err )
@@ -246,9 +249,17 @@ func TestServiceBodyBuilderWithLargeSpec(t *testing.T) {
246249 assert .Equal (t , endpoints , sb .Endpoints )
247250 assert .Equal (t , tags , sb .Tags )
248251 assert .NotEmpty (t , sb .SpecDefinition )
252+ assert .Empty (t , sb .originalSpecHash )
249253 assert .Equal (t , Oas3 , sb .ResourceType )
250254
251255 // Verify spec size constraints
252256 assert .Less (t , len (sb .SpecDefinition ), len (specBytes ), "processed spec should be smaller than original" )
253257 assert .Less (t , len (sb .SpecDefinition ), tenMB , "spec should be under 10MB limit" )
258+
259+ sb , err = serviceBuilder .
260+ SetStripOASServersBeforePublish ().
261+ Build ()
262+ assert .Nil (t , err )
263+ assert .NotNil (t , sb )
264+ assert .NotEqual (t , sb .originalSpecHash , sb .specHash )
254265}
0 commit comments