ShardsIntegrationSpec.scala

package com.twitter.gizzard.shards

import java.sql.SQLException
import scala.collection.mutable
import com.twitter.gizzard.thrift.conversions.Sequences._
import com.twitter.querulous.evaluator.QueryEvaluator
import org.specs.Specification
import org.specs.mock.{ClassMocker, JMocker}
import nameserver.{ManagingNameServer, SqlNameServer, ShardRepository}


object ShardsIntegrationSpec extends Specification with JMocker with ClassMocker with Database {

//  SqlNameServer.rebuildSchema(queryEvaluator)

  val shardInfo1 = new ShardInfo("com.example.UserShard", "table1", "localhost")
  val shardInfo2 = new ShardInfo("com.example.UserShard", "table2", "localhost")

  class UserShard(val shardInfo: ShardInfo, val weight: Int, val children: Seq[Shard]) extends Shard {
    val data = new mutable.HashMap[Int, String]

    def setName(id: Int, name: String) {
      data(id) = name
    }

    def getName(id: Int) = data(id)
  }

  val factory = new ShardFactory[UserShard] {
    def instantiate(shardInfo: ShardInfo, weight: Int, children: Seq[Shard]) = {
      new UserShard(shardInfo, weight, children)
    }

    def materialize(shardInfo: ShardInfo) {
      // nothing.
    }
  }

  "Shards" should {
    var shardRepository: ShardRepository[UserShard] = null
    var nameServer: ManagingNameServer = null

    doBefore {
      shardRepository = new ShardRepository
      shardRepository += (("com.example.UserShard", factory))
      shardRepository += (("com.example.SqlShard", factory))
      nameServer = new SqlNameServer(queryEvaluator, (a: ShardInfo) => ())

      nameServer.createShard(shardInfo1)
      nameServer.createShard(shardInfo2)
//      nameServer.reload()
    }

    "WriteOnlyShard" in {
      3 mustEqual 3
    }
  }
}