kytos-ng / flow_stats

        assert response.json == expected

        assert response.status_code == 200

    @patch("napps.amlight.flow_stats.main.Main.flow_stats_by_dpid_flow_id")

    def test_flow_stats_by_dpid_flow_id_with_dpid(self, mock_from_flow):

        """Test flow_stats rest call."""

        flow_stats = {

            'byte_count': 148,

            'duration_sec': 1589,

            'duration_nsec': 556000000,

            'packet_count': 2

            }

        flow_stats_dict_mock = {'6055f13593fad45e0b4699f49d56b105': flow_stats}

        flow_by_sw = {"00:00:00:00:00:00:00:01": flow_stats_dict_mock}

        mock_from_flow.return_value = flow_by_sw

        api = get_test_client(self.napp.controller, self.napp)

        endpoint = "/flow/stats?dpid=00:00:00:00:00:00:00:01"

        url = f"{self.server_name_url}"+endpoint

        response = api.get(url)

        expected = flow_by_sw

        assert response.json == expected

        assert response.status_code == 200

    def _patch_switch_flow(self, flow_id):

        """Helper method to patch controller to return switch/flow data."""

            json_response = response.json

            assert len(json_response) == 1

    @patch("napps.amlight.flow_stats.main.Main.flow_stats_by_dpid_flow_id")

    def test_flow_stats_by_dpid_flow_id(self, mock_from_flow):

        """Test flow_stats rest call."""

        flow_stats = {

            'byte_count': 148,

            'duration_sec': 1589,

            'duration_nsec': 556000000,

            'packet_count': 2

            }

        flow_stats_dict_mock = {'6055f13593fad45e0b4699f49d56b105': flow_stats}

        flow_by_sw = {"00:00:00:00:00:00:00:01": flow_stats_dict_mock}

        mock_from_flow.return_value = flow_by_sw

        api = get_test_client(self.napp.controller, self.napp)

        endpoint = "/flow/stats?dpid=00:00:00:00:00:00:00:01"

        url = f"{self.server_name_url}"+endpoint

        response = api.get(url)

        expected = flow_by_sw

        assert response.json == expected

        assert response.status_code == 200

    @patch("napps.amlight.flow_stats.main.Main.flow_stats_by_dpid_flow_id")

    def test_flow_stats_by_dpid_flow_id_with_dpid(self, mock_from_flow):