public void TestGetNodeForPartition() { var node = new NodeMock(); var node2 = new NodeMock(); var node3 = new NodeMock(); var routes = new Dictionary <string, Partition[]> { { "test2p", new[] { new Partition { Id = 1, Leader = node }, new Partition { Id = 2, Leader = node2 }, new Partition { Id = 3, Leader = node3 } } }, }; var rt = new RoutingTable(routes); Assert.AreSame(node, rt.GetLeaderForPartition("test2p", 1)); Assert.AreSame(node2, rt.GetLeaderForPartition("test2p", 2)); Assert.AreSame(node3, rt.GetLeaderForPartition("test2p", 3)); Assert.IsNull(rt.GetLeaderForPartition("test2p", 8)); Assert.IsNull(rt.GetLeaderForPartition("test2poulpe", 8423)); }