Browse Source

Extract generic #will: aBlock to be run when request is made.

Herby Vojčík 6 years ago
parent
commit
b892ef5328
1 changed files with 8 additions and 6 deletions
  1. 8 6
      src/Znock/ZnockBase.class.st

+ 8 - 6
src/Znock/ZnockBase.class.st

@@ -72,8 +72,7 @@ ZnockBase >> handleClientMessage: aMessage [
 { #category : #building }
 ZnockBase >> handleResponseCreationFromRequestMessage: aMessage [
 	aMessage argument ifNotNil: [ ^ self handleResponseCreationMessage: aMessage ].
-	self response: self newCleanResponse.
-	builtClient customizeResponseBlock: [ :req :res | 
+	self will: [ :req :res | 
 		aMessage argument: req.
 		res customizeFrom: (aMessage sendTo: ZnResponse) ]
 ]
@@ -81,8 +80,7 @@ ZnockBase >> handleResponseCreationFromRequestMessage: aMessage [
 { #category : #building }
 ZnockBase >> handleResponseCreationFromUrlMessage: aMessage [
 	aMessage argument ifNotNil: [ ^ self handleResponseCreationMessage: aMessage ].
-	self response: self newCleanResponse.
-	builtClient customizeResponseBlock: [ :req :res | 
+	self will: [ :req :res | 
 		aMessage argument: req url.
 		res customizeFrom: (aMessage sendTo: ZnResponse) ]
 ]
@@ -169,8 +167,12 @@ ZnockBase >> waitFor: aDuration [
 ]
 
 { #category : #public }
-ZnockBase >> willError: aString [
+ZnockBase >> will: aBlock [
 	self response: self newCleanResponse.
-	builtClient customizeResponseBlock: [ :req :res | self error: aString ]
+	builtClient customizeResponseBlock: aBlock
+]
 
+{ #category : #public }
+ZnockBase >> willError: aString [
+	self will: [ :req :res | self error: aString ]
 ]