Merge 19e7cc2 into 9490736

.dockerignore

@@ -0,0 +1,5 @@
+Dockerfile*
+*.pyc
+build
+asyncio_apns.egg-info
+output

.travis.yml

@@ -1,14 +1,26 @@
+sudo: required
+
+services:
+  - docker
+
+env:
+  - DOCKER_PYTHON="3.5"
+
 language: python
-python:
-    - "3.4"
-    - "3.5"
+
+before_install:
+  - eval "$(gimme 1.7)"
+
 install:
-    - "pip install -r requirements-test.txt"
-    - "pip install python-coveralls"
-    - "pip install flake8"
-    - "pip install -e ."
+  - pushd tests_mocked_server && CGO_ENABLED=0 go build -v go-apns-server.go && popd
+  - sed "s/{{python_version}}/$DOCKER_PYTHON/" Dockerfile.template > Dockerfile
+  - docker build -t etataurov/asyncio_apns .
+  - pip install python-coveralls
+  - mkdir -p output
+
 script:
-    - "flake8 asyncio_apns"
-    - "py.test -v --cov=asyncio_apns"
+  - docker run --rm etataurov/asyncio_apns flake8 asyncio_apns
+  - docker run --rm -v $(pwd)/output:/app/output etataurov/asyncio_apns /bin/sh -c "py.test tests tests_mocked_server -v --cov=asyncio_apns && cp .coverage output/"
+
 after_success:
-    - "coveralls"
+  -  docker run --rm -e "TRAVIS_JOB_ID=$TRAVIS_JOB_ID" -e "TRAVIS_BRANCH=$TRAVIS_BRANCH" -v $(pwd)/output:/app/output etataurov/asyncio_apns /bin/sh -c "coveralls -d output/.coverage"

Dockerfile.template

@@ -0,0 +1,8 @@
+FROM python:{{python_version}}-alpine
+RUN apk add --no-cache git
+RUN pip install --upgrade pip
+RUN pip install --upgrade setuptools
+COPY . app
+WORKDIR app
+RUN pip install -r requirements-test.txt && pip install python-coveralls && pip install flake8
+RUN pip install -e .

setup.py

@@ -1,5 +1,5 @@
 import sys
-from distutils.core import setup
+from setuptools import setup
 
 install_requires = ["h2>=2.2.0"]
 if sys.version_info < (3, 5):

tests_mocked_server/cert.pem

@@ -0,0 +1,19 @@
+-----BEGIN CERTIFICATE-----
+MIIC/zCCAeegAwIBAgIQTytcpl1uBwcQAkJ0KDJnwjANBgkqhkiG9w0BAQsFADAS
+MRAwDgYDVQQKEwdBY21lIENvMB4XDTE2MDcxOTEwMjcxMloXDTE3MDcxOTEwMjcx
+MlowEjEQMA4GA1UEChMHQWNtZSBDbzCCASIwDQYJKoZIhvcNAQEBBQADggEPADCC
+AQoCggEBAPpI2mMcKh0kuc92RELRUZxHCOBHiVNIQqq135bUubFfkV6Wf+rwjgAD
+gk7ihKmrwDe1JeHFu4AnlJw6vvoAz25MwLLfn0tf1oE5BCKNRmeiiUjt+SSR512c
+JW7XgebbTGxW9pVDz8Un3814Z1mzyegPtRtwY3px4HdKnqWPo1F5/WZKEWlKrnMk
+vnYa5yTUh3V39q6wf0ZWVth5iKMQeZHtXw+FiSPIs2qAOvfpNoZDAdBEqtMDyjdl
+4DGkPmSO1oiUUKreOgIKl+R/Bqg93VJP+GPn4sZc7FEWctdlZdILVEwNIbMgG/uY
+ti1b4vj0gvOo+gHddcji0wVJmnvgDkECAwEAAaNRME8wDgYDVR0PAQH/BAQDAgWg
+MBMGA1UdJQQMMAoGCCsGAQUFBwMBMAwGA1UdEwEB/wQCMAAwGgYDVR0RBBMwEYIP
+Y29tcGFuZXJvcy5jbHViMA0GCSqGSIb3DQEBCwUAA4IBAQCUwwAtfLRB2G1aLSHZ
+t/dr279rRo0KKWH8JTpqpSl0scPyVPyWNuTORsDozM29b32E4o9XxsfkKhhALGv2
+0KVZVkxUtJIOLaM7ogQIci1ToQq18ap4OPjMr8iB+kZA4uZp8iYPuanv6AUjRjsl
+eRE5qieqvP0kP6ixGCYtZ6hGri/vHwVuDMIlsHFkOr5Gf+Xu3LDJuLL8lfN4Yff3
+3+WI+mjrzhNrirU8q5czyjLMLaCkTq7+OUoRjG4w+SC5HrPD9eexVa4L6I+25VBh
+HhxYS1sIyB4PtOnfSedBMjsYCH5O+RFQSF+WZIJv3oQD0pjH/1GYUe7Hz8Y9CcBE
+zMLi
+-----END CERTIFICATE-----

tests_mocked_server/go-apns-server.go

@@ -0,0 +1,59 @@
+// Copyright (c) 2016, CleverTap
+// All rights reserved.
+// Licensed under the New 3-Clause BSD License
+//
+// A mock server for the HTTP2 Apple push gateway
+
+package main
+
+import (
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"math/rand"
+	"net/http"
+	"strconv"
+)
+
+func main() {
+	serverPort := flag.Int("port", 8443, "The port for the mock server to listen to")
+	serverCert := flag.String("cert", "cert.pem", "The path to the certificate")
+	serverKey := flag.String("key", "key.pem", "The path to the certificate key")
+
+	flag.Parse()
+
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		_, err := ioutil.ReadAll(r.Body)
+		if err != nil {
+			log.Println("Error:", err)
+		}
+		n := rand.Int31n(10)
+
+		statusCode := 200
+
+		var body string
+
+		if n == 7 {
+			statusCode = 400
+			body = "{\"reason\": \"BadDeviceToken\"}"
+		} else if n > 7 {
+			statusCode = 410
+			body = "{\"reason\": \"Unregistered\"}"
+		}
+
+		w.WriteHeader(statusCode)
+
+		if statusCode != 200 {
+			w.Header().Set("Content-Type", "application/json")
+			fmt.Fprintf(w, body)
+		}
+	})
+
+	log.Println("Using certificate", *serverCert, "with key", *serverKey)
+	log.Println("Starting server on port", *serverPort)
+	log.Println("Hit ctrl + c to stop...")
+
+	port := ":" + strconv.Itoa(*serverPort)
+	log.Fatal(http.ListenAndServeTLS(port, *serverCert, *serverKey, nil))
+}

tests_mocked_server/key.pem

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEpQIBAAKCAQEA+kjaYxwqHSS5z3ZEQtFRnEcI4EeJU0hCqrXfltS5sV+RXpZ/
+6vCOAAOCTuKEqavAN7Ul4cW7gCeUnDq++gDPbkzAst+fS1/WgTkEIo1GZ6KJSO35
+JJHnXZwlbteB5ttMbFb2lUPPxSffzXhnWbPJ6A+1G3BjenHgd0qepY+jUXn9ZkoR
+aUqucyS+dhrnJNSHdXf2rrB/RlZW2HmIoxB5ke1fD4WJI8izaoA69+k2hkMB0ESq
+0wPKN2XgMaQ+ZI7WiJRQqt46AgqX5H8GqD3dUk/4Y+fixlzsURZy12Vl0gtUTA0h
+syAb+5i2LVvi+PSC86j6Ad11yOLTBUmae+AOQQIDAQABAoIBAQCCG99TiG+oKNBj
+NhL0PUZUnGTT2rxbzAPEJLnv+fPybkDu0AP9iUH4ZdqW0ZCvd3Z8PzJLNPRtAaZy
+tONUqZkruMCv/CoeD9s93/rzT0fbVhZSEEvlI+gHtMWM4oFIFpaa93ZGUDMxRbJF
+aSMo8ZqzhX7bnchI/ev3d5YTcSEwl9fi+DuTftXhRDS6KR2QvvBQPTsZEYFXHKvv
+9DJ431tCMUWNJwAEQq22iNojOWiq10NoXR7UiUMi/yhBbtvuzJyAu7wwwyGoRLRH
+4T/wof8Rc/zc8SuvizFsCfVhNe5Y/rTDc87U7eWD14Ihc6evKi+pqeV+kpGVJbhn
+zfauz3JlAoGBAP1cftClWv0N2xCiSRwBp1qwT259rpNZkMIFu8s/ipd2jffE87RS
+lCxfHKTf+oJq5fQrbwfxly4qsagzlwqKM4rLlh/tC5NfEj+Bhr85yZeEovVjJiwi
+2lrZXWhzYdvL/1MSJv3wC6SEyqYLFjD9xfQqogob5Qnt1Fab90kCKTMjAoGBAPzk
+J5U6AsDpYO2JiSI0fil+HYda6NI6tWcplr7gRFI4u+mJ4T1eB8fntwxu9wHr8xv3
++3YJ4yfLy2AfABR/uWak1Ua/KlxPnst78WPQNsOUYUJTSNQQ+i3OOwKJf+jM4PU9
+aUwZm4mz8o88eco3OJNPVqsvb5ZV27dVYjrG4FFLAoGAak8kWxH3EtV8hQdDXAAJ
+o9D1E2yjEurzBaZ4PPlFgvZY7YzzoUdBKzkhgMEGkTCfmoubBB9ICzQY5XX4O5Dv
+vZ5dOLozMDwlIQaNbrGlt/EMm6BlxQE+0/3+5rP7bMm1MbsdJ1JQ4FBj47YlMptZ
+M3GsGkfIHBlyDoFVCopglTcCgYEAlr4FHj2DEKzMWytFc/8oSOFlBp7Y8x+NrB2t
+Pr/92aj50f6fJW5cYvurMB0g3aNguYY6/SRto/Yll3lDGfT8dcPBYbdnr7jCJfcO
+b3nYewKCav+sBMTN6+nnYN7HQOSHH4fosTT+crmx11GsIy6BacW8GrRe5pTtMD4U
+HIcZhTsCgYEAwbbSOm6hhkxGIV8cz5IdYw4xxj6Fw2kZc178dhVfeWX4bOHcRH6H
+yS//fD94lnAgV3punW1BibcK3nAxMBN2sLIWIacor41USJGp2TWe2cREqOD+3GlQ
+170PWseGrQiDm/EfkjEfcJTB6PePe9y0QMvarnzafHNTjlQAGCc5zeo=
+-----END RSA PRIVATE KEY-----

tests_mocked_server/test_load.py

@@ -0,0 +1,41 @@
+import asyncio
+import os
+import subprocess
+import time
+
+import pytest
+from asyncio_apns import APNsConnection, APNsError
+
+CWD = os.path.dirname(os.path.realpath(__file__))
+
+
+@pytest.yield_fixture(scope="module")
+def mock_server(request):
+    server = subprocess.Popen("./go-apns-server", shell=True, cwd=CWD)
+    time.sleep(0.5)
+    yield server
+    server.terminate()
+
+
+@pytest.mark.skipif(not os.path.exists(os.path.join(CWD, 'go-apns-server')),
+                    reason="No binary for test server")
+@pytest.mark.asyncio
+def test_batch_send_messages(event_loop, mock_server):
+    connection = APNsConnection(os.path.join(CWD, "cert.pem"), os.path.join(CWD, "key.pem"),
+                                loop=event_loop, server_addr="127.0.0.1", server_port=8443)
+    yield from connection.connect()
+
+    @asyncio.coroutine
+    def send_message():
+        token = "fsd76fh23jhd6s2j2h"
+        message = "Hello my world"
+        try:
+            yield from connection.send_message(message, token=token)
+        except APNsError as exc:
+            assert exc.status in ("BadDeviceToken", "Unregistered")
+
+    tasks = []
+    for _ in range(1000):
+        tasks.append(asyncio.async(send_message(), loop=event_loop))
+    yield from asyncio.wait(tasks, loop=event_loop)
+    connection.disconnect()