2019-06-10 09:33:19 +01:00
|
|
|
package home
|
2019-02-28 17:41:18 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
2021-02-09 13:51:44 +00:00
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
2019-02-28 17:41:18 +00:00
|
|
|
)
|
|
|
|
|
2021-03-23 09:32:07 +00:00
|
|
|
// TODO(a.garipov): Cover all migrations, use a testdata/ dir.
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
func TestUpgradeSchema1to2(t *testing.T) {
|
|
|
|
diskConf := testDiskConf(1)
|
2019-03-01 08:27:15 +00:00
|
|
|
|
2021-03-24 14:17:44 +00:00
|
|
|
err := upgradeSchema1to2(diskConf)
|
2021-02-09 13:51:44 +00:00
|
|
|
require.Nil(t, err)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
require.Equal(t, diskConf["schema_version"], 2)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
_, ok := diskConf["coredns"]
|
|
|
|
require.False(t, ok)
|
|
|
|
|
2021-03-23 09:32:07 +00:00
|
|
|
newDNSConf, ok := diskConf["dns"]
|
2021-02-09 13:51:44 +00:00
|
|
|
require.True(t, ok)
|
|
|
|
|
2021-03-23 09:32:07 +00:00
|
|
|
oldDNSConf := testDNSConf(1)
|
2021-02-09 13:51:44 +00:00
|
|
|
assert.Equal(t, oldDNSConf, newDNSConf)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2019-03-01 08:27:15 +00:00
|
|
|
oldExcludedEntries := []string{"coredns", "schema_version"}
|
|
|
|
newExcludedEntries := []string{"dns", "schema_version"}
|
2021-02-09 13:51:44 +00:00
|
|
|
oldDiskConf := testDiskConf(1)
|
|
|
|
assertEqualExcept(t, oldDiskConf, diskConf, oldExcludedEntries, newExcludedEntries)
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
func TestUpgradeSchema2to3(t *testing.T) {
|
|
|
|
diskConf := testDiskConf(2)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-03-24 14:17:44 +00:00
|
|
|
err := upgradeSchema2to3(diskConf)
|
2021-02-09 13:51:44 +00:00
|
|
|
require.Nil(t, err)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
require.Equal(t, diskConf["schema_version"], 3)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
dnsMap, ok := diskConf["dns"]
|
|
|
|
require.True(t, ok)
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-03-23 09:32:07 +00:00
|
|
|
newDNSConf, ok := dnsMap.(yobj)
|
|
|
|
require.True(t, ok)
|
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
bootstrapDNS := newDNSConf["bootstrap_dns"]
|
2019-02-28 17:41:18 +00:00
|
|
|
switch v := bootstrapDNS.(type) {
|
|
|
|
case []string:
|
2021-02-09 13:51:44 +00:00
|
|
|
require.Len(t, v, 1)
|
|
|
|
require.Equal(t, "8.8.8.8:53", v[0])
|
2019-02-28 17:41:18 +00:00
|
|
|
default:
|
2021-02-09 13:51:44 +00:00
|
|
|
t.Fatalf("wrong type for bootsrap dns: %T", v)
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
|
|
|
|
2019-03-01 08:27:15 +00:00
|
|
|
excludedEntries := []string{"bootstrap_dns"}
|
2021-03-23 09:32:07 +00:00
|
|
|
oldDNSConf := testDNSConf(2)
|
2021-02-09 13:51:44 +00:00
|
|
|
assertEqualExcept(t, oldDNSConf, newDNSConf, excludedEntries, excludedEntries)
|
2019-03-01 08:27:15 +00:00
|
|
|
|
|
|
|
excludedEntries = []string{"dns", "schema_version"}
|
2021-02-09 13:51:44 +00:00
|
|
|
oldDiskConf := testDiskConf(2)
|
|
|
|
assertEqualExcept(t, oldDiskConf, diskConf, excludedEntries, excludedEntries)
|
2019-03-01 08:27:15 +00:00
|
|
|
}
|
|
|
|
|
2021-03-23 09:32:07 +00:00
|
|
|
func TestUpgradeSchema7to8(t *testing.T) {
|
|
|
|
const host = "1.2.3.4"
|
|
|
|
oldConf := yobj{
|
|
|
|
"dns": yobj{
|
|
|
|
"bind_host": host,
|
|
|
|
},
|
|
|
|
"schema_version": 7,
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
|
|
|
|
2021-03-24 14:17:44 +00:00
|
|
|
err := upgradeSchema7to8(oldConf)
|
2021-03-23 09:32:07 +00:00
|
|
|
require.Nil(t, err)
|
|
|
|
|
|
|
|
require.Equal(t, oldConf["schema_version"], 8)
|
|
|
|
|
|
|
|
dnsVal, ok := oldConf["dns"]
|
|
|
|
require.True(t, ok)
|
|
|
|
|
|
|
|
newDNSConf, ok := dnsVal.(yobj)
|
|
|
|
require.True(t, ok)
|
|
|
|
|
|
|
|
newBindHosts, ok := newDNSConf["bind_hosts"].(yarr)
|
|
|
|
require.True(t, ok)
|
|
|
|
require.Len(t, newBindHosts, 1)
|
|
|
|
assert.Equal(t, host, newBindHosts[0])
|
2019-03-01 08:27:15 +00:00
|
|
|
}
|
2019-02-28 17:41:18 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
// assertEqualExcept removes entries from configs and compares them.
|
2021-03-23 09:32:07 +00:00
|
|
|
func assertEqualExcept(t *testing.T, oldConf, newConf yobj, oldKeys, newKeys []string) {
|
2021-02-09 13:51:44 +00:00
|
|
|
t.Helper()
|
2019-03-01 08:27:15 +00:00
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
for _, k := range oldKeys {
|
|
|
|
delete(oldConf, k)
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
2021-02-09 13:51:44 +00:00
|
|
|
for _, k := range newKeys {
|
|
|
|
delete(newConf, k)
|
2019-03-01 08:27:15 +00:00
|
|
|
}
|
2021-02-09 13:51:44 +00:00
|
|
|
|
|
|
|
assert.Equal(t, oldConf, newConf)
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
|
|
|
|
2021-03-23 09:32:07 +00:00
|
|
|
func testDiskConf(schemaVersion int) (diskConf yobj) {
|
|
|
|
filters := []filter{{
|
|
|
|
URL: "https://filters.adtidy.org/android/filters/111_optimized.txt",
|
|
|
|
Name: "Latvian filter",
|
|
|
|
RulesCount: 100,
|
|
|
|
}, {
|
|
|
|
URL: "https://easylist.to/easylistgermany/easylistgermany.txt",
|
|
|
|
Name: "Germany filter",
|
|
|
|
RulesCount: 200,
|
|
|
|
}}
|
|
|
|
diskConf = yobj{
|
2021-02-09 13:51:44 +00:00
|
|
|
"language": "en",
|
|
|
|
"filters": filters,
|
|
|
|
"user_rules": []string{},
|
|
|
|
"schema_version": schemaVersion,
|
|
|
|
"bind_host": "0.0.0.0",
|
|
|
|
"bind_port": 80,
|
|
|
|
"auth_name": "name",
|
|
|
|
"auth_pass": "pass",
|
|
|
|
}
|
|
|
|
|
|
|
|
dnsConf := testDNSConf(schemaVersion)
|
2019-02-28 17:41:18 +00:00
|
|
|
if schemaVersion > 1 {
|
2021-02-09 13:51:44 +00:00
|
|
|
diskConf["dns"] = dnsConf
|
2019-02-28 17:41:18 +00:00
|
|
|
} else {
|
2021-02-09 13:51:44 +00:00
|
|
|
diskConf["coredns"] = dnsConf
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
2021-02-09 13:51:44 +00:00
|
|
|
|
|
|
|
return diskConf
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
|
|
|
|
2021-02-09 13:51:44 +00:00
|
|
|
// testDNSConf creates a DNS config for test the way gopkg.in/yaml.v2 would
|
|
|
|
// unmarshal it. In YAML, keys aren't guaranteed to always only be strings.
|
2021-03-23 09:32:07 +00:00
|
|
|
func testDNSConf(schemaVersion int) (dnsConf yobj) {
|
|
|
|
dnsConf = yobj{
|
2021-02-09 13:51:44 +00:00
|
|
|
"port": 53,
|
|
|
|
"blocked_response_ttl": 10,
|
|
|
|
"querylog_enabled": true,
|
|
|
|
"ratelimit": 20,
|
|
|
|
"bootstrap_dns": "8.8.8.8:53",
|
|
|
|
"parental_sensitivity": 13,
|
|
|
|
"ratelimit_whitelist": []string{},
|
|
|
|
"upstream_dns": []string{"tls://1.1.1.1", "tls://1.0.0.1", "8.8.8.8"},
|
|
|
|
"filtering_enabled": true,
|
|
|
|
"refuse_any": true,
|
|
|
|
"parental_enabled": true,
|
|
|
|
"bind_host": "0.0.0.0",
|
|
|
|
"protection_enabled": true,
|
|
|
|
"safesearch_enabled": true,
|
|
|
|
"safebrowsing_enabled": true,
|
|
|
|
}
|
|
|
|
|
2019-02-28 17:41:18 +00:00
|
|
|
if schemaVersion > 2 {
|
2021-02-09 13:51:44 +00:00
|
|
|
dnsConf["bootstrap_dns"] = []string{"8.8.8.8:53"}
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|
2021-02-09 13:51:44 +00:00
|
|
|
|
|
|
|
return dnsConf
|
2019-02-28 17:41:18 +00:00
|
|
|
}
|