Fix PiHole AAAA tests

Signed-off-by: PseudoResonance <kaio11604@gmail.com>
This commit is contained in:
PseudoResonance 2024-04-26 06:20:14 -07:00
parent 332f679cc0
commit b4d76a9d94
No known key found for this signature in database
GPG Key ID: D2A8677B0898FB8A

View File

@ -242,7 +242,7 @@ func TestProvider(t *testing.T) {
t.Error("Unexpected delete request, got:", requests.deleteRequests[0], "expected:", recordToDeleteA) t.Error("Unexpected delete request, got:", requests.deleteRequests[0], "expected:", recordToDeleteA)
} }
if !reflect.DeepEqual(requests.deleteRequests[1], &recordToDeleteAAAA) { if !reflect.DeepEqual(requests.deleteRequests[1], &recordToDeleteAAAA) {
t.Error("Unexpected delete request, got:", requests.deleteRequests[0], "expected:", recordToDeleteAAAA) t.Error("Unexpected delete request, got:", requests.deleteRequests[1], "expected:", recordToDeleteAAAA)
} }
requests.clear() requests.clear()
@ -326,7 +326,7 @@ func TestProvider(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if len(newRecords) != 4 { if len(newRecords) != 4 {
t.Fatal("Expected list of 4 records, got:", records) t.Fatal("Expected list of 4 records, got:", newRecords)
} }
if len(requests.createRequests) != 2 { if len(requests.createRequests) != 2 {
t.Fatal("Expected 2 create request, got:", requests.createRequests) t.Fatal("Expected 2 create request, got:", requests.createRequests)
@ -365,17 +365,32 @@ func TestProvider(t *testing.T) {
RecordType: endpoint.RecordTypeAAAA, RecordType: endpoint.RecordTypeAAAA,
} }
if !reflect.DeepEqual(requests.createRequests[0], &expectedCreateA) { for _, request := range requests.createRequests {
t.Error("Unexpected create request, got:", requests.createRequests[0], "expected:", &expectedCreateA) switch request.RecordType {
case endpoint.RecordTypeA:
if !reflect.DeepEqual(request, &expectedCreateA) {
t.Error("Unexpected create request, got:", request, "expected:", &expectedCreateA)
}
case endpoint.RecordTypeAAAA:
if !reflect.DeepEqual(request, &expectedCreateAAAA) {
t.Error("Unexpected create request, got:", request, "expected:", &expectedCreateAAAA)
}
default:
}
} }
if !reflect.DeepEqual(requests.deleteRequests[0], &expectedDeleteA) {
t.Error("Unexpected delete request, got:", requests.deleteRequests[0], "expected:", &expectedDeleteA) for _, request := range requests.deleteRequests {
} switch request.RecordType {
if !reflect.DeepEqual(requests.createRequests[1], &expectedCreateAAAA) { case endpoint.RecordTypeA:
t.Error("Unexpected create request, got:", requests.createRequests[0], "expected:", &expectedCreateAAAA) if !reflect.DeepEqual(request, &expectedDeleteA) {
} t.Error("Unexpected delete request, got:", request, "expected:", &expectedDeleteA)
if !reflect.DeepEqual(requests.deleteRequests[1], &expectedDeleteAAAA) { }
t.Error("Unexpected delete request, got:", requests.deleteRequests[0], "expected:", &expectedDeleteAAAA) case endpoint.RecordTypeAAAA:
if !reflect.DeepEqual(request, &expectedDeleteAAAA) {
t.Error("Unexpected delete request, got:", request, "expected:", &expectedDeleteAAAA)
}
default:
}
} }
requests.clear() requests.clear()