def test_convert_args(self):
client = BlivetGUIClient(MagicMock())
# 'normal' arguments
args = ["abcdef", 1, 1.01, True, None, Size("8 GiB")]
converted_args = client._args_convertTo_id(args)
self.assertEqual(converted_args, args)
# ClientProxyObject (id of object should be returned)
args = [ClientProxyObject(MagicMock(), ProxyID())]
converted_args = client._args_convertTo_id(args)
self.assertTrue(isinstance(converted_args[0], ProxyID))
self.assertEqual(converted_args, [arg.proxy_id for arg in args])
# ProxyDataContainer
args = [ProxyDataContainer(data1="abcdef", data2=Size("8 GiB"), data3=ClientProxyObject(MagicMock(), ProxyID()))]
converted_args = client._args_convertTo_id(args)
self.assertEqual(converted_args[0].data1, args[0].data1)
self.assertEqual(converted_args[0].data2, args[0].data2)
self.assertTrue(isinstance(converted_args[0].data3, ProxyID))
self.assertEqual(converted_args[0].data3, args[0].data3.proxy_id)
# ProxyDataContainer inside ProxyDataContainer
args = [ProxyDataContainer(data1="abcdef", data2=Size("8 GiB"), data3=ProxyDataContainer(dataA="xyz", dataB=ClientProxyObject(MagicMock(), ProxyID())))]
converted_args = client._args_convertTo_id(args)
self.assertEqual(converted_args[0].data1, args[0].data1)
self.assertEqual(converted_args[0].data2, args[0].data2)
self.assertTrue(isinstance(converted_args[0].data3, ProxyDataContainer))
self.assertEqual(converted_args[0].data3.dataA, args[0].data3.dataA)
self.assertTrue(isinstance(converted_args[0].data3.dataB, ProxyID))
self.assertEqual(converted_args[0].data3.dataB, args[0].data3.dataB.proxy_id)
def test_convert_answer(self):
# string (shouldn't be converted at all)
msg = "abcdef"
converted_answer = BlivetGUIClient._answer_convertTo_object(MagicMock(), msg)
self.assertEqual(msg, converted_answer)
# blivet.size.Size (is picklable too)
msg = Size("8 GiB")
converted_answer = BlivetGUIClient._answer_convertTo_object(MagicMock(), msg)
self.assertEqual(msg, converted_answer)
# ProxyID object
test_dict = {}
msg = ProxyID()
converted_answer = BlivetGUIClient._answer_convertTo_object(MagicMock(id_dict=test_dict), msg)
self.assertTrue(isinstance(converted_answer, ClientProxyObject))
self.assertEqual(converted_answer.proxy_id, msg) # pylint: disable=no-member
def test_convert_answer(self):
# string (shouldn't be converted at all)
msg = "abcdef"
converted_answer = BlivetGUIClient._answer_convertTo_object(
MagicMock(), msg)
self.assertEqual(msg, converted_answer)
# blivet.size.Size (is picklable too)
msg = Size("8 GiB")
converted_answer = BlivetGUIClient._answer_convertTo_object(
MagicMock(), msg)
self.assertEqual(msg, converted_answer)
# ProxyID object
test_dict = {}
msg = ProxyID()
converted_answer = BlivetGUIClient._answer_convertTo_object(
MagicMock(id_dict=test_dict), msg)
self.assertTrue(isinstance(converted_answer, ClientProxyObject))
self.assertEqual(converted_answer.proxy_id, msg) # pylint: disable=no-member
def test_convert_args(self):
client = BlivetGUIClient(MagicMock())
# 'normal' arguments
args = ["abcdef", 1, 1.01, True, None, Size("8 GiB")]
converted_args = client._args_convertTo_id(args)
self.assertEqual(converted_args, args)
# ClientProxyObject (id of object should be returned)
args = [ClientProxyObject(MagicMock(), ProxyID())]
converted_args = client._args_convertTo_id(args)
self.assertTrue(isinstance(converted_args[0], ProxyID))
self.assertEqual(converted_args, [arg.proxy_id for arg in args])
# ProxyDataContainer
args = [
ProxyDataContainer(data1="abcdef",
data2=Size("8 GiB"),
data3=ClientProxyObject(MagicMock(), ProxyID()))
]
converted_args = client._args_convertTo_id(args)
self.assertEqual(converted_args[0].data1, args[0].data1)
self.assertEqual(converted_args[0].data2, args[0].data2)
self.assertTrue(isinstance(converted_args[0].data3, ProxyID))
self.assertEqual(converted_args[0].data3, args[0].data3.proxy_id)
# ProxyDataContainer inside ProxyDataContainer
args = [
ProxyDataContainer(data1="abcdef",
data2=Size("8 GiB"),
data3=ProxyDataContainer(
dataA="xyz",
dataB=ClientProxyObject(
MagicMock(), ProxyID())))
]
converted_args = client._args_convertTo_id(args)
self.assertEqual(converted_args[0].data1, args[0].data1)
self.assertEqual(converted_args[0].data2, args[0].data2)
self.assertTrue(isinstance(converted_args[0].data3,
ProxyDataContainer))
self.assertEqual(converted_args[0].data3.dataA, args[0].data3.dataA)
self.assertTrue(isinstance(converted_args[0].data3.dataB, ProxyID))
self.assertEqual(converted_args[0].data3.dataB,
args[0].data3.dataB.proxy_id)
