add setenv, unsetenv function to os

pull/701/head
musou1500 2019-06-27 20:51:03 +09:00 committed by Alexander Medvednikov
parent b464adec43
commit 8c4f7749df
2 changed files with 35 additions and 0 deletions

14
os/os.v
View File

@ -316,6 +316,20 @@ pub fn getenv(key string) string {
return string(s) return string(s)
} }
pub fn setenv(name string, value string, overwrite bool) int {
return C.setenv(name.cstr(), value.cstr(), overwrite)
}
pub fn unsetenv(name string) int {
return C.unsetenv(name.cstr())
}
fn exit(code int) {
C.exit(code)
}
// `file_exists` returns true if `path` exists. // `file_exists` returns true if `path` exists.
pub fn file_exists(path string) bool { pub fn file_exists(path string) bool {
res := false res := false

21
os/os_test.v 100644
View File

@ -0,0 +1,21 @@
import os
fn test_setenv() {
os.setenv('foo', 'bar', true)
assert os.getenv('foo') == 'bar'
// `setenv` should not set if `overwrite` is false
os.setenv('foo', 'bar2', false)
assert os.getenv('foo') == 'bar'
// `setenv` should overwrite if `overwrite` is true
os.setenv('foo', 'bar2', true)
assert os.getenv('foo') == 'bar2'
}
fn test_unsetenv() {
os.setenv('foo', 'bar', true)
os.unsetenv('foo')
assert os.getenv('foo') == ''
}