Files
container/Tests/DNSServerTests/HostTableResolverTest.swift
T
Kathryn Baldauf d6f052d206 Update license header on all files to include the current year (#1024)
## Motivation and Context
Now that we're in 2026, we need to update the license headers on all the
files. Unfortunately, Hawkeye doesn't have an attribute for the current
year to help us avoid this in the future. Instead, I had to work around
this by doing the following:

1. Update licenserc.toml with:
     ```
      [properties]
       ... (other properties)
       currentYear = "2026"
     ```
 
2. Update scripts/license-header.txt with
    ```
Copyright ©{{ " " }}{%- set created = attrs.git_file_created_year or
attrs.disk_file_created_year -%}{%- set modified = props["currentYear"]
-%}{%- if created != modified -%} {{created}}-{{modified}}{%- else
-%}{{created}}{%- endif -%}{{ " " }}{{ props["copyrightOwner"] }}.
    ```

Then I removed these two changes before committing. After this PR is
merged, all files will have recently had git updates, so the existing
code for setting the modified year should work as intended.

Signed-off-by: Kathryn Baldauf <k_baldauf@apple.com>
2026-01-05 13:09:34 -08:00

132 lines
4.6 KiB
Swift

//===----------------------------------------------------------------------===//
// Copyright © 2025-2026 Apple Inc. and the container project authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//===----------------------------------------------------------------------===//
import DNS
import Testing
@testable import DNSServer
struct HostTableResolverTest {
@Test func testUnsupportedQuestionType() async throws {
guard let ip = IPv4("1.2.3.4") else {
throw DNSResolverError.serverError("cannot create IP address in test")
}
let handler = HostTableResolver(hosts4: ["foo": ip])
let query = Message(
id: UInt16(1),
type: .query,
questions: [
Question(name: "foo", type: .mailExchange)
])
let response = try await handler.answer(query: query)
#expect(.notImplemented == response?.returnCode)
#expect(1 == response?.id)
#expect(.response == response?.type)
#expect(1 == response?.questions.count)
#expect(0 == response?.answers.count)
}
@Test func testAAAAQueryReturnsNoDataWhenARecordExists() async throws {
guard let ip = IPv4("1.2.3.4") else {
throw DNSResolverError.serverError("cannot create IP address in test")
}
let handler = HostTableResolver(hosts4: ["foo": ip])
let query = Message(
id: UInt16(1),
type: .query,
questions: [
Question(name: "foo", type: .host6)
])
let response = try await handler.answer(query: query)
// AAAA queries should return NODATA (noError with empty answers) when A record exists
// to avoid musl libc issues where NXDOMAIN causes complete DNS resolution failure
#expect(.noError == response?.returnCode)
#expect(1 == response?.id)
#expect(.response == response?.type)
#expect(1 == response?.questions.count)
#expect(0 == response?.answers.count)
}
@Test func testAAAAQueryReturnsNilWhenHostDoesNotExist() async throws {
guard let ip = IPv4("1.2.3.4") else {
throw DNSResolverError.serverError("cannot create IP address in test")
}
let handler = HostTableResolver(hosts4: ["foo": ip])
let query = Message(
id: UInt16(1),
type: .query,
questions: [
Question(name: "bar", type: .host6)
])
let response = try await handler.answer(query: query)
// AAAA queries for non-existent hosts should return nil (which becomes NXDOMAIN)
#expect(nil == response)
}
@Test func testHostNotPresent() async throws {
guard let ip = IPv4("1.2.3.4") else {
throw DNSResolverError.serverError("cannot create IP address in test")
}
let handler = HostTableResolver(hosts4: ["foo": ip])
let query = Message(
id: UInt16(1),
type: .query,
questions: [
Question(name: "bar", type: .host)
])
let response = try await handler.answer(query: query)
#expect(nil == response)
}
@Test func testHostPresent() async throws {
guard let ip = IPv4("1.2.3.4") else {
throw DNSResolverError.serverError("cannot create IP address in test")
}
let handler = HostTableResolver(hosts4: ["foo": ip])
let query = Message(
id: UInt16(1),
type: .query,
questions: [
Question(name: "foo", type: .host)
])
let response = try await handler.answer(query: query)
#expect(.noError == response?.returnCode)
#expect(1 == response?.id)
#expect(.response == response?.type)
#expect(1 == response?.questions.count)
#expect("foo" == response?.questions[0].name)
#expect(.host == response?.questions[0].type)
#expect(1 == response?.answers.count)
let answer = response?.answers[0] as? HostRecord<IPv4>
#expect(IPv4("1.2.3.4") == answer?.ip)
}
}