add unit tests

tests/cli/mod.rs

@@ -23,6 +23,7 @@ mod optimization;
 mod output_dir;
 mod overwrite;
 mod supress;
+mod version;
 mod vyper;
 
 /// The solidity contract name

tests/cli/version.rs

@@ -0,0 +1,30 @@
+use crate::{cli, common};
+use predicates::prelude::*;
+
+#[test]
+fn test_version() -> anyhow::Result<()> {
+    let _ = common::setup();
+
+    let args = &["--version"];
+    let result = cli::execute_zkvyper(args)?;
+
+    result
+        .success()
+        .stdout(predicate::str::contains("Vyper compiler for ZKsync"));
+
+    Ok(())
+}
+
+#[test]
+fn test_version_with_extra_args() -> anyhow::Result<()> {
+    let _ = common::setup();
+
+    let args = &["--version", cli::TEST_VYPER_CONTRACT_PATH];
+    let result = cli::execute_zkvyper(args)?;
+
+    result.failure().stderr(predicate::str::contains(
+        "Error: No other options are allowed while getting the compiler version.",
+    ));
+
+    Ok(())
+}