Add counter for mock invocations
This commit is contained in:
parent
876c29e57d
commit
15ed01cc0a
5 changed files with 131 additions and 112 deletions
|
@ -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
|
||||||
|
}
|
||||||
|
}
|
|
@ -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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -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}
|
||||||
|
}
|
||||||
|
}
|
|
@ -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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -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
|
||||||
|
}
|
||||||
|
}
|
Loading…
Add table
Add a link
Reference in a new issue