From 22db6bcb7c593673a1403eadf54e37e065c8109b Mon Sep 17 00:00:00 2001
From: Florian Fromm <florian.fromm@camao-tec.com>
Date: Mon, 20 Mar 2023 23:01:41 +0100
Subject: [PATCH] test: add tests for file

---
 .../tests/standard_apis/directory.rs          |  2 +-
 crates/dagger-sdk/tests/standard_apis/file.rs | 62 +++++++++++++++++++
 crates/dagger-sdk/tests/standard_apis/git.rs  |  2 +-
 crates/dagger-sdk/tests/standard_apis/mod.rs  |  1 +
 4 files changed, 65 insertions(+), 2 deletions(-)
 create mode 100644 crates/dagger-sdk/tests/standard_apis/file.rs

diff --git a/crates/dagger-sdk/tests/standard_apis/directory.rs b/crates/dagger-sdk/tests/standard_apis/directory.rs
index f54c091..cf8f942 100644
--- a/crates/dagger-sdk/tests/standard_apis/directory.rs
+++ b/crates/dagger-sdk/tests/standard_apis/directory.rs
@@ -1,4 +1,4 @@
-use dagger_sdk::{connect, ContainerExecOptsBuilder};
+use dagger_sdk::connect;
 use pretty_assertions::assert_eq;
 
 #[tokio::test]
diff --git a/crates/dagger-sdk/tests/standard_apis/file.rs b/crates/dagger-sdk/tests/standard_apis/file.rs
new file mode 100644
index 0000000..8abc043
--- /dev/null
+++ b/crates/dagger-sdk/tests/standard_apis/file.rs
@@ -0,0 +1,62 @@
+use dagger_sdk::connect;
+use pretty_assertions::assert_eq;
+
+static TEST_FILE_PATH: &str = "./data/";
+static TEST_FILE_NAME: &str = "test.txt";
+static TEST_FILE_CONTENT: &str = "This is a test file.";
+static ALPINE_VERSION: &str = "3.16.2";
+
+#[tokio::test]
+async fn test_file_from_host() -> eyre::Result<()> {
+    setup_file().await?;
+
+    let client = connect().await.unwrap();
+
+    let file = client.host().directory(TEST_FILE_PATH).file(TEST_FILE_NAME);
+    let file_content = file.contents().await?;
+    assert_eq!(file_content, TEST_FILE_CONTENT);
+
+    let file_retrieved_by_id = client.file(file.id().await?);
+    let file_retrieved_by_id_content = file_retrieved_by_id.contents().await?;
+    assert_eq!(file_retrieved_by_id_content, TEST_FILE_CONTENT);
+
+    cleanup_file().await
+}
+
+#[tokio::test]
+async fn test_file_from_container() -> eyre::Result<()> {
+    let client = connect().await.unwrap();
+
+    let alpine = client.container().from(format!("alpine:{}", ALPINE_VERSION));
+    let file = alpine.file("/etc/alpine-release");
+
+    let file_content = file.contents().await?;
+    let expected_file_content = format!("{}\n", ALPINE_VERSION);
+    assert_eq!(file_content, expected_file_content); 
+
+    let file_retrieved_by_id = client.file(file.id().await?);
+    let file_retrieved_by_id_contents = file_retrieved_by_id.contents().await?;
+    assert_eq!(file_retrieved_by_id_contents, expected_file_content);
+
+    Ok(())
+}
+
+async fn setup_file() -> eyre::Result<()> {
+    let full_file_path = String::from(TEST_FILE_PATH) + TEST_FILE_NAME;
+    tokio::fs::create_dir(TEST_FILE_PATH)
+        .await
+        .map_err(eyre::Error::from)?;
+    tokio::fs::write(full_file_path, TEST_FILE_CONTENT)
+        .await
+        .map_err(eyre::Error::from)
+}
+
+async fn cleanup_file() -> eyre::Result<()> {
+    let full_file_path = String::from(TEST_FILE_PATH) + TEST_FILE_NAME;
+    tokio::fs::remove_file(full_file_path)
+        .await
+        .map_err(eyre::Error::from)?;
+    tokio::fs::remove_dir(TEST_FILE_PATH)
+        .await
+        .map_err(eyre::Error::from)
+}
\ No newline at end of file
diff --git a/crates/dagger-sdk/tests/standard_apis/git.rs b/crates/dagger-sdk/tests/standard_apis/git.rs
index 590499d..087f68b 100644
--- a/crates/dagger-sdk/tests/standard_apis/git.rs
+++ b/crates/dagger-sdk/tests/standard_apis/git.rs
@@ -1,4 +1,4 @@
-use dagger_sdk::{connect, ContainerExecOptsBuilder};
+use dagger_sdk::connect;
 use pretty_assertions::assert_eq;
 
 #[tokio::test]
diff --git a/crates/dagger-sdk/tests/standard_apis/mod.rs b/crates/dagger-sdk/tests/standard_apis/mod.rs
index 3f69e0a..14c871f 100644
--- a/crates/dagger-sdk/tests/standard_apis/mod.rs
+++ b/crates/dagger-sdk/tests/standard_apis/mod.rs
@@ -1,5 +1,6 @@
 mod container;
 mod directory;
+mod file;
 mod git;
 
 use dagger_sdk::connect;