mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-18 15:51:11 +00:00
tests: udpsrc: split out socket setup
This commit is contained in:
parent
e6f77948ac
commit
7b1774513e
1 changed files with 63 additions and 51 deletions
|
@ -26,85 +26,97 @@ static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
|
|||
GST_PAD_ALWAYS,
|
||||
GST_STATIC_CAPS_ANY);
|
||||
|
||||
GST_START_TEST (test_udpsrc_empty_packet)
|
||||
static gboolean
|
||||
udpsrc_setup (GstElement ** udpsrc, GSocket ** socket,
|
||||
GstPad ** sinkpad, GSocketAddress ** sa)
|
||||
{
|
||||
GstElement *udpsrc;
|
||||
GSocket *socket;
|
||||
GstPad *sinkpad;
|
||||
GInetAddress *ia;
|
||||
int port = 0;
|
||||
gchar *s;
|
||||
|
||||
udpsrc = gst_check_setup_element ("udpsrc");
|
||||
fail_unless (udpsrc != NULL);
|
||||
g_object_set (udpsrc, "port", 0, NULL);
|
||||
*udpsrc = gst_check_setup_element ("udpsrc");
|
||||
fail_unless (*udpsrc != NULL);
|
||||
g_object_set (*udpsrc, "port", 0, NULL);
|
||||
|
||||
sinkpad = gst_check_setup_sink_pad_by_name (udpsrc, &sinktemplate, "src");
|
||||
fail_unless (sinkpad != NULL);
|
||||
gst_pad_set_active (sinkpad, TRUE);
|
||||
*sinkpad = gst_check_setup_sink_pad_by_name (*udpsrc, &sinktemplate, "src");
|
||||
fail_unless (*sinkpad != NULL);
|
||||
gst_pad_set_active (*sinkpad, TRUE);
|
||||
|
||||
gst_element_set_state (udpsrc, GST_STATE_PLAYING);
|
||||
g_object_get (udpsrc, "port", &port, NULL);
|
||||
gst_element_set_state (*udpsrc, GST_STATE_PLAYING);
|
||||
g_object_get (*udpsrc, "port", &port, NULL);
|
||||
GST_INFO ("udpsrc port = %d", port);
|
||||
|
||||
socket = g_socket_new (G_SOCKET_FAMILY_IPV4, G_SOCKET_TYPE_DATAGRAM,
|
||||
*socket = g_socket_new (G_SOCKET_FAMILY_IPV4, G_SOCKET_TYPE_DATAGRAM,
|
||||
G_SOCKET_PROTOCOL_UDP, NULL);
|
||||
|
||||
if (socket != NULL) {
|
||||
GSocketAddress *sa;
|
||||
GInetAddress *ia;
|
||||
gchar *s;
|
||||
if (*socket == NULL) {
|
||||
GST_WARNING ("Could not create IPv4 UDP socket for unit test");
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
ia = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
|
||||
s = g_inet_address_to_string (ia);
|
||||
GST_LOG ("inet address %s", s);
|
||||
g_free (s);
|
||||
sa = g_inet_socket_address_new (ia, port);
|
||||
ia = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
|
||||
s = g_inet_address_to_string (ia);
|
||||
GST_LOG ("inet address %s", s);
|
||||
g_free (s);
|
||||
*sa = g_inet_socket_address_new (ia, port);
|
||||
g_object_unref (ia);
|
||||
|
||||
if (g_socket_send_to (socket, sa, "HeLL0", 0, NULL, NULL) == 0) {
|
||||
GST_INFO ("sent 0 bytes");
|
||||
if (g_socket_send_to (socket, sa, "HeLL0", 6, NULL, NULL) == 6) {
|
||||
GstMapInfo map;
|
||||
GstBuffer *buf;
|
||||
guint len;
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
GST_INFO ("sent 6 bytes");
|
||||
GST_START_TEST (test_udpsrc_empty_packet)
|
||||
{
|
||||
GSocketAddress *sa = NULL;
|
||||
GstElement *udpsrc = NULL;
|
||||
GSocket *socket = NULL;
|
||||
GstPad *sinkpad = NULL;
|
||||
|
||||
g_usleep (G_USEC_PER_SEC / 2);
|
||||
if (!udpsrc_setup (&udpsrc, &socket, &sinkpad, &sa))
|
||||
goto no_socket;
|
||||
|
||||
len = g_list_length (buffers);
|
||||
GST_INFO ("%u buffers", len);
|
||||
fail_unless (len == 1 || len == 2);
|
||||
if (g_socket_send_to (socket, sa, "HeLL0", 0, NULL, NULL) == 0) {
|
||||
GST_INFO ("sent 0 bytes");
|
||||
if (g_socket_send_to (socket, sa, "HeLL0", 6, NULL, NULL) == 6) {
|
||||
GstMapInfo map;
|
||||
GstBuffer *buf;
|
||||
guint len;
|
||||
|
||||
/* last buffer should be our HeLL0 string */
|
||||
buf = GST_BUFFER (g_list_nth_data (buffers, len - 1));
|
||||
gst_buffer_map (buf, &map, GST_MAP_READ);
|
||||
fail_unless_equals_int (map.size, 6);
|
||||
fail_unless_equals_string ((gchar *) map.data, "HeLL0");
|
||||
gst_buffer_unmap (buf, &map);
|
||||
GST_INFO ("sent 6 bytes");
|
||||
|
||||
/* if there's another buffer, it should be 0 bytes */
|
||||
if (len == 2) {
|
||||
buf = GST_BUFFER (g_list_nth_data (buffers, 0));
|
||||
fail_unless_equals_int (gst_buffer_get_size (buf), 0);
|
||||
}
|
||||
} else {
|
||||
GST_WARNING ("send_to(6 bytes) failed");
|
||||
g_usleep (G_USEC_PER_SEC / 2);
|
||||
|
||||
len = g_list_length (buffers);
|
||||
GST_INFO ("%u buffers", len);
|
||||
fail_unless (len == 1 || len == 2);
|
||||
|
||||
/* last buffer should be our HeLL0 string */
|
||||
buf = GST_BUFFER (g_list_nth_data (buffers, len - 1));
|
||||
gst_buffer_map (buf, &map, GST_MAP_READ);
|
||||
fail_unless_equals_int (map.size, 6);
|
||||
fail_unless_equals_string ((gchar *) map.data, "HeLL0");
|
||||
gst_buffer_unmap (buf, &map);
|
||||
|
||||
/* if there's another buffer, it should be 0 bytes */
|
||||
if (len == 2) {
|
||||
buf = GST_BUFFER (g_list_nth_data (buffers, 0));
|
||||
fail_unless_equals_int (gst_buffer_get_size (buf), 0);
|
||||
}
|
||||
} else {
|
||||
GST_WARNING ("send_to(0 bytes) failed");
|
||||
GST_WARNING ("send_to(6 bytes) failed");
|
||||
}
|
||||
|
||||
g_object_unref (sa);
|
||||
g_object_unref (ia);
|
||||
} else {
|
||||
GST_WARNING ("Could not create IPv4 UDP socket for unit test");
|
||||
GST_WARNING ("send_to(0 bytes) failed");
|
||||
}
|
||||
|
||||
no_socket:
|
||||
|
||||
gst_element_set_state (udpsrc, GST_STATE_NULL);
|
||||
|
||||
gst_check_teardown_pad_by_name (udpsrc, "src");
|
||||
gst_check_teardown_element (udpsrc);
|
||||
|
||||
g_object_unref (socket);
|
||||
g_object_unref (sa);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
|
Loading…
Reference in a new issue