`waitFor:`, refactoring

src/Znock/ZnockBase.class.st

@@ -124,3 +124,9 @@ ZnockBase >> url: anObject [
 	builtClient url: anObject.
 	builtClient request url scheme: scheme
 ]
+
+{ #category : #public }
+ZnockBase >> waitFor: aDuration [
+	self response.
+	builtClient delay: aDuration
+]

src/Znock/ZnockClient.class.st

@@ -16,12 +16,27 @@ ZnockClient class >> newWithInterceptor: aZnock [
 
 { #category : #'private protocol' }
 ZnockClient >> executeRequestResponse [
-	^ (interceptor consumeExpectationFor: self)
-		ifNotNil: [ :expectation |
-			response := expectation response.
-			(response respondsTo: #signal) ifTrue: [ response signal ].
-			response contents ]
-		ifNil: [ super executeRequestResponse ]
+	| expectation |
+	expectation := interceptor consumeExpectationFor: self.
+	expectation ifNil: [ ^ super executeRequestResponse ].
+	expectation delay ifNotNil: [ :delay | self fakeDelay: delay ].
+	self fakeResponse: expectation response.
+	^ response contents
+]
+
+{ #category : #'private protocol' }
+ZnockClient >> fakeDelay: aDuration [
+	| localTimeout |
+	localTimeout := self timeout seconds.
+	localTimeout <= aDuration
+		ifTrue: [ localTimeout asDelay wait. ConnectionTimedOut signal: 'Znock timeout' ]
+		ifFalse: [ aDuration asDelay wait ]
+]
+
+{ #category : #'private protocol' }
+ZnockClient >> fakeResponse: aZnResponse [
+	response := aZnResponse.
+	(response respondsTo: #signal) ifTrue: [ response signal ]
 ]
 
 { #category : #accessing }

src/Znock/ZnockExpectation.class.st

@@ -1,14 +1,36 @@
 Class {
 	#name : #ZnockExpectation,
 	#superclass : #ZnClient,
+	#instVars : [
+		'delay'
+	],
 	#category : #Znock
 }
 
+{ #category : #accessing }
+ZnockExpectation >> delay [
+	^ delay
+]
+
+{ #category : #accessing }
+ZnockExpectation >> delay: anObject [
+	delay := anObject
+]
+
 { #category : #operations }
 ZnockExpectation >> execute [
 	"Do nothing"
 ]
 
+{ #category : #initialization }
+ZnockExpectation >> initialize [
+
+	super initialize.
+
+	delay := nil.
+	finalizer := nil.
+]
+
 { #category : #accessing }
 ZnockExpectation >> response: aZnResponse [
 	response := aZnResponse