def test_get_status_return_smth(self, m):
m.register_uri('GET',
self.MIST_APP_URL + 'status',
text="""
{
"mistVersion": "1.2.3",
"sparkVersion": "1.2.3"
}
""")
mist = MistApp()
status = mist.get_status()
self.assertIsNotNone(status)
def test_mist_cli_get_mist_versions(self):
mist = MistApp()
actual_status = dict(mistVersion="1.2.3", sparkVersion="4.5.6", javaVersion=dict(runtimeVersion="7.8.9"))
mist.get_status = MagicMock(return_value=actual_status)
t = get_mist_versions(mist)
self.assertEqual(t[0], "1.2.3")
self.assertEqual(t[1], "4.5.6")
self.assertEqual(t[2], "7.8.9")
status_wo_java_version = dict(mistVersion="1.2.3", sparkVersion="4.5.6")
mist.get_status = MagicMock(return_value=status_wo_java_version)
t = get_mist_versions(mist)
self.assertEqual(t[0], "1.2.3")
self.assertEqual(t[1], "4.5.6")
self.assertEqual(t[2], "UNKNOWN")
status_wo_java_version = dict(mistVersion="1.2.3", sparkVersion="4.5.6", javaVersion=dict())
mist.get_status = MagicMock(return_value=status_wo_java_version)
t = get_mist_versions(mist)
self.assertEqual(t[0], "1.2.3")
self.assertEqual(t[1], "4.5.6")
self.assertEqual(t[2], "UNKNOWN")
status_wo_mist_spark_version = dict(javaVersion=dict(runtimeVersion="7.8.9"))
mist.get_status = MagicMock(return_value=status_wo_mist_spark_version)
t = get_mist_versions(mist)
self.assertEqual(t[0], "UNKNOWN")
self.assertEqual(t[1], "UNKNOWN")
self.assertEqual(t[2], "7.8.9")
def test_get_status_return_empty_dict(self, m):
m.register_uri('GET', self.MIST_APP_URL + 'status', status_code=404)
mist = MistApp()
status = mist.get_status()
self.assertIsNotNone(status)
self.assertEqual(len(status), 0)
