Browse Source

Fix dns tests by setting v2ray.location.asset

Felix Yan 5 years ago
parent
commit
11bf22b234
1 changed files with 5 additions and 3 deletions
  1. 5 3
      infra/conf/dns_test.go

+ 5 - 3
infra/conf/dns_test.go

@@ -20,11 +20,12 @@ func init() {
 	wd, err := os.Getwd()
 	common.Must(err)
 
-	if _, err := os.Stat(platform.GetAssetLocation("geosite.dat")); err != nil && os.IsNotExist(err) {
-		common.Must(filesystem.CopyFile(platform.GetAssetLocation("geosite.dat"), filepath.Join(wd, "..", "..", "release", "config", "geosite.dat")))
+	if _, err := os.Stat(platform.GetAssetLocation("geoip.dat")); err != nil && os.IsNotExist(err) {
+		common.Must(filesystem.CopyFile(platform.GetAssetLocation("geoip.dat"), filepath.Join(wd, "..", "..", "release", "config", "geoip.dat")))
 	}
 
-	geositeFilePath := platform.GetAssetLocation("geosite.dat")
+	geositeFilePath := filepath.Join(wd, "geosite.dat")
+	os.Setenv("v2ray.location.asset", wd)
 	geositeFile, err := os.OpenFile(geositeFilePath, os.O_CREATE|os.O_WRONLY, 0600)
 	common.Must(err)
 	defer geositeFile.Close()
@@ -48,6 +49,7 @@ func TestDnsConfigParsing(t *testing.T) {
 	geositePath := platform.GetAssetLocation("geosite.dat")
 	defer func() {
 		os.Remove(geositePath)
+		os.Unsetenv("v2ray.location.asset")
 	}()
 
 	parserCreator := func() func(string) (proto.Message, error) {