From 73cfce4ec2ebc2006b758d75576bc104df27ffac Mon Sep 17 00:00:00 2001
From: Stefan Hahmann <stefan.hahmann@tu-dresden.de>
Date: Mon, 10 Jun 2024 12:44:43 +0200
Subject: [PATCH] Add new unit test ProjectSaverTest

* tests saving a project to a zip file and a directory
---
 .../mastodon/mamut/io/ProjectSaverTest.java   | 97 +++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 src/test/java/org/mastodon/mamut/io/ProjectSaverTest.java

diff --git a/src/test/java/org/mastodon/mamut/io/ProjectSaverTest.java b/src/test/java/org/mastodon/mamut/io/ProjectSaverTest.java
new file mode 100644
index 000000000..28dd94d7b
--- /dev/null
+++ b/src/test/java/org/mastodon/mamut/io/ProjectSaverTest.java
@@ -0,0 +1,97 @@
+package org.mastodon.mamut.io;
+
+import ij.ImagePlus;
+import net.imagej.ImgPlus;
+import net.imagej.axis.Axes;
+import net.imagej.axis.AxisType;
+import net.imglib2.img.Img;
+import net.imglib2.img.array.ArrayImgs;
+import net.imglib2.img.display.imagej.ImgToVirtualStack;
+import net.imglib2.type.numeric.real.FloatType;
+import org.junit.Test;
+import org.mastodon.mamut.ProjectModel;
+import org.mastodon.mamut.io.project.MamutProject;
+import org.mastodon.mamut.model.Model;
+import org.mastodon.views.bdv.SharedBigDataViewerData;
+import org.scijava.Context;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.util.Objects;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+public class ProjectSaverTest
+{
+	@Test
+	public void testSaveProjectToZip() throws IOException
+	{
+		try (Context context = new Context())
+		{
+			Model model = new Model();
+			Img< FloatType > image = ArrayImgs.floats( 1, 1, 1 );
+			File mastodonFile = File.createTempFile( "test", ".mastodon" );
+			ProjectModel appModel = wrapAsAppModel( image, model, context, mastodonFile );
+			ProjectSaver.saveProject( mastodonFile, appModel );
+			assertNotNull( mastodonFile );
+			assertTrue( mastodonFile.exists() );
+			assertTrue( mastodonFile.isFile() );
+			assertTrue( mastodonFile.length() > 0 );
+			assertEquals( mastodonFile, appModel.getProject().getProjectRoot() );
+
+			ProjectSaver.saveProject( mastodonFile, appModel ); // Overwrite again
+			assertTrue( mastodonFile.exists() );
+			assertTrue( mastodonFile.isFile() );
+			assertTrue( mastodonFile.length() > 0 );
+			assertEquals( mastodonFile, appModel.getProject().getProjectRoot() );
+		}
+	}
+
+	@Test
+	public void testSaveProjectToDirectory() throws IOException
+	{
+		try (Context context = new Context())
+		{
+			Model model = new Model();
+			Img< FloatType > image = ArrayImgs.floats( 1, 1, 1 );
+			File mastodonDirectory = Files.createTempDirectory( "test" ).toFile();
+			ProjectModel appModel = wrapAsAppModel( image, model, context, mastodonDirectory );
+			ProjectSaver.saveProject( mastodonDirectory, appModel );
+			assertNotNull( mastodonDirectory );
+			assertTrue( mastodonDirectory.exists() );
+			assertTrue( mastodonDirectory.isDirectory() );
+			assertEquals( mastodonDirectory, appModel.getProject().getProjectRoot() );
+			File[] files = mastodonDirectory.listFiles();
+			assertNotNull( files );
+			assertEquals( 6, files.length );
+
+			ProjectSaver.saveProject( mastodonDirectory, appModel ); // Overwrite again
+			assertTrue( mastodonDirectory.exists() );
+			assertTrue( mastodonDirectory.isDirectory() );
+			assertEquals( mastodonDirectory, appModel.getProject().getProjectRoot() );
+			files = mastodonDirectory.listFiles();
+			assertNotNull( files );
+			assertEquals( 6, files.length );
+		}
+	}
+
+	private static ProjectModel wrapAsAppModel( final Img< FloatType > image, final Model model, final Context context, final File file )
+			throws IOException
+	{
+		final SharedBigDataViewerData sharedBigDataViewerData = asSharedBdvDataXyz( image );
+		MamutProject mamutProject = new MamutProject( file );
+		File datasetXmlFile = File.createTempFile( "test", ".xml" );
+		mamutProject.setDatasetXmlFile( datasetXmlFile );
+		return ProjectModel.create( context, model, sharedBigDataViewerData, mamutProject );
+	}
+
+	private static SharedBigDataViewerData asSharedBdvDataXyz( final Img< FloatType > image1 )
+	{
+		final ImagePlus image =
+				ImgToVirtualStack.wrap( new ImgPlus<>( image1, "image", new AxisType[] { Axes.X, Axes.Y, Axes.Z, Axes.TIME } ) );
+		return Objects.requireNonNull( SharedBigDataViewerData.fromImagePlus( image ) );
+	}
+}