def test_get_country_from_service_proposal(self): node = Node("key") node.proposal = json.dumps({ "service_definition": { "location_originate": { "country": "country code" }, }, }) self.assertEqual("country code", node.get_country_from_service_proposal())
def test_get_country_from_invalid_service_proposal(self): node = Node("key") node.proposal = json.dumps({"service_definition": {}}) self.assertIsNone(node.get_country_from_service_proposal())
def test_get_country_from_empty_service_proposal(self): node = Node("key") node.proposal = json.dumps({}) self.assertIsNone(node.get_country_from_service_proposal())