226 lines
8.6 KiB
Go
226 lines
8.6 KiB
Go
package playstore
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
type testSignature struct {
|
|
PrivateKeyID string `json:"private_key_id"`
|
|
PrivateKey string `json:"private_key"`
|
|
ClientEmail string `json:"client_email"`
|
|
ClientID string `json:"client_id"`
|
|
Type string `json:"type"`
|
|
}
|
|
|
|
var testJSON = testSignature{
|
|
PrivateKeyID: "dummyKeyID",
|
|
PrivateKey: "-----BEGIN PRIVATE KEY-----\nMIIBOQIBAAJBANXOa7wgs5KHMEVJmVo2eoRxEgeqiYF2oABPGYrebU+cQiE7Mwdy\nxv153DHME+9L9QzAj+fR4y5Rwva/fAsGAssCAwEAAQJATQwrFMtwCtC+22kvYywY\nsJuSlMKm9MmL1TCsErgfCj2rksRK1U+/ZY709tE3XJVYlZalWCeVhHTjs5p0pnk6\nYQIhAOw0FksytfIfpdfcREbful+LhFp1um5WjcVf7kQ73JDxAiEA57nJkG9pwnUd\nBCyIcElTVIAKU0+iFpd1208OnGxyT3sCIGaEBNkGXWmEytnxQ8DvAVjOmNcaGZwh\n/M4ZYLREtupBAiAsrpFkTWdqPKTcsi2Y4Tq1N39GMzvA+XGbWTIrDWo5UwIgHhp9\nEOnHuUuPCjpLfYM2vSFiYzaj8UJCImjkMtDwzbA=\n-----END PRIVATE KEY-----\n",
|
|
ClientEmail: "dummyEmail",
|
|
ClientID: "dummyClientID",
|
|
Type: "service_account",
|
|
}
|
|
|
|
func TestNew(t *testing.T) {
|
|
// Exception scenario
|
|
jsonKey, _ := json.Marshal(testJSON)
|
|
expected := "oauth2: cannot fetch token: 400 Bad Request\nResponse: {\n \"error\" : \"invalid_grant\"\n}"
|
|
|
|
actual, _ := New(jsonKey)
|
|
val := actual.httpClient.Transport.(*oauth2.Transport)
|
|
token, err := val.Source.Token()
|
|
if token != nil {
|
|
t.Errorf("got %#v", token)
|
|
}
|
|
if err.Error() != expected {
|
|
t.Errorf("got %v\nwant %v", err, expected)
|
|
}
|
|
|
|
// TODO Normal scenario
|
|
}
|
|
|
|
func TestSetTimeout(t *testing.T) {
|
|
_timeout := time.Second * 3
|
|
SetTimeout(_timeout)
|
|
|
|
if timeout != _timeout {
|
|
t.Errorf("got %#v\nwant %#v", timeout, _timeout)
|
|
}
|
|
}
|
|
|
|
func TestVerifySubscription(t *testing.T) {
|
|
// Exception scenario
|
|
jsonKey, _ := json.Marshal(testJSON)
|
|
|
|
expected := "Get https://www.googleapis.com/androidpublisher/v2/applications/package/purchases/subscriptions/subscriptionID/tokens/purchaseToken?alt=json: oauth2: cannot fetch token: 400 Bad Request\nResponse: {\n \"error\" : \"invalid_grant\"\n}"
|
|
|
|
client, _ := New(jsonKey)
|
|
_, err := client.VerifySubscription("package", "subscriptionID", "purchaseToken")
|
|
|
|
if err.Error() != expected {
|
|
t.Errorf("got %v\nwant %v", err, expected)
|
|
}
|
|
|
|
// TODO Normal scenario
|
|
}
|
|
|
|
func TestVerifySubscriptionAndroidPublisherError(t *testing.T) {
|
|
client := Client{nil}
|
|
expected := errors.New("client is nil")
|
|
_, actual := client.VerifySubscription("package", "subscriptionID", "purchaseToken")
|
|
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
t.Errorf("got %v\nwant %v", actual, expected)
|
|
}
|
|
}
|
|
|
|
func TestVerifyProduct(t *testing.T) {
|
|
// Exception scenario
|
|
jsonKey, _ := json.Marshal(testJSON)
|
|
|
|
expected := "Get https://www.googleapis.com/androidpublisher/v2/applications/package/purchases/products/productID/tokens/purchaseToken?alt=json: oauth2: cannot fetch token: 400 Bad Request\nResponse: {\n \"error\" : \"invalid_grant\"\n}"
|
|
|
|
client, _ := New(jsonKey)
|
|
_, err := client.VerifyProduct("package", "productID", "purchaseToken")
|
|
|
|
if err.Error() != expected {
|
|
t.Errorf("got %v", err)
|
|
}
|
|
|
|
// TODO Normal scenario
|
|
}
|
|
|
|
func TestVerifyProductAndroidPublisherError(t *testing.T) {
|
|
client := Client{nil}
|
|
expected := errors.New("client is nil")
|
|
_, actual := client.VerifyProduct("package", "productID", "purchaseToken")
|
|
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
t.Errorf("got %v\nwant %v", actual, expected)
|
|
}
|
|
}
|
|
|
|
func TestCancelSubscription(t *testing.T) {
|
|
// Exception scenario
|
|
client := Client{nil}
|
|
expected := errors.New("client is nil")
|
|
actual := client.CancelSubscription("package", "productID", "purchaseToken")
|
|
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
t.Errorf("got %v\nwant %v", actual, expected)
|
|
}
|
|
|
|
jsonKey, _ := json.Marshal(testJSON)
|
|
client, _ = New(jsonKey)
|
|
expectedStr := "Post https://www.googleapis.com/androidpublisher/v2/applications/package/purchases/subscriptions/productID/tokens/purchaseToken:cancel?alt=json: oauth2: cannot fetch token: 400 Bad Request\nResponse: {\n \"error\" : \"invalid_grant\"\n}"
|
|
actual = client.CancelSubscription("package", "productID", "purchaseToken")
|
|
|
|
if actual.Error() != expectedStr {
|
|
t.Errorf("got %v\nwant %v", actual, expectedStr)
|
|
}
|
|
|
|
// TODO Normal scenario
|
|
}
|
|
|
|
func TestRefundSubscription(t *testing.T) {
|
|
// Exception scenario
|
|
client := Client{nil}
|
|
expected := errors.New("client is nil")
|
|
actual := client.RefundSubscription("package", "productID", "purchaseToken")
|
|
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
t.Errorf("got %v\nwant %v", actual, expected)
|
|
}
|
|
|
|
jsonKey, _ := json.Marshal(testJSON)
|
|
client, _ = New(jsonKey)
|
|
expectedStr := "Post https://www.googleapis.com/androidpublisher/v2/applications/package/purchases/subscriptions/productID/tokens/purchaseToken:refund?alt=json: oauth2: cannot fetch token: 400 Bad Request\nResponse: {\n \"error\" : \"invalid_grant\"\n}"
|
|
actual = client.RefundSubscription("package", "productID", "purchaseToken")
|
|
|
|
if actual.Error() != expectedStr {
|
|
t.Errorf("got %v\nwant %v", actual, expectedStr)
|
|
}
|
|
|
|
// TODO Normal scenario
|
|
}
|
|
|
|
func TestRevokeSubscription(t *testing.T) {
|
|
// Exception scenario
|
|
client := Client{nil}
|
|
expected := errors.New("client is nil")
|
|
actual := client.RevokeSubscription("package", "productID", "purchaseToken")
|
|
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
t.Errorf("got %v\nwant %v", actual, expected)
|
|
}
|
|
|
|
jsonKey, _ := json.Marshal(testJSON)
|
|
client, _ = New(jsonKey)
|
|
expectedStr := "Post https://www.googleapis.com/androidpublisher/v2/applications/package/purchases/subscriptions/productID/tokens/purchaseToken:revoke?alt=json: oauth2: cannot fetch token: 400 Bad Request\nResponse: {\n \"error\" : \"invalid_grant\"\n}"
|
|
actual = client.RevokeSubscription("package", "productID", "purchaseToken")
|
|
|
|
if actual.Error() != expectedStr {
|
|
t.Errorf("got %v\nwant %v", actual, expectedStr)
|
|
}
|
|
|
|
// TODO Normal scenario
|
|
}
|
|
|
|
func TestVerifySignature(t *testing.T) {
|
|
receipt := `{"orderId":"GPA.xxxx-xxxx-xxxx-xxxxx","packageName":"my.package","productId":"myproduct","purchaseTime":1437564796303,"purchaseState":0,"developerPayload":"user001","purchaseToken":"some-token"}`
|
|
|
|
// when public key format is invalid base64
|
|
pubkey := "dummy_public_key"
|
|
sig := "gj0N8LANKXOw4OhWkS1UZmDVUxM1UIP28F6bDzEp7BCqcVAe0DuDxmAY5wXdEgMRx/VM1Nl2crjogeV60OqCsbIaWqS/ZJwdP127aKR0jk8sbX36ssyYZ0DdZdBdCr1tBZ/eSW1GlGuD/CgVaxns0JaWecXakgoV7j+RF2AFbS4="
|
|
expectedStr := "failed to decode public key"
|
|
_, err := VerifySignature(pubkey, []byte(receipt), sig)
|
|
if err.Error() != expectedStr {
|
|
t.Errorf("got %v\nwant %v", err, expectedStr)
|
|
}
|
|
|
|
// when pub key is not rsa public key
|
|
pubkey = "JTbngOdvBE0rfdOs3GeuBnPB+YEP1w/peM4VJbnVz+hN9Td25vPjAznX9YKTGQN4iDohZ07wtl+zYygIcpSCc2ozNZUs9pV0s5itayQo22aT5myJrQmkp94ZSGI2npDP4+FE6ZiF+7khl3qoE0rVZq4G2mfk5LIIyTPTSA4UvyQ="
|
|
sig = "gj0N8LANKXOw4OhWkS1UZmDVUxM1UIP28F6bDzEp7BCqcVAe0DuDxmAY5wXdEgMRx/VM1Nl2crjogeV60OqCsbIaWqS/ZJwdP127aKR0jk8sbX36ssyYZ0DdZdBdCr1tBZ/eSW1GlGuD/CgVaxns0JaWecXakgoV7j+RF2AFbS4="
|
|
expectedStr = "failed to parse public key"
|
|
_, err = VerifySignature(pubkey, []byte(receipt), sig)
|
|
if err.Error() != expectedStr {
|
|
t.Errorf("got %v\nwant %v", err, expectedStr)
|
|
}
|
|
|
|
// when signature is invalid base64 format
|
|
pubkey = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDGvModvVUrqJ9C5fy8J77ZQ7JDC6+tf5iK8C74/3mjmcvwo4nmprCgzR/BQIEuZWJi8KX+jiJUXKXF90JPsXHkKAPq6A1SCga7kWvs/M8srMpjNS9zJdwZF+eDOR0+lJEihO04zlpAV9ybPJ3Q621y1HUeVpwdxDNLQpJTuIflnwIDAQAB"
|
|
sig = "invalid_signature"
|
|
expectedStr = "failed to decode signature"
|
|
_, err = VerifySignature(pubkey, []byte(receipt), sig)
|
|
if err.Error() != expectedStr {
|
|
t.Errorf("got %v\nwant %v", err, expectedStr)
|
|
}
|
|
|
|
// when signature is invalid
|
|
pubkey = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDGvModvVUrqJ9C5fy8J77ZQ7JDC6+tf5iK8C74/3mjmcvwo4nmprCgzR/BQIEuZWJi8KX+jiJUXKXF90JPsXHkKAPq6A1SCga7kWvs/M8srMpjNS9zJdwZF+eDOR0+lJEihO04zlpAV9ybPJ3Q621y1HUeVpwdxDNLQpJTuIflnwIDAQAB"
|
|
sig = "JTbngOdvBE0rfdOs3GeuBnPB+YEP1w/peM4VJbnVz+hN9Td25vPjAznX9YKTGQN4iDohZ07wtl+zYygIcpSCc2ozNZUs9pV0s5itayQo22aT5myJrQmkp94ZSGI2npDP4+FE6ZiF+7khl3qoE0rVZq4G2mfk5LIIyTPTSA4UvyQ="
|
|
isValid, err := VerifySignature(pubkey, []byte(receipt), sig)
|
|
if err != nil {
|
|
t.Errorf("got %v\n", err)
|
|
}
|
|
if isValid {
|
|
t.Errorf("got %v\nwant %v", isValid, false)
|
|
}
|
|
|
|
// when all arguments are valid
|
|
pubkey = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDGvModvVUrqJ9C5fy8J77ZQ7JDC6+tf5iK8C74/3mjmcvwo4nmprCgzR/BQIEuZWJi8KX+jiJUXKXF90JPsXHkKAPq6A1SCga7kWvs/M8srMpjNS9zJdwZF+eDOR0+lJEihO04zlpAV9ybPJ3Q621y1HUeVpwdxDNLQpJTuIflnwIDAQAB"
|
|
sig = "gj0N8LANKXOw4OhWkS1UZmDVUxM1UIP28F6bDzEp7BCqcVAe0DuDxmAY5wXdEgMRx/VM1Nl2crjogeV60OqCsbIaWqS/ZJwdP127aKR0jk8sbX36ssyYZ0DdZdBdCr1tBZ/eSW1GlGuD/CgVaxns0JaWecXakgoV7j+RF2AFbS4="
|
|
isValid, err = VerifySignature(pubkey, []byte(receipt), sig)
|
|
if err != nil {
|
|
t.Errorf("got %v\n", err)
|
|
}
|
|
if !isValid {
|
|
t.Errorf("got %v\nwant %v", isValid, true)
|
|
}
|
|
}
|