kytos-ng / kytos_stats

        assert response.status_code == 404

        assert response.json()["description"] == "Flow does not exist"

    @patch("napps.amlight.kytos_stats.main.Main.flow_from_id")

    async def test_bytes_count(self, mock_from_flow):

        """Test bytes_count rest call."""

        flow_id = '1'

        mock_from_flow.return_value = self._get_mocked_flow_base()

        self._patch_switch_flow(flow_id)

        endpoint = f"{self.base_endpoint}/bytes_count/{flow_id}"

        response = await self.api_client.get(endpoint)

        assert response.status_code == 200

        json_response = response.json()

        assert json_response["flow_id"] == flow_id

        assert json_response["bytes_counter"] == 10

        assert json_response["bits_per_second"] == 4.0

    async def test_packet_count_per_flow_empty(self):

        """Test packet_count rest call with a flow that does not exist ."""

        assert response.status_code == 404

        assert response.json()["description"] == "Flow does not exist"

    @patch("napps.amlight.kytos_stats.main.Main.flow_from_id")

    async def test_packet_count(self, mock_from_flow):

        """Test packet_count rest call."""

        flow_id = '1'

        mock_from_flow.return_value = self._get_mocked_flow_base()

        self._patch_switch_flow(flow_id)

        endpoint = f"{self.base_endpoint}/packet_count/{flow_id}"

        response = await self.api_client.get(endpoint)

        assert response.status_code == 200

        json_response = response.json()

        assert json_response["flow_id"] == flow_id

        assert json_response["packet_counter"] == 40

        assert json_response["packet_per_second"] == 2.0

    async def test_bytes_count_not_found(self):

        """Test bytes_count rest call with wrong flow_id."""