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)