whois-parser/parser_ru_test.go at master · icamys/whois-parser · GitHub

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
package whoisparser

import (
	"github.com/stretchr/testify/assert"
	"io/ioutil"
	"testing"
)

func TestParserRu(t *testing.T) {
	assertParamsMap := AssertParamsMap{
		"Registrar": {
			{TargetField: "DomainName", ExpectedResult: "GG.RU", AssertType: AssertTypeEqual},
			{TargetField: "CreatedDate", ExpectedResult: "2001-12-03T21:00:00Z", AssertType: AssertTypeEqual},
			{TargetField: "DomainStatus", ExpectedResult: "REGISTERED, DELEGATED, VERIFIED", AssertType: AssertTypeEqual},
			{TargetField: "ExpirationDate", ExpectedResult: "2020-01-05", AssertType: AssertTypeEqual},
			{TargetField: "NameServers", ExpectedResult: "ns1.privateperson.ru", AssertType: AssertTypeContains},
			{TargetField: "NameServers", ExpectedResult: "ns2.privateperson.ru", AssertType: AssertTypeContains},
			{TargetField: "RegistrarName", ExpectedResult: "SALENAMES-RU", AssertType: AssertTypeEqual},
		},
		"Registrant": {
			{TargetField: "Organization", ExpectedResult: "Private Person", AssertType: AssertTypeEqual},
		},
	}

	testDataFilepath := "test_data/whois_ru/gg.ru.txt"
	parserName := ".ru parser"

	runParserAssertions(t, ruParser, parserName, testDataFilepath, assertParamsMap)
}

func TestParserRuRateLimit(t *testing.T) {
	var fileBytes []byte
	var err error
	var text string
	var whoisRecord *Record

	fileBytes, err = ioutil.ReadFile("test_data/whois_ru/rate_limit.txt")
	assert.NoError(t, err, "failed to open file with test data")

	text = string(fileBytes)

	whoisRecord = ruParser.Parse(text)

	assert.True(t, whoisRecord.ErrCode == ErrCodeRequestRateLimit)
	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 TestParserRuNoSuchDomainErr(t *testing.T) {
	var fileBytes []byte
	var err error
	var text string
	var whoisRecord *Record

	fileBytes, err = ioutil.ReadFile("test_data/whois_ru/no_such_domain.txt")
	assert.NoError(t, err, "failed to open file with test data")

	text = string(fileBytes)

	whoisRecord = ruParser.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)
}