Add counter for mock invocations

This commit is contained in:
Dominik Adam Przybysz 2014-12-07 21:47:39 +01:00
parent 876c29e57d
commit 15ed01cc0a
5 changed files with 131 additions and 112 deletions

View file

@ -0,0 +1,44 @@
package com.blogspot.przybyszd.mockserver
import com.sun.net.httpserver.HttpExchange
import groovy.util.slurpersupport.GPathResult
import java.util.concurrent.CopyOnWriteArrayList
class ContextExecutor {
private final HttpServerWraper httpServerWraper
private final String path
private final List<Mock> mocks
ContextExecutor(HttpServerWraper httpServerWraper, String path, Mock initialMock) {
this.httpServerWraper = httpServerWraper
this.path = path
this.mocks = new CopyOnWriteArrayList<>([initialMock])
httpServerWraper.createContext(path,{
HttpExchange ex ->
ex.sendResponseHeaders(200, 0)
String input = ex.requestBody.text
println "Mock received input"
GPathResult xml = new XmlSlurper().parseText(input)
for (Mock mock : mocks){
if(mock.predicate(xml)){
println "Mock ${mock.name} invoked"
++mock.counter
ex.responseBody << mock.responseOk(xml)
ex.responseBody.close()
return
}
}
ex.responseBody << "<invalidInput/>"
ex.responseBody.close()
})
}
int removeMock(String name) {
Mock mock = mocks.find {it.name == name}
if(mock){
mocks.remove(mock)
}
return mock.counter
}
}

View file

@ -1,20 +1,16 @@
package com.blogspot.przybyszd.mockserver package com.blogspot.przybyszd.mockserver
import com.sun.net.httpserver.HttpExchange import com.sun.net.httpserver.HttpExchange
import com.sun.net.httpserver.HttpHandler
import com.sun.net.httpserver.HttpServer
import groovy.transform.EqualsAndHashCode
import groovy.util.slurpersupport.GPathResult import groovy.util.slurpersupport.GPathResult
import java.util.concurrent.CopyOnWriteArrayList import java.util.concurrent.CopyOnWriteArrayList
import java.util.concurrent.CopyOnWriteArraySet import java.util.concurrent.CopyOnWriteArraySet
import java.util.concurrent.Executors
class HttpMockServer { class HttpMockServer {
HttpServerWraper httpServerWraper private final HttpServerWraper httpServerWraper
List<HttpServerWraper> childServers = new CopyOnWriteArrayList<>() private final List<HttpServerWraper> childServers = new CopyOnWriteArrayList<>()
Set<String> mockNames = new CopyOnWriteArraySet<>() private final Set<String> mockNames = new CopyOnWriteArraySet<>()
HttpMockServer(int port = 9999){ HttpMockServer(int port = 9999){
httpServerWraper= new HttpServerWraper(port) httpServerWraper= new HttpServerWraper(port)
@ -63,9 +59,10 @@ class HttpMockServer {
if (! (name in mockNames)) { if (! (name in mockNames)) {
throw new RuntimeException('mock not registered') throw new RuntimeException('mock not registered')
} }
childServers.each {it.removeMock(name)} println "Removing $name"
int used = childServers.inject(0) { int res, HttpServerWraper server-> server.removeMock(name) + res}
mockNames.remove(name) mockNames.remove(name)
ex.responseBody << '<mockRemoved/>' ex.responseBody << "<mockRemoved>$used</mockRemoved>"
ex.responseBody.close() ex.responseBody.close()
} }
@ -78,107 +75,4 @@ class HttpMockServer {
childServers.each {it.stop()} childServers.each {it.stop()}
httpServerWraper.stop() httpServerWraper.stop()
} }
private static final class HttpServerWraper{
final HttpServer httpServer
final int port
List<ContextExecutor> executors = []
HttpServerWraper(int port){
this.port = port
InetSocketAddress addr = new InetSocketAddress(Inet4Address.getByName("0.0.0.0"), port)
httpServer = HttpServer.create(addr, 0)
httpServer.executor = Executors.newCachedThreadPool()
println "Http server statrting on port $port..."
httpServer.start()
println 'Http server is started'
}
void createContext(String context, HttpHandler handler){
httpServer.createContext(context, handler)
}
void addMock(String path, Mock mock){
ContextExecutor executor = executors.find {it.path == path}
if(executor){
executor.mocks << mock
}else {
executors << new ContextExecutor(this, path, mock)
}
}
void stop(){
executors.each {httpServer.removeContext(it.path)}
httpServer.stop(0)
}
void removeMock(String name) {
executors.each {it.removeMock(name)}
}
}
private static final class ContextExecutor{
final HttpServerWraper httpServerWraper
final String path
List<Mock> mocks
ContextExecutor(HttpServerWraper httpServerWraper, String path, Mock initialMock) {
this.httpServerWraper = httpServerWraper
this.path = path
this.mocks = new CopyOnWriteArrayList<>([initialMock])
httpServerWraper.createContext(path,{
HttpExchange ex ->
ex.sendResponseHeaders(200, 0)
String input = ex.requestBody.text
println "Mock received input"
GPathResult xml = new XmlSlurper().parseText(input)
for (Mock mock : mocks){
if(mock.predicate(xml)){
ex.responseBody << mock.responseOk(xml)
ex.responseBody.close()
return
}
}
ex.responseBody << "<invalidInput/>"
ex.responseBody.close()
})
}
void removeMock(String name) {
Mock mock = mocks.find {it.name == name}
if(mock){
mocks.remove(mock)
}
}
}
@EqualsAndHashCode
private static final class Mock {
final String name
final Closure predicate
final Closure responseOk
//TODO add http method
//TODO add is soap method
Mock(String name, Closure predicate, Closure responseOk) {
this.name = name
this.predicate = predicate
this.responseOk = responseOk
}
}
static void main(String [] args) {
HttpMockServer httpMockServer = new HttpMockServer()
Runtime.runtime.addShutdownHook(new Thread({
println 'Http server is stopping...'
httpMockServer.stop()
println 'Http server is stopped'
} as Runnable))
while(true){
Thread.sleep(10000)
}
}
} }

View file

@ -0,0 +1,45 @@
package com.blogspot.przybyszd.mockserver
import com.sun.net.httpserver.HttpHandler
import com.sun.net.httpserver.HttpServer
import java.util.concurrent.Executors
class HttpServerWraper {
private final HttpServer httpServer
private final int port
private List<ContextExecutor> executors = []
HttpServerWraper(int port) {
this.port = port
InetSocketAddress addr = new InetSocketAddress(Inet4Address.getByName("0.0.0.0"), port)
httpServer = HttpServer.create(addr, 0)
httpServer.executor = Executors.newCachedThreadPool()
println "Http server statrting on port $port..."
httpServer.start()
println 'Http server is started'
}
void createContext(String context, HttpHandler handler) {
httpServer.createContext(context, handler)
}
void addMock(String path, Mock mock) {
ContextExecutor executor = executors.find { it.path == path }
if (executor) {
executor.mocks << mock
} else {
executors << new ContextExecutor(this, path, mock)
}
}
void stop() {
executors.each { httpServer.removeContext(it.path) }
httpServer.stop(0)
}
int removeMock(String name) {
executors.inject(0) { int res, ContextExecutor e -> e.removeMock(name) + res}
}
}

View file

@ -0,0 +1,17 @@
package com.blogspot.przybyszd.mockserver
class Main {
static void main(String [] args) {
HttpMockServer httpMockServer = new HttpMockServer()
Runtime.runtime.addShutdownHook(new Thread({
println 'Http server is stopping...'
httpMockServer.stop()
println 'Http server is stopped'
} as Runnable))
while(true){
Thread.sleep(10000)
}
}
}

View file

@ -0,0 +1,19 @@
package com.blogspot.przybyszd.mockserver
import groovy.transform.EqualsAndHashCode
@EqualsAndHashCode(excludes = ["counter"])
class Mock {
final String name
final Closure predicate
final Closure responseOk
//TODO add http method
//TODO add is soap method
int counter = 0
Mock(String name, Closure predicate, Closure responseOk) {
this.name = name
this.predicate = predicate
this.responseOk = responseOk
}
}