def test_get_ports(self):
        ports = ['80:80', '443:443', '8001:80001']

        example_service = {'image': 'nginx:1.19', 'ports': ports}

        service = ServiceDeclaration('phillyabc', example_service)
        self.assertEqual(ports, service.get_ports())
    def test_get_ports_returned_as_dict(self):
        """
        In newer versions of docker-compose the list of ports is returned as dict ex. {"published": 80, "target": 8000}
        :return:
        """

        service = ServiceDeclaration(
            'phillyabc', {
                'image': 'quay.io/riotkit/infracheck:v2.0',
                'ports': [{
                    'published': 80,
                    'target': 8000
                }]
            })

        self.assertEqual(['80:8000'], service.get_ports())