diff options
Diffstat (limited to 'yarnutils/vars_tests.py')
-rw-r--r-- | yarnutils/vars_tests.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/yarnutils/vars_tests.py b/yarnutils/vars_tests.py new file mode 100644 index 0000000..a49903e --- /dev/null +++ b/yarnutils/vars_tests.py @@ -0,0 +1,43 @@ +# Copyright 2017 Lars Wirzenius +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. +# +# =*= License: GPL-3+ =*= + + +import shutil +import tempfile +import unittest + +import yarnutils + + +class VariablesTests(unittest.TestCase): + + def setUp(self): + self.tempdir = tempfile.mkdtemp() + + def tearDown(self): + shutil.rmtree(self.tempdir) + + def test_unset_variable_is_None(self): + vars = yarnutils.Variables(self.tempdir) + self.assertEqual(vars['foo'], None) + + def test_set_value_is_persistent(self): + vars1 = yarnutils.Variables(self.tempdir) + vars1['foo'] = 'bar' + vars2 = yarnutils.Variables(self.tempdir) + self.assertEqual(vars1['foo'], vars2['foo']) + self.assertEqual(vars2['foo'], 'bar') |