From b5f16818977cb9b4ee71d6c1a5665abba54d5a64 Mon Sep 17 00:00:00 2001
From: black-desk <me@black-desk.cn>
Date: Sat, 17 Jun 2023 14:41:59 +0800
Subject: [PATCH] test(config): fix tests for remove repeater

Signed-off-by: black-desk <me@black-desk.cn>
---
 internal/config/config_test.go |  2 --
 internal/config/error.go       |  2 --
 test/data/example_config.yaml  | 26 ++++++++++----------------
 test/data/too_few_ports.yaml   | 28 ----------------------------
 test/data/validation_fail.yaml | 17 -----------------
 test/data/wrong_type.yaml      | 29 ++++++++++-------------------
 6 files changed, 20 insertions(+), 84 deletions(-)
 delete mode 100644 test/data/too_few_ports.yaml

diff --git a/internal/config/config_test.go b/internal/config/config_test.go
index 1b3c0b0..5e954ed 100644
--- a/internal/config/config_test.go
+++ b/internal/config/config_test.go
@@ -74,8 +74,6 @@ var _ = Describe("Configuration", func() {
 			Expect(err).To(MatchErr(expectErr))
 		})
 	},
-		ContextTableEntry("../../test/data/too_few_ports.yaml", ErrTooFewPorts).
-			WithFmt("../../test/data/too_few_ports.yaml"),
 		ContextTableEntry("../../test/data/wrong_type.yaml", new(yaml.TypeError)).
 			WithFmt("../../test/data/wrong_type.yaml"),
 		ContextTableEntry("../../test/data/validation_fail.yaml", validator.ValidationErrors{}).
diff --git a/internal/config/error.go b/internal/config/error.go
index 522d6ba..344574e 100644
--- a/internal/config/error.go
+++ b/internal/config/error.go
@@ -5,7 +5,5 @@ import (
 )
 
 var (
-	ErrTooFewPorts              = errors.New("Too few ports for tproxy")
-	ErrTooFewMarks              = errors.New("Too few marks for tproxy")
 	ErrCannotFoundCgroupv2Mount = errors.New("`cgroup2` mount point not found in /proc/mounts.")
 )
diff --git a/test/data/example_config.yaml b/test/data/example_config.yaml
index d1f497a..cc8b3c8 100644
--- a/test/data/example_config.yaml
+++ b/test/data/example_config.yaml
@@ -1,28 +1,22 @@
 version: 1
 cgroup-root: AUTO
-marks: '[100,300)'
+route-table: 300
 
 tproxies:
   clash:
     port: 7893
-    addr: 127.0.0.1
-
-proxies:
-  proxy1:
-    udp: true
-    port: 1080
-    protocol: socks
-    addr: 127.0.0.1
+    mark: 4000
+  clash-meta:
+    port: 8893
+    mark: 4001
+  v2ray:
+    port: 12345
+    mark: 4002
 
 rules:
   - match: /user.slice/user-1000.slice/user@1000.service/app.slice/*
     tproxy: clash
   - match: /user.slice/user-1001.slice/user@1001.service/app.slice/*
-    proxy: proxy1
+    tproxy: clash-meta
   - match: /user.slice/user-1002.slice/user@1002.service/app.slice/*
-    proxy: proxy1
-
-repeater:
-  listens:
-    - 127.0.0.1
-  tproxy-ports: "[13030,13040)"
+    tproxy: v2ray
diff --git a/test/data/too_few_ports.yaml b/test/data/too_few_ports.yaml
deleted file mode 100644
index dd86ec9..0000000
--- a/test/data/too_few_ports.yaml
+++ /dev/null
@@ -1,28 +0,0 @@
-version: 1
-cgroup-root: AUTO
-marks: '[100,300)'
-
-proxies:
-  socks1:
-    udp: true
-    port: 1080
-    protocol: socks
-    addr: 127.0.0.1
-  socks2:
-    udp: true
-    port: 1081
-    protocol: socks
-    addr: 127.0.0.1
-
-rules:
-  - match: /user.slice/user-1000.slice/user@1000.service/app.slice/*
-    proxy: socks1
-  - match: /user.slice/user-1001.slice/user@1001.service/app.slice/*
-    drop: true
-  - match: /user.slice/user-1002.slice/user@1002.service/app.slice/*
-    proxy: socks2
-
-repeater:
-  listens:
-    - 127.0.0.1
-  tproxy-ports: "[13030,13031)"
diff --git a/test/data/validation_fail.yaml b/test/data/validation_fail.yaml
index 6b6560f..4f67c74 100644
--- a/test/data/validation_fail.yaml
+++ b/test/data/validation_fail.yaml
@@ -1,28 +1,11 @@
 version: 1
 cgroup-root: AUTO
-marks: '[100,300)'
 
 tproxies:
   clash:
     port: 7893
     addr: 127.0.0.1
 
-proxies:
-  proxy1:
-    udp: true
-    port: 1080
-    protocol: socks
-    addr: 127.0.0.1
-
 rules:
   - match: /user.slice/user-1000.slice/user@1000.service/app.slice/*
     tproxy: clash
-  - match: /user.slice/user-1001.slice/user@1001.service/app.slice/*
-    proxy: proxy1
-  - match: /user.slice/user-1002.slice/user@1002.service/app.slice/*
-    proxy: proxy1
-
-repeater:
-  listens:
-    - 127.0.0
-  tproxy-ports: "[13030,13040)"
diff --git a/test/data/wrong_type.yaml b/test/data/wrong_type.yaml
index 0515e51..62d2154 100644
--- a/test/data/wrong_type.yaml
+++ b/test/data/wrong_type.yaml
@@ -1,28 +1,19 @@
 version: 1
 cgroup-root: AUTO
-marks: '[100,300)'
+route-table: 300
 
-proxies:
-  - socks1:
-      udp: true
-      port: 1080
-      protocol: socks
-      addr: 127.0.0.1
-  - socks2:
-      udp: true
-      port: 1081
-      protocol: socks
-      addr: 127.0.0.1
+tproxies:
+  - tproxy1:
+      port: 7893
+      mark: 4000
+  - tproxy2:
+      port: 8893
+      mark: 4001
 
 rules:
   - match: /user.slice/user-1000.slice/user@1000.service/app.slice/*
-    proxy: socks1
+    tproxy: socks1
   - match: /user.slice/user-1001.slice/user@1001.service/app.slice/*
     drop: true
   - match: /user.slice/user-1002.slice/user@1002.service/app.slice/*
-    proxy: socks2
-
-repeater:
-  listens:
-    - 127.0.0.1
-  tproxy-ports: "[13030,13031)"
+    tproxy: socks2