test fixes

Automattic · Oct 4, 2023 · 306cdcf · 306cdcf
1 parent d6fc89e
commit 306cdcf
Showing 1 changed file with 19 additions and 4 deletions.
diff --git a/tests/files/test-vip-filesystem.php b/tests/files/test-vip-filesystem.php
@@ -266,8 +266,12 @@ public function test__filter_validate_file__valid_file() {
 		$basepath = $this->get_upload_path();
 
 		$stub = $this->getMockBuilder( VIP_Filesystem::class )
-				->setMethods( [ 'validate_file_type' ] )
+				->setMethods( [ 'validate_file_unique_name', 'validate_file_type' ] )
 				->getMock();
+
+		$stub->expects( $this->once() )
+				->method( 'validate_file_unique_name' )
+				->will( $this->returnValue( true ) );
 
 		$stub->expects( $this->once() )
 				->method( 'validate_file_type' )
@@ -287,13 +291,16 @@ public function test__filter_validate_file__file_exists() {
 		$basepath = $this->get_upload_path();
 
 		$stub = $this->getMockBuilder( VIP_Filesystem::class )
-				->setMethods( [ 'validate_file_unique_name' ] )
+				->setMethods( [ 'validate_file_unique_name', 'validate_file_type' ] )
 				->getMock();
 
 		$stub->expects( $this->once() )
 				->method( 'validate_file_unique_name' )
 				->with( $file['name'], $basepath . '/' )
 				->will( $this->returnValue( 'testfile_1696439069.txt' ) );
+
+		$stub->expects( $this->once() )
+				->method( 'validate_file_type' );
 
 		$actual = $stub->filter_validate_file( $file );
 
@@ -306,8 +313,12 @@ public function test__filter_validate_file__invalid_file_length() {
 		];
 
 		$stub = $this->getMockBuilder( VIP_Filesystem::class )
-				->setMethods( [ 'validate_file_type' ] )
+				->setMethods( [ 'validate_file_unique_name', 'validate_file_type' ] )
 				->getMock();
+
+		$stub->expects( $this->once() )
+				->method( 'validate_file_unique_name' )
+				->will( $this->returnValue( true ) );
 
 		$stub->expects( $this->once() )
 				->method( 'validate_file_type' );
@@ -328,9 +339,13 @@ public function test__filter_validate_file__invalid_file_type() {
 		$basepath = $this->get_upload_path();
 
 		$stub = $this->getMockBuilder( VIP_Filesystem::class )
-				->setMethods( [ 'validate_file_type' ] )
+				->setMethods( [ 'validate_file_unique_name', 'validate_file_type' ] )
 				->getMock();
 
+		$stub->expects( $this->once() )
+				->method( 'validate_file_unique_name' )
+				->will( $this->returnValue( true ) );
+
 		$stub->expects( $this->once() )
 				->method( 'validate_file_type' )
 				->with( $basepath . '/' . $file['name'] )