| OLD | NEW |
| 1 // Copyright (c) 2015, the Fletch project authors. Please see the AUTHORS file | 1 // Copyright (c) 2015, the Fletch project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE.md file. | 3 // BSD-style license that can be found in the LICENSE.md file. |
| 4 | 4 |
| 5 import 'dart:io'; | 5 import 'dart:io'; |
| 6 | 6 |
| 7 import 'package:expect/expect.dart'; | 7 import 'package:expect/expect.dart'; |
| 8 import 'package:mdns/src/constants.dart' show RRType; |
| 9 import 'package:mdns/src/lookup_resolver.dart'; |
| 8 import 'package:mdns/src/packet.dart'; | 10 import 'package:mdns/src/packet.dart'; |
| 9 import 'package:mdns/src/lookup_resolver.dart'; | |
| 10 | 11 |
| 11 main() async { | 12 main() async { |
| 12 await testTimeout(); | 13 await testTimeout(); |
| 13 await testResult(); | 14 await testResult(); |
| 14 await testResult2(); | 15 await testResult2(); |
| 15 await testResult3(); | 16 await testResult3(); |
| 16 } | 17 } |
| 17 | 18 |
| 19 ResourceRecord ip4Result(String name, InternetAddress address) { |
| 20 int validUntil = new DateTime.now().millisecondsSinceEpoch + 2000; |
| 21 return new ResourceRecord(RRType.A, name, address, validUntil); |
| 22 } |
| 23 |
| 18 testTimeout() async { | 24 testTimeout() async { |
| 19 var shortTimeout = new Duration(milliseconds: 1); | 25 var shortTimeout = new Duration(milliseconds: 1); |
| 20 var resolver = new LookupResolver(); | 26 var resolver = new LookupResolver(); |
| 21 var result = await resolver.addPendingRequest('xxx', shortTimeout); | 27 var result = resolver.addPendingRequest(RRType.A, 'xxx', shortTimeout); |
| 22 Expect.isNull(result); | 28 Expect.isTrue(await result.isEmpty); |
| 23 } | 29 } |
| 24 | 30 |
| 25 // One pending request and one response. | 31 // One pending request and one response. |
| 26 testResult() async { | 32 testResult() async { |
| 27 var noTimeout = new Duration(days: 1); | 33 var noTimeout = new Duration(days: 1); |
| 28 var resolver = new LookupResolver(); | 34 var resolver = new LookupResolver(); |
| 29 var futureResult = resolver.addPendingRequest('xxx', noTimeout); | 35 var futureResult = resolver.addPendingRequest(RRType.A, 'xxx', noTimeout); |
| 30 var response = new DecodeResult('xxx', new InternetAddress('1.2.3.4')); | 36 var response = ip4Result('xxx', new InternetAddress('1.2.3.4')); |
| 31 resolver.handleResponse([response]); | 37 resolver.handleResponse([response]); |
| 32 var result = await futureResult; | 38 var result = await futureResult.first; |
| 33 Expect.equals('1.2.3.4', result.address); | 39 Expect.equals('1.2.3.4', result.address.address); |
| 40 resolver.clearPendingRequests(); |
| 34 } | 41 } |
| 35 | 42 |
| 36 testResult2() async { | 43 testResult2() async { |
| 37 var noTimeout = new Duration(days: 1); | 44 var noTimeout = new Duration(days: 1); |
| 38 var resolver = new LookupResolver(); | 45 var resolver = new LookupResolver(); |
| 39 var futureResult1 = resolver.addPendingRequest('xxx', noTimeout); | 46 var futureResult1 = resolver.addPendingRequest(RRType.A, 'xxx', noTimeout); |
| 40 var futureResult2 = resolver.addPendingRequest('yyy', noTimeout); | 47 var futureResult2 = resolver.addPendingRequest(RRType.A, 'yyy', noTimeout); |
| 41 var response1 = new DecodeResult('xxx', new InternetAddress('1.2.3.4')); | 48 var response1 = ip4Result('xxx', new InternetAddress('1.2.3.4')); |
| 42 var response2 = new DecodeResult('yyy', new InternetAddress('2.3.4.5')); | 49 var response2 = ip4Result('yyy', new InternetAddress('2.3.4.5')); |
| 43 resolver.handleResponse([response2, response1]); | 50 resolver.handleResponse([response2, response1]); |
| 44 var result1 = await futureResult1; | 51 var result1 = await futureResult1.first; |
| 45 var result2 = await futureResult2; | 52 var result2 = await futureResult2.first; |
| 46 Expect.equals('1.2.3.4', result1.address); | 53 Expect.equals('1.2.3.4', result1.address.address); |
| 47 Expect.equals('2.3.4.5', result2.address); | 54 Expect.equals('2.3.4.5', result2.address.address); |
| 55 resolver.clearPendingRequests(); |
| 48 } | 56 } |
| 49 | 57 |
| 50 testResult3() async { | 58 testResult3() async { |
| 51 var noTimeout = new Duration(days: 1); | 59 var noTimeout = new Duration(days: 1); |
| 52 var resolver = new LookupResolver(); | 60 var resolver = new LookupResolver(); |
| 53 var response0 = new DecodeResult('zzz', new InternetAddress('2.3.4.5')); | 61 var response0 = ip4Result('zzz', new InternetAddress('2.3.4.5')); |
| 54 resolver.handleResponse([response0]); | 62 resolver.handleResponse([response0]); |
| 55 var futureResult1 = resolver.addPendingRequest('xxx', noTimeout); | 63 var futureResult1 = resolver.addPendingRequest(RRType.A, 'xxx', noTimeout); |
| 56 resolver.handleResponse([response0]); | 64 resolver.handleResponse([response0]); |
| 57 var futureResult2 = resolver.addPendingRequest('yyy', noTimeout); | 65 var futureResult2 = resolver.addPendingRequest(RRType.A, 'yyy', noTimeout); |
| 58 resolver.handleResponse([response0]); | 66 resolver.handleResponse([response0]); |
| 59 var response1 = new DecodeResult('xxx', new InternetAddress('1.2.3.4')); | 67 var response1 = ip4Result('xxx', new InternetAddress('1.2.3.4')); |
| 60 resolver.handleResponse([response0]); | 68 resolver.handleResponse([response0]); |
| 61 var response2 = new DecodeResult('yyy', new InternetAddress('2.3.4.5')); | 69 var response2 = ip4Result('yyy', new InternetAddress('2.3.4.5')); |
| 62 resolver.handleResponse([response0]); | 70 resolver.handleResponse([response0]); |
| 63 resolver.handleResponse([response2, response1]); | 71 resolver.handleResponse([response2, response1]); |
| 64 resolver.handleResponse([response0]); | 72 resolver.handleResponse([response0]); |
| 65 var result1 = await futureResult1; | 73 var result1 = await futureResult1.first; |
| 66 var result2 = await futureResult2; | 74 var result2 = await futureResult2.first; |
| 67 Expect.equals('1.2.3.4', result1.address); | 75 Expect.equals('1.2.3.4', result1.address.address); |
| 68 Expect.equals('2.3.4.5', result2.address); | 76 Expect.equals('2.3.4.5', result2.address.address); |
| 77 resolver.clearPendingRequests(); |
| 69 } | 78 } |
| OLD | NEW |