kytos-ng / flow_stats

        assert response.data == b"Flow does not exist"

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

    def test_bytes_count(self, mock_from_flow):

        """Test bytes_count rest call."""

        flow_id = '1'

        dpid_id = '1'

        mock_from_flow.return_value = self._get_mocked_flow_base()

        rest_name = "bytes_count"

        self._patch_switch_flow(flow_id)

        response = self._get_rest_response(rest_name, dpid_id)

        json_response = json.loads(response.data)

        assert json_response["flow_id"] == flow_id

        assert json_response["bytes_counter"] == 10

        assert json_response["bits_per_second"] == 4.0

    def test_packet_count_per_flow__empty(self):

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

        assert response.data == b"Flow does not exist"

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

    def test_packet_count(self, mock_from_flow):

        """Test packet_count rest call."""

        flow_id = '1'

        dpid_id = '1'

        mock_from_flow.return_value = self._get_mocked_flow_base()

        rest_name = "packet_count"

        self._patch_switch_flow(flow_id)

        response = self._get_rest_response(rest_name, dpid_id)

        json_response = json.loads(response.data)

        assert json_response["flow_id"] == flow_id

        assert json_response["packet_counter"] == 40

        assert json_response["packet_per_second"] == 2.0

    def test_bytes_count__fail(self):

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