Improve config test.

Add step to clean config file. Now exercises both the AWSShell attributes and the configobj itself.

Author: donnemartin

tests/integration/test_config.py

@@ -10,46 +10,51 @@
 # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
+import mock
+import os
 import unittest
 
 from awsshell.app import AWSShell
 from awsshell.config import Config
+from awsshell.utils import build_config_file_path
 
 
 class ConfigTest(unittest.TestCase):
 
-    def setUp(self):
-        self.load_config()
-
-    def load_config(self):
-        config = Config()
-        self.config_obj = config.load('awsshellrc', 'test-awsshellrc')
-        self.config_section = self.config_obj['aws-shell']
-
     def test_config_off(self):
-        self.config_section['match_fuzzy'] = False
-        self.config_section['enable_vi_bindings'] = False
-        self.config_section['show_completion_columns'] = False
-        self.config_section['show_help'] = False
-        self.config_section['theme'] = 'none'
-        self.config_obj.write()
-        self.load_config()
-        assert self.config_section.as_bool('match_fuzzy') == False
-        assert self.config_section.as_bool('enable_vi_bindings') == False
-        assert self.config_section.as_bool('show_completion_columns') == False
-        assert self.config_section.as_bool('show_help') == False
-        assert self.config_section['theme'] == 'none'
+        try:
+            os.remove(build_config_file_path('test-awsshellrc'))
+        except OSError:
+            pass
+        self.aws_shell = AWSShell(None, mock.Mock(),
+                                  None, mock.Mock())
+        self.aws_shell.model_completer.match_fuzzy = False
+        self.aws_shell.enable_vi_bindings = False
+        self.aws_shell.show_completion_columns = False
+        self.aws_shell.show_help = False
+        self.aws_shell.theme = 'none'
+        self.aws_shell.save_config()
+        self.aws_shell.load_config()
+        assert self.aws_shell.model_completer.match_fuzzy == False
+        assert self.aws_shell.enable_vi_bindings == False
+        assert self.aws_shell.show_completion_columns == False
+        assert self.aws_shell.show_help == False
+        assert self.aws_shell.theme == 'none'
 
     def test_config_on(self):
-        self.config_section['match_fuzzy'] = True
-        self.config_section['enable_vi_bindings'] = True
-        self.config_section['show_completion_columns'] = True
-        self.config_section['show_help'] = True
-        self.config_section['theme'] = 'vim'
-        self.config_obj.write()
-        self.load_config()
-        assert self.config_section.as_bool('match_fuzzy') == True
-        assert self.config_section.as_bool('enable_vi_bindings') == True
-        assert self.config_section.as_bool('show_completion_columns') == True
-        assert self.config_section.as_bool('show_help') == True
-        assert self.config_section['theme'] == 'vim'
+        self.aws_shell = AWSShell(None, mock.Mock(),
+                                  None, mock.Mock())
+        self.aws_shell.model_completer.match_fuzzy = True
+        self.aws_shell.enable_vi_bindings = True
+        self.aws_shell.show_completion_columns = True
+        self.aws_shell.show_help = True
+        self.aws_shell.theme = 'vim'
+        self.aws_shell.save_config()
+        self.aws_shell.load_config()
+        assert self.aws_shell.config_section.as_bool('match_fuzzy') == True
+        assert self.aws_shell.config_section.as_bool(
+            'enable_vi_bindings') == True
+        assert self.aws_shell.config_section.as_bool(
+            'show_completion_columns') == True
+        assert self.aws_shell.config_section.as_bool('show_help') == True
+        assert self.aws_shell.config_section['theme'] == 'vim'