FileOrData: Handle None object
Return when object is None in FileOrData class Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com>
This commit is contained in:
parent
3ea80032c3
commit
a270ea2946
@ -97,6 +97,8 @@ class FileOrData(object):
|
||||
self._file = None
|
||||
self._data = None
|
||||
self._base64_file_content = base64_file_content
|
||||
if not obj:
|
||||
return
|
||||
if data_key_name in obj:
|
||||
self._data = obj[data_key_name]
|
||||
elif file_key_name in obj:
|
||||
|
||||
@ -255,6 +255,16 @@ class TestFileOrData(BaseTestCase):
|
||||
data_key_name=TEST_DATA_KEY, base64_file_content=False)
|
||||
self.assertEqual(TEST_DATA, self.get_file_content(t.as_file()))
|
||||
|
||||
def test_file_given_no_object(self):
|
||||
t = FileOrData(obj=None, file_key_name=TEST_FILE_KEY,
|
||||
data_key_name=TEST_DATA_KEY)
|
||||
self.assertEqual(t.as_file(), None)
|
||||
|
||||
def test_file_given_no_object_data(self):
|
||||
t = FileOrData(obj=None, file_key_name=TEST_FILE_KEY,
|
||||
data_key_name=TEST_DATA_KEY)
|
||||
self.assertEqual(t.as_data(), None)
|
||||
|
||||
|
||||
class TestConfigNode(BaseTestCase):
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user