Refactor packages

This commit is contained in:
Dominik Adam Przybysz 2014-12-09 23:09:01 +01:00
parent dfc0332d5a
commit fc4bda11de
6 changed files with 8 additions and 10 deletions

View file

@ -0,0 +1,80 @@
package pl.touk.mockserver.server
import com.sun.net.httpserver.HttpExchange
import groovy.util.slurpersupport.GPathResult
import java.util.concurrent.CopyOnWriteArrayList
import java.util.concurrent.CopyOnWriteArraySet
class HttpMockServer {
private final HttpServerWraper httpServerWraper
private final List<HttpServerWraper> childServers = new CopyOnWriteArrayList<>()
private final Set<String> mockNames = new CopyOnWriteArraySet<>()
HttpMockServer(int port = 9999){
httpServerWraper= new HttpServerWraper(port)
httpServerWraper.createContext('/serverControl', {
HttpExchange ex ->
ex.sendResponseHeaders(200, 0)
try{
GPathResult request = new XmlSlurper().parse(ex.requestBody)
if(ex.requestMethod== 'POST' && request.name() == 'addMock'){
addMock(request, ex)
}else if(ex.requestMethod == 'POST' && request.name() == 'removeMock'){
removeMock(request, ex)
}
//TODO add get mock report
//TODO add list mock
}catch(Exception e){
createErrorResponse(ex, e)
}
})
}
private void addMock(GPathResult request, HttpExchange ex) {
String name = request.name
if (name in mockNames) {
throw new RuntimeException('mock already registered')
}
println "Adding $name"
String mockPath = request.path
int mockPort = Integer.valueOf(request.port as String)
Closure predicate = Eval.me(request.predicate as String) as Closure
Closure okResponse = Eval.me(request.response as String) as Closure
boolean soap = Boolean.valueOf(request.soap as String)
Mock mock = new Mock(name, predicate, okResponse, soap)
HttpServerWraper child = childServers.find { it.port == mockPort }
if (!child) {
child = new HttpServerWraper(mockPort)
childServers << child
}
child.addMock(mockPath, mock)
mockNames << name
ex.responseBody << '<mockAdded/>'
ex.responseBody.close()
}
private void removeMock(GPathResult request, HttpExchange ex) {
String name = request.name
if (! (name in mockNames)) {
throw new RuntimeException('mock not registered')
}
println "Removing $name"
int used = childServers.inject(0) { int res, HttpServerWraper server-> server.removeMock(name) + res}
mockNames.remove(name)
ex.responseBody << "<mockRemoved>$used</mockRemoved>"
ex.responseBody.close()
}
private static void createErrorResponse(HttpExchange ex, Exception e) {
ex.responseBody << """<exceptionOccured>${e.message}</exceptionOccured>"""
ex.responseBody.close()
}
void stop(){
childServers.each {it.stop()}
httpServerWraper.stop()
}
}