import 'headers.dart';

/// Headers that can be locked to editing, i.e. after a request body has been written.
class LockableMockHttpHeaders extends MockHttpHeaders {
  bool _locked = false;

  StateError _stateError() =>
      StateError('Cannot modify headers after they have been write-locked.');

  void lock() {
    _locked = true;
  }

  @override
  void add(String name, Object value, {bool preserveHeaderCase = false}) {
    if (_locked) {
      throw _stateError();
    } else {
      super.add(name, value, preserveHeaderCase: preserveHeaderCase);
    }
  }

  @override
  void clear() {
    if (_locked) {
      throw _stateError();
    } else {
      super.clear();
    }
  }

  @override
  void noFolding(String name) {
    if (_locked) {
      throw _stateError();
    } else {
      super.noFolding(name);
    }
  }

  @override
  void remove(String name, Object value) {
    if (_locked) {
      throw _stateError();
    } else {
      super.remove(name, value);
    }
  }

  @override
  void removeAll(String name) {
    if (_locked) {
      throw _stateError();
    } else {
      super.removeAll(name);
    }
  }

  @override
  void set(String name, Object value, {bool preserveHeaderCase = false}) {
    if (_locked) {
      throw _stateError();
    } else {
      super.set(name, value, preserveHeaderCase: preserveHeaderCase);
    }
  }
}