test/run_pass/test_dgram_multicast_membership.js

   1 /* Copyright 2016-present Samsung Electronics Co., Ltd. and other contributors
   2  *
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 var assert = require('assert');
  17 var dgram = require('dgram');
  18
  19 var port = 41239;
  20 var multicast_address = '230.255.255.250';
  21 var interval = 100;
  22
  23 var recv_count = 0, send_count = 0;
  24
  25 var msg = 'Hello IoT.js';
  26 var client = dgram.createSocket('udp4');
  27 var server = dgram.createSocket('udp4');
  28
  29 server.on('error', function(err) {
  30   assert.fail();
  31   server.close();
  32 });
  33
  34 server.on('message', function(data, rinfo) {
  35   console.log('server got data : ' + data);
  36   recv_count++;
  37   if (recv_count == 1) {
  38     server.dropMembership(multicast_address);
  39   }
  40 });
  41
  42 server.bind(port, function() {
  43   server.addMembership(multicast_address);
  44 });
  45
  46 var timer = setInterval(function () {
  47   send_count++;
  48   client.send(msg, port, multicast_address);
  49   if (send_count == 3) {
  50     clearInterval(timer);
  51   }
  52 }, interval);
  53
  54 setTimeout(function() {
  55   server.close();
  56   client.close();
  57 }, 1000);
  58
  59 process.on('exit', function(code) {
  60   assert.equal(code, 0);
  61   assert.equal(recv_count, 1);
  62   assert.equal(send_count, 3);
  63 });