示例#1
0
    def test_get_env(self, mock_os):
        """Get Linux env_variable value"""
        def_value = "42"
        test_data = {"my_test_key_1": "my_test_value_1", "fake_path": "fake_path_value", "empty_env": " "}
        test_results = ["my_test_value_1", "fake_path_value", def_value]

        # Mock os.environ
        mock_os.environ = test_data

        test_cases_keys = test_data.keys()
        test_cases_keys.append("SHOULD_BE_NONEXISTENT_VARIABLE_FOR_SMM_TESTING_ONLY")
        test_results.append(def_value)

        for index, key in enumerate(test_cases_keys):
            self.assertEqual(get_env(key, def_value), test_results[index])
示例#2
0
from smm.utils import get_env, get_abs_path

# Templates
tmpl_missing_key = string.Template('Missing key [$key].')

DB_FILE_NAME = 'smmdbstore.json'
DB_DEF_PATH = '~/.smmdbstore/'
DB_FILE_NAME_BKP = '.' + DB_FILE_NAME + '.backup'
GROUP_ID = '__groups__'
DATA_ID = '__data__'
COUNT_ID = '__count__'
OWNER_ID = '__owner__'
COMM_TEXT = 'text'
COMM_GROUP = 'group'
COMM_DESC = 'description'
DB_PATH = get_env('SMMENVDBPATH', def_value=get_abs_path('~/.smmdbstore'))

INIT_STATE = {
    DATA_ID: {},
    GROUP_ID: [],
    COUNT_ID: 0,
    OWNER_ID: "anonymous"
}


class SmmDBCorrupted(Exception):
    pass


class SmmDBKeyError(Exception):
    pass