1 files changed, 7 insertions, 6 deletions
diff --git a/server/tests/table_test.rs b/server/tests/table_test.rs
index 1db7aa8..a131768 100644
--- a/server/tests/table_test.rs
+++ b/server/tests/table_test.rs
@@ -1,5 +1,5 @@
 use protocol::bridge_engine::TableState;
-use rand::{thread_rng, Rng};
+use rand::{thread_rng, Rng, random};
 use server::{
     error::BridgeError,
     table::{DbTable, InMemoryTable, Table},
@@ -13,13 +13,13 @@ where
     T: Table + Send,
 {
     assert!(matches!(table.state(), TableState::Unknown));
-    table = table.new_deal().await?;
+    table = table.set_board(0, random()).await?;
     assert!(matches!(table.state(), TableState::Game(_)));
     while matches!(table.state(), TableState::Game(_)) {
         table = server::table::advance_play(table).await?;
     }
     assert!(matches!(table.state(), TableState::Result(_)));
-    table = table.new_deal().await?;
+    table = table.set_board(1, random()).await?;
     assert!(matches!(table.state(), TableState::Game(_)));
 
     Ok(())
@@ -29,10 +29,11 @@ async fn advance_table<T>(table: Box<T>) -> Result<Box<T>, BridgeError>
 where
     T: Table + Send,
 {
+    let board_number = table.board_number() + 1;
     match table.state() {
         TableState::Unknown => panic!("unexpected state"),
         TableState::Game(_) => server::table::advance_play(table).await,
-        TableState::Result(_) => table.new_deal().await,
+        TableState::Result(_) => table.set_board(board_number, random()).await,
     }
 }
 
@@ -83,8 +84,8 @@ async fn db_table_persistence() -> Result<(), anyhow::Error> {
         DbTable::new(db.db().clone(), Uuid::new_v4(), InMemoryTable::new())
             .await?,
     );
-    table = table.new_deal().await?;
-    for i in 0..(thread_rng().gen_range(0..200)) {
+    table = table.set_board(1, random()).await?;
+    for _i in 0..(thread_rng().gen_range(0..200)) {
         table = advance_table(table).await?;
     }
     Ok(())