Add tests for setVersion

1 files changed, 40 insertions, 0 deletions

diff --git a/apps/encryption/tests/lib/KeyManagerTest.php b/apps/encryption/tests/lib/KeyManagerTest.php
index a8cb2dcc789..ea1830db4d5 100644
--- a/apps/encryption/tests/lib/KeyManagerTest.php
+++ b/apps/encryption/tests/lib/KeyManagerTest.php
@@ -606,4 +606,44 @@ class KeyManagerTest extends TestCase {
 		$this->assertSame(1337, $this->instance->getVersion('/admin/files/myfile.txt', $view));
 	}
 
+	public function testSetVersionWithFileInfo() {
+		$view = $this->getMockBuilder('\\OC\\Files\\View')
+			->disableOriginalConstructor()->getMock();
+		$cache = $this->getMockBuilder('\\OCP\\Files\\Cache\\ICache')
+			->disableOriginalConstructor()->getMock();
+		$cache->expects($this->once())
+			->method('update')
+			->with(123, ['encrypted' => 5, 'encryptedVersion' => 5]);
+		$storage = $this->getMockBuilder('\\OCP\\Files\\Storage')
+			->disableOriginalConstructor()->getMock();
+		$storage->expects($this->once())
+			->method('getCache')
+			->willReturn($cache);
+		$fileInfo = $this->getMockBuilder('\\OC\\Files\\FileInfo')
+			->disableOriginalConstructor()->getMock();
+		$fileInfo->expects($this->once())
+			->method('getStorage')
+			->willReturn($storage);
+		$fileInfo->expects($this->once())
+			->method('getId')
+			->willReturn(123);
+		$view->expects($this->once())
+			->method('getFileInfo')
+			->with('/admin/files/myfile.txt')
+			->willReturn($fileInfo);
+
+		$this->instance->setVersion('/admin/files/myfile.txt', 5, $view);
+	}
+
+	public function testSetVersionWithoutFileInfo() {
+		$view = $this->getMockBuilder('\\OC\\Files\\View')
+			->disableOriginalConstructor()->getMock();
+		$view->expects($this->once())
+			->method('getFileInfo')
+			->with('/admin/files/myfile.txt')
+			->willReturn(false);
+
+		$this->instance->setVersion('/admin/files/myfile.txt', 5, $view);
+	}
+
 }