-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathparser_pro_test.go
81 lines (66 loc) · 3.14 KB
/
parser_pro_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package whoisparser
import (
"github.com/stretchr/testify/assert"
"io/ioutil"
"testing"
)
func TestParserPro(t *testing.T) {
assertParamsMap := AssertParamsMap{
"Registrar": {
{TargetField: "DomainName", ExpectedResult: "GOOGLE.PRO", AssertType: AssertTypeEqual},
{TargetField: "DomainID", ExpectedResult: "D107300000000011545-LRMS", AssertType: AssertTypeEqual},
{TargetField: "WhoisServer", ExpectedResult: "whois.markmonitor.com", AssertType: AssertTypeEqual},
{TargetField: "ReferralURL", ExpectedResult: "http://www.markmonitor.com", AssertType: AssertTypeEqual},
{TargetField: "UpdatedDate", ExpectedResult: "2019-08-07T09:30:57Z", AssertType: AssertTypeEqual},
{TargetField: "CreatedDate", ExpectedResult: "2008-07-22T00:00:00Z", AssertType: AssertTypeEqual},
{TargetField: "ExpirationDate", ExpectedResult: "2020-09-08T00:00:00Z", AssertType: AssertTypeEqual},
{TargetField: "RegistrarID", ExpectedResult: "292", AssertType: AssertTypeEqual},
{TargetField: "RegistrarName", ExpectedResult: "MarkMonitor Inc.", AssertType: AssertTypeEqual},
{TargetField: "Emails", ExpectedResult: "[email protected]", AssertType: AssertTypeEqual},
{TargetField: "DomainStatus", ExpectedResult: "clientDeleteProhibited https://icann.org/epp#clientDeleteProhibited", AssertType: AssertTypeContains},
{TargetField: "NameServers", ExpectedResult: "NS1.GOOGLE.COM", AssertType: AssertTypeContains},
{TargetField: "DomainDNSSEC", ExpectedResult: "unsigned", AssertType: AssertTypeEqual},
},
"Registrant": {
{TargetField: "Organization", ExpectedResult: "Google Inc.", AssertType: AssertTypeEqual},
{TargetField: "Province", ExpectedResult: "CA", AssertType: AssertTypeEqual},
{TargetField: "Country", ExpectedResult: "US", AssertType: AssertTypeEqual},
},
}
testDataFilepath := "test_data/whois_pro/google.pro.txt"
parserName := ".pro parser"
runParserAssertions(t, proParser, parserName, testDataFilepath, assertParamsMap)
}
func TestParserProNoSuchDomainErr(t *testing.T) {
var fileBytes []byte
var err error
var text string
var whoisRecord *Record
fileBytes, err = ioutil.ReadFile("test_data/whois_pro/no_such_domain.txt")
assert.NoError(t, err, "failed to open file with test data")
text = string(fileBytes)
whoisRecord = proParser.Parse(text)
assert.True(t, whoisRecord.ErrCode == ErrCodeNoSuchDomain)
assert.Nil(t, whoisRecord.Registrar)
assert.Nil(t, whoisRecord.Registrant)
assert.Nil(t, whoisRecord.Bill)
assert.Nil(t, whoisRecord.Admin)
assert.Nil(t, whoisRecord.Tech)
}
func TestParserProMalformedRequest(t *testing.T) {
var fileBytes []byte
var err error
var text string
var whoisRecord *Record
fileBytes, err = ioutil.ReadFile("test_data/whois_pro/malformed_request.txt")
assert.NoError(t, err, "failed to open file with test data")
text = string(fileBytes)
whoisRecord = proParser.Parse(text)
assert.NoError(t, err, "failed to open file with test data")
assert.True(t, whoisRecord.ErrCode == ErrCodeMalformedRequest)
assert.Nil(t, whoisRecord.Registrar)
assert.Nil(t, whoisRecord.Registrant)
assert.Nil(t, whoisRecord.Bill)
assert.Nil(t, whoisRecord.Admin)
assert.Nil(t, whoisRecord.Tech)
}