fix: read config data with bytes (python3)

Author: tomplus

Diff for: config/kube_config.py

@@ -100,8 +100,12 @@ def as_file(self):
         use_data_if_no_file = not self._file and self._data
         if use_data_if_no_file:
             if self._base64_file_content:
+                if isinstance(self._data, str):
+                    content = self._data.encode()
+                else:
+                    content = self._data
                 self._file = _create_temp_file_with_content(
-                    base64.decodestring(self._data.encode()))
+                    base64.decodestring(content))
             else:
                 self._file = _create_temp_file_with_content(self._data)
         if self._file and not os.path.isfile(self._file):

Diff for: config/kube_config_test.py

@@ -65,6 +65,9 @@ def _raise_exception(st):
 TEST_DATA = "test-data"
 TEST_DATA_BASE64 = _base64(TEST_DATA)
 
+TEST_DATA_BYTES = b"test-data"
+TEST_DATA_BASE64_BYTES = b"dGVzdC1kYXRh"
+
 TEST_ANOTHER_DATA = "another-test-data"
 TEST_ANOTHER_DATA_BASE64 = _base64(TEST_ANOTHER_DATA)
 
@@ -201,6 +204,18 @@ def test_create_temp_file_with_content(self):
                              _create_temp_file_with_content(TEST_DATA)))
         _cleanup_temp_files()
 
+    def test_file_given_data_bytes(self):
+        obj = {TEST_DATA_KEY: TEST_DATA_BASE64_BYTES}
+        t = FileOrData(obj=obj, file_key_name=TEST_FILE_KEY,
+                       data_key_name=TEST_DATA_KEY)
+        self.assertEqual(TEST_DATA, self.get_file_content(t.as_file()))
+
+    def test_file_given_data_bytes_no_base64(self):
+        obj = {TEST_DATA_KEY: TEST_DATA_BYTES}
+        t = FileOrData(obj=obj, file_key_name=TEST_FILE_KEY,
+                       data_key_name=TEST_DATA_KEY, base64_file_content=False)
+        self.assertEqual(TEST_DATA, self.get_file_content(t.as_file()))
+
 
 class TestConfigNode(BaseTestCase):