// Copyright (c) 2013, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
//
// VMOptions=
// VMOptions=--short_socket_read
// VMOptions=--short_socket_write
// VMOptions=--short_socket_read --short_socket_write

import 'package:expect/expect.dart';
import 'dart:io';
import 'dart:typed_data';

void testServerCompress({bool clientAutoUncompress: true}) {
  void test(List<int> data) {
    HttpServer.bind("127.0.0.1", 0).then((server) {
      server.autoCompress = true;
      server.listen((request) {
        request.response.add(data);
        request.response.close();
      });
      var client = new HttpClient();
      client.autoUncompress = clientAutoUncompress;
      client.get("127.0.0.1", server.port, "/").then((request) {
        request.headers.set(HttpHeaders.ACCEPT_ENCODING, "gzip,deflate");
        return request.close();
      }).then((response) {
        Expect.equals(
            "gzip", response.headers.value(HttpHeaders.CONTENT_ENCODING));
        response.fold([], (list, b) {
          list.addAll(b);
          return list;
        }).then((list) {
          if (clientAutoUncompress) {
            Expect.listEquals(data, list);
          } else {
            Expect.listEquals(data, GZIP.decode(list));
          }
          server.close();
          client.close();
        });
      });
    });
  }

  test("My raw server provided data".codeUnits);
  var longBuffer = new Uint8List(1024 * 1024);
  for (int i = 0; i < longBuffer.length; i++) {
    longBuffer[i] = i & 0xFF;
  }
  test(longBuffer);
}

void testAcceptEncodingHeader() {
  void test(String encoding, bool valid) {
    HttpServer.bind("127.0.0.1", 0).then((server) {
      server.autoCompress = true;
      server.listen((request) {
        request.response.write("data");
        request.response.close();
      });
      var client = new HttpClient();
      client.get("127.0.0.1", server.port, "/").then((request) {
        request.headers.set(HttpHeaders.ACCEPT_ENCODING, encoding);
        return request.close();
      }).then((response) {
        Expect.equals(valid,
            ("gzip" == response.headers.value(HttpHeaders.CONTENT_ENCODING)));
        response.listen((_) {}, onDone: () {
          server.close();
          client.close();
        });
      });
    });
  }

  test('gzip', true);
  test('deflate', false);
  test('gzip, deflate', true);
  test('gzip ,deflate', true);
  test('gzip  ,  deflate', true);
  test('deflate,gzip', true);
  test('deflate, gzip', true);
  test('deflate ,gzip', true);
  test('deflate  ,  gzip', true);
  test('abc,deflate  ,  gzip,def,,,ghi  ,jkl', true);
  test('xgzip', false);
  test('gzipx;', false);
}

void testDisableCompressTest() {
  HttpServer.bind("127.0.0.1", 0).then((server) {
    Expect.equals(false, server.autoCompress);
    server.listen((request) {
      Expect.equals('gzip', request.headers.value(HttpHeaders.ACCEPT_ENCODING));
      request.response.write("data");
      request.response.close();
    });
    var client = new HttpClient();
    client
        .get("127.0.0.1", server.port, "/")
        .then((request) => request.close())
        .then((response) {
      Expect.equals(null, response.headers.value(HttpHeaders.CONTENT_ENCODING));
      response.listen((_) {}, onDone: () {
        server.close();
        client.close();
      });
    });
  });
}

void main() {
  testServerCompress();
  testServerCompress(clientAutoUncompress: false);
  testAcceptEncodingHeader();
  testDisableCompressTest();
}