-- get some information about the test being run from an external file
-- so we can modify ourselves.
local mode = dofile("/tmp/proxyconfigmode.lua")

function mcp_config_pools(old)
    mcp.backend_use_iothread(true)
    mcp.backend_read_timeout(4)
    mcp.backend_connect_timeout(5)

    if mode == "none" then
        return {}
    elseif mode == "start" then
        local b1 = mcp.backend('b1', '127.0.0.1', 11511)
        local b2 = mcp.backend('b2', '127.0.0.1', 11512)
        local b3 = mcp.backend('b3', '127.0.0.1', 11513)

        local pools = {
            test = mcp.pool({b1, b2, b3})
        }
        return pools
    elseif mode == "betable" then
        local b1 = mcp.backend({ label = "b1", host = "127.0.0.1", port = 11511,
            connecttimeout = 2, retrytimeout = 5, readtimeout = 1,
            failurelimit = 0 })
        local b2 = mcp.backend({ label = "b2", host = "127.0.0.1", port = 11512,
            connecttimeout = 2, retrytimeout = 5, readtimeout = 5 })
        local b3 = mcp.backend({ label = "b3", host = "127.0.0.1", port = 11513,
            connecttimeout = 5, retrytimeout = 5, readtimeout = 5 })

        local pools = {
            test = mcp.pool({b1, b2, b3})
        }
        return pools
    elseif mode == "noiothread" then
        local b1 = mcp.backend('b1', '127.0.0.1', 11514)
        local b2 = mcp.backend('b2', '127.0.0.1', 11515)
        local b3 = mcp.backend('b3', '127.0.0.1', 11516)

        local pools = {
            test = mcp.pool({b1, b2, b3}, { iothread = false })
        }
        return pools
    elseif mode == "connections" then
        local b1 = mcp.backend({ label = "b1c", host = "127.0.0.1", port = 11511,
                               connections = 3})
        local pools = {
            test = mcp.pool({b1})
        }
        return pools
    elseif mode == "connectionsreload" then
        local b1 = mcp.backend({ label = "b1c", host = "127.0.0.1", port = 11511,
                               connections = 1})
        local pools = {
            test = mcp.pool({b1})
        }
        return pools
    elseif mode == "down" then
        local down = mcp.backend({ label = "down", host = "127.0.0.1", port = 11517,
                                   down = true })
        local pools = {
            test = mcp.pool({down})
        }
        return pools
    elseif mode == "notdown" then
        local down = mcp.backend({ label = "down", host = "127.0.0.1", port = 11517,
                                   down = false })
        local pools = {
            test = mcp.pool({down})
        }
        return pools
    end
end

-- At least to start we don't need to test every command, but we should do
-- some tests against the two broad types of commands (gets vs sets with
-- payloads)
function mcp_config_routes(zones)
    if mode == "none" then
        mcp.attach(mcp.CMD_MG, function(r) return "SERVER_ERROR no mg route\r\n" end)
        mcp.attach(mcp.CMD_MS, function(r) return "SERVER_ERROR no ms route\r\n" end)
    else
        local fg = mcp.funcgen_new()
        local h = fg:new_handle(zones["test"])
        fg:ready({
            f = function(rctx)
                return function(r)
                    return rctx:enqueue_and_wait(r, h)
                end
            end
        })

        mcp.attach(mcp.CMD_MG, fg)
        mcp.attach(mcp.CMD_MS, fg)
    end
end