fix(bindingtester): push error during get bytes on directoryPartition

Author: PierreZ

foundationdb-bindingtester/src/main.rs

@@ -2274,7 +2274,7 @@ impl StackMachine {
                     buf.push(element);
                 }
 
-                match self.get_current_item() {
+                match self.get_current_directory_item() {
                     Some(DirectoryStackItem::DirectoryOutput(
                         DirectoryOutput::DirectoryPartition(d),
                     )) => {
@@ -2300,7 +2300,7 @@ impl StackMachine {
             // the specified key. Push 1 if it does and 0 if it doesn't.
             DirectoryContains => {
                 let raw_prefix = self.pop_bytes().await;
-                let b = match self.get_current_item() {
+                let b = match self.get_current_directory_item() {
                     None => panic!("not found"),
                     Some(DirectoryStackItem::Subspace(s)) => s.is_start_of(&raw_prefix.to_vec()),
                     Some(DirectoryStackItem::DirectoryOutput(d)) => match d {
@@ -2355,23 +2355,44 @@ impl StackMachine {
                 };
                 let key = Subspace::from_bytes(&*raw_prefix).pack(&self.directory_index);
 
-                let value = match self.directory_stack.get(self.directory_index) {
+                match self.directory_stack.get(self.directory_index) {
                     None => panic!("nothing in the stack"),
                     Some(DirectoryStackItem::Null) => panic!("Directory is NULL"),
                     Some(DirectoryStackItem::Directory(_)) => {
                         panic!("trying to get a subspace, got a Directory")
                     }
-                    Some(DirectoryStackItem::DirectoryOutput(d)) => d.bytes(),
-                    Some(DirectoryStackItem::Subspace(s)) => s.bytes(),
+                    Some(DirectoryStackItem::DirectoryOutput(
+                        DirectoryOutput::DirectorySubspace(d),
+                    )) => {
+                        txn.set(&key, d.bytes());
+                        println!(
+                            "logging subspace [{}] {:?}={:?}",
+                            self.directory_index,
+                            unpack::<Vec<Element>>(&key).unwrap(),
+                            d.bytes(),
+                        );
+                    }
+                    Some(DirectoryStackItem::DirectoryOutput(
+                        DirectoryOutput::DirectoryPartition(_),
+                    )) => {
+                        self.push_directory_err(
+                            &instr.code,
+                            number,
+                            DirectoryError::Other(String::from(
+                                "cannot get key for the root of a directory partition",
+                            )),
+                        );
+                    }
+                    Some(DirectoryStackItem::Subspace(s)) => {
+                        txn.set(&key, s.bytes());
+                        println!(
+                            "logging subspace [{}] {:?}={:?}",
+                            self.directory_index,
+                            unpack::<Vec<Element>>(&key).unwrap(),
+                            s.bytes(),
+                        );
+                    }
                 };
-
-                txn.set(&key, &value);
-                println!(
-                    "logging subspace [{}] {:?}={:?}",
-                    self.directory_index,
-                    unpack::<Vec<Element>>(&key).unwrap(),
-                    &value
-                );
             }
 
             // Use the current directory for this operation.
@@ -2543,7 +2564,7 @@ impl StackMachine {
         }
     }
 
-    fn get_current_item(&self) -> Option<&DirectoryStackItem> {
+    fn get_current_directory_item(&mut self) -> Option<&DirectoryStackItem> {
         self.directory_stack.get(self.directory_index)
     }