def test_default_profile(self, m_datastore): m_update_rules = Mock() m_datastore.profile_update_rules = m_update_rules m_profile = Mock() m_datastore.get_profile.return_value = m_profile calico_mesos._create_profile_with_default_mesos_rules("TESTPROF") new_rules = m_profile.rules host_net = calico_mesos._get_host_ip_net() # TODO: Better test for getting host ip self.assertIn(Rule(action="allow", src_net=host_net), new_rules.inbound_rules) self.assertIn(Rule(action="allow", src_tag="TESTPROF"), new_rules.inbound_rules) self.assertIn(Rule(action="allow"), new_rules.outbound_rules)
def test_default_profile(self, m_datastore): m_update_rules = Mock() m_datastore.profile_update_rules = m_update_rules m_profile = Mock() m_datastore.get_profile.return_value = m_profile calico_mesos._create_profile_with_default_mesos_rules("TESTPROF") new_rules = m_profile.rules host_net = calico_mesos._get_host_ip_net() #TODO: Better test for getting host ip self.assertIn(Rule(action="allow", src_net=host_net), new_rules.inbound_rules) self.assertIn(Rule(action="allow", src_tag="TESTPROF"), new_rules.inbound_rules) self.assertIn(Rule(action="allow"), new_rules.outbound_rules)
def test_get_host_ip_net_mainline(self, m_check_output): m_check_output.return_value = self.IP_ADDR1_OUTPUT ip_net = calico_mesos._get_host_ip_net() self.assertEqual(self.IP_ADDR1_NET, ip_net)