/finagle-memcached/src/test/scala/com/twitter/finagle/memcached/stress/InterpreterServiceSpec.scala

package com.twitter.finagle.memcached.stress

import com.twitter.finagle.Service
import com.twitter.finagle.builder.ClientBuilder
import com.twitter.finagle.memcached.integration.InProcessMemcached
import com.twitter.finagle.memcached.protocol._
import com.twitter.finagle.memcached.protocol.text.Memcached
import com.twitter.finagle.memcached.util.ChannelBufferUtils._
import com.twitter.util.{Await, Time}
import java.net.InetSocketAddress
import org.specs.SpecificationWithJUnit

class InterpreterServiceSpec extends SpecificationWithJUnit {
  "InterpreterService" should {
    var server: InProcessMemcached = null
    var client: Service[Command, Response] = null

    doBefore {
      server = new InProcessMemcached(new InetSocketAddress(0))
      val address = server.start().localAddress
      client = ClientBuilder()
        .hosts(address)
        .codec(new Memcached)
        .hostConnectionLimit(1)
        .build()
    }

    doAfter {
      server.stop()
    }

    "set & get" in {
      val _key   = "key"
      val value = "value"
      val zero = "0"
      val start = System.currentTimeMillis
      (0 until 100) map { i =>
        val key = _key + i
        Await.result(client(Delete(key)))
        Await.result(client(Set(key, 0, Time.epoch, value)))
        Await.result(client(Get(Seq(key)))) mustEqual Values(Seq(Value(key, value, None, Some(zero))))
      }
      val end = System.currentTimeMillis
      // println("%d ms".format(end - start))
    }
  }
}