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())
