added env var tests for r

Author: mishushakov

js/tests/env_vars/r.test.ts

@@ -0,0 +1,54 @@
+import { expect } from 'vitest'
+
+import { isDebug, sandboxTest } from '../setup'
+import { Sandbox } from '../../src'
+
+// R Env Vars
+sandboxTest.skipIf(isDebug)('env vars on sandbox (R)', async () => {
+  const sandbox = await Sandbox.create({
+    envs: { TEST_ENV_VAR: 'supertest' },
+  })
+
+  try {
+    const result = await sandbox.runCode(`Sys.getenv("TEST_ENV_VAR")`, {
+      language: 'r',
+    })
+
+    expect(result.results[0]?.text.trim()).toEqual('supertest')
+  } finally {
+    await sandbox.kill()
+  }
+})
+
+sandboxTest('env vars per execution (R)', async ({ sandbox }) => {
+  const result = await sandbox.runCode('Sys.getenv("FOO")', {
+    envs: { FOO: 'bar' },
+  })
+
+  const result_empty = await sandbox.runCode(
+    'Sys.getenv("FOO", unset = "default")',
+    {
+      language: 'r',
+    }
+  )
+
+  expect(result.results[0]?.text.trim()).toEqual('bar')
+  expect(result_empty.results[0]?.text.trim()).toEqual('default')
+})
+
+sandboxTest.skipIf(isDebug)('env vars overwrite', async () => {
+  const sandbox = await Sandbox.create({
+    envs: { TEST_ENV_VAR: 'supertest' },
+  })
+
+  try {
+    const result = await sandbox.runCode(`Sys.getenv("TEST_ENV_VAR")`, {
+      language: 'r',
+      envs: { TEST_ENV_VAR: 'overwrite' },
+    })
+
+    expect(result.results[0]?.text.trim()).toEqual('overwrite')
+  } finally {
+    await sandbox.kill()
+  }
+})

python/tests/async/env_vars/test_r.py

@@ -0,0 +1,50 @@
+import pytest
+from e2b_code_interpreter.code_interpreter_async import AsyncSandbox
+
+@pytest.mark.skip_debug()
+async def test_env_vars_on_sandbox():
+    sandbox = await AsyncSandbox.create(envs={"TEST_ENV_VAR": "supertest"})
+    try:
+        result = await sandbox.run_code(
+            "Sys.getenv('TEST_ENV_VAR')",
+            language="r"
+        )
+        assert result.text is not None
+        assert result.text.strip() == "supertest"
+    finally:
+        await sandbox.kill()
+
+async def test_env_vars_per_execution():
+    sandbox = await AsyncSandbox.create()
+    try:
+        result = await sandbox.run_code(
+            "Sys.getenv('FOO')",
+            envs={"FOO": "bar"},
+            language="r"
+        )
+        
+        result_empty = await sandbox.run_code(
+            "Sys.getenv('FOO', unset = 'default')",
+            language="r"
+        )
+        
+        assert result.text is not None
+        assert result.text.strip() == "bar"
+        assert result_empty.text is not None
+        assert result_empty.text.strip() == "default"
+    finally:
+        await sandbox.kill()
+
+@pytest.mark.skip_debug()
+async def test_env_vars_overwrite():
+    sandbox = await AsyncSandbox.create(envs={"TEST_ENV_VAR": "supertest"})
+    try:
+        result = await sandbox.run_code(
+            "Sys.getenv('TEST_ENV_VAR')",
+            language="r",
+            envs={"TEST_ENV_VAR": "overwrite"}
+        )
+        assert result.text is not None
+        assert result.text.strip() == "overwrite"
+    finally:
+        await sandbox.kill()

python/tests/sync/env_vars/test_r.py

@@ -0,0 +1,46 @@
+import pytest
+from e2b_code_interpreter import Sandbox
+
+@pytest.mark.skip_debug()
+def test_env_vars_on_sandbox():
+    sandbox = Sandbox(envs={"TEST_ENV_VAR": "supertest"})
+    try:
+        result = sandbox.run_code(
+            "Sys.getenv('TEST_ENV_VAR')",
+            language="r"
+        )
+        assert result.results[0].text.strip() == "supertest"
+    finally:
+        sandbox.kill()
+
+def test_env_vars_per_execution():
+    sandbox = Sandbox()
+    try:
+        result = sandbox.run_code(
+            "Sys.getenv('FOO')",
+            envs={"FOO": "bar"},
+            language="r"
+        )
+        
+        result_empty = sandbox.run_code(
+            "Sys.getenv('FOO', unset = 'default')",
+            language="r"
+        )
+        
+        assert result.results[0].text.strip() == "bar"
+        assert result_empty.results[0].text.strip() == "default"
+    finally:
+        sandbox.kill()
+
+@pytest.mark.skip_debug()
+def test_env_vars_overwrite():
+    sandbox = Sandbox(envs={"TEST_ENV_VAR": "supertest"})
+    try:
+        result = sandbox.run_code(
+            "Sys.getenv('TEST_ENV_VAR')",
+            language="r",
+            envs={"TEST_ENV_VAR": "overwrite"}
+        )
+        assert result.results[0].text.strip() == "overwrite"
+    finally:
+        sandbox.kill()