| 
									
										
										
										
											2019-08-29 15:41:59 +01:00
										 |  |  | package sign | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import ( | 
					
						
							|  |  |  | 	"testing" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-09-24 18:14:41 +02:00
										 |  |  | 	"github.com/coredns/caddy" | 
					
						
							| 
									
										
										
										
											2019-08-29 15:41:59 +01:00
										 |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func TestParse(t *testing.T) { | 
					
						
							|  |  |  | 	tests := []struct { | 
					
						
							|  |  |  | 		input     string | 
					
						
							|  |  |  | 		shouldErr bool | 
					
						
							|  |  |  | 		exp       *Signer | 
					
						
							|  |  |  | 	}{ | 
					
						
							|  |  |  | 		{`sign testdata/db.miek.nl miek.nl { | 
					
						
							|  |  |  | 			key file testdata/Kmiek.nl.+013+59725 | 
					
						
							|  |  |  | 		 }`, | 
					
						
							|  |  |  | 			false, | 
					
						
							|  |  |  | 			&Signer{ | 
					
						
							|  |  |  | 				keys:       []Pair{}, | 
					
						
							|  |  |  | 				origin:     "miek.nl.", | 
					
						
							|  |  |  | 				dbfile:     "testdata/db.miek.nl", | 
					
						
							|  |  |  | 				directory:  "/var/lib/coredns", | 
					
						
							|  |  |  | 				signedfile: "db.miek.nl.signed", | 
					
						
							|  |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{`sign testdata/db.miek.nl example.org { | 
					
						
							|  |  |  | 			key file testdata/Kmiek.nl.+013+59725 | 
					
						
							|  |  |  | 			directory testdata | 
					
						
							|  |  |  | 		 }`, | 
					
						
							|  |  |  | 			false, | 
					
						
							|  |  |  | 			&Signer{ | 
					
						
							|  |  |  | 				keys:       []Pair{}, | 
					
						
							|  |  |  | 				origin:     "example.org.", | 
					
						
							|  |  |  | 				dbfile:     "testdata/db.miek.nl", | 
					
						
							|  |  |  | 				directory:  "testdata", | 
					
						
							|  |  |  | 				signedfile: "db.example.org.signed", | 
					
						
							|  |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		// errors | 
					
						
							|  |  |  | 		{`sign db.example.org { | 
					
						
							|  |  |  | 			key file /etc/coredns/keys/Kexample.org | 
					
						
							|  |  |  | 		 }`, | 
					
						
							|  |  |  | 			true, | 
					
						
							|  |  |  | 			nil, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	for i, tc := range tests { | 
					
						
							|  |  |  | 		c := caddy.NewTestController("dns", tc.input) | 
					
						
							|  |  |  | 		sign, err := parse(c) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if err == nil && tc.shouldErr { | 
					
						
							|  |  |  | 			t.Fatalf("Test %d expected errors, but got no error", i) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		if err != nil && !tc.shouldErr { | 
					
						
							|  |  |  | 			t.Fatalf("Test %d expected no errors, but got '%v'", i, err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		if tc.shouldErr { | 
					
						
							|  |  |  | 			continue | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		signer := sign.signers[0] | 
					
						
							|  |  |  | 		if x := signer.origin; x != tc.exp.origin { | 
					
						
							|  |  |  | 			t.Errorf("Test %d expected %s as origin, got %s", i, tc.exp.origin, x) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		if x := signer.dbfile; x != tc.exp.dbfile { | 
					
						
							|  |  |  | 			t.Errorf("Test %d expected %s as dbfile, got %s", i, tc.exp.dbfile, x) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		if x := signer.directory; x != tc.exp.directory { | 
					
						
							|  |  |  | 			t.Errorf("Test %d expected %s as directory, got %s", i, tc.exp.directory, x) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		if x := signer.signedfile; x != tc.exp.signedfile { | 
					
						
							|  |  |  | 			t.Errorf("Test %d expected %s as signedfile, got %s", i, tc.exp.signedfile, x) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } |