flip/dcrd
1
0
Fork 0
mirror of https://github.com/FlipsideCrypto/dcrd.git synced 2026-02-06 10:56:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

main: move cert tests to a separated file

Browse Source
This commit is contained in:
J Fixby 2018-10-18 19:50:49 +02:00
parent 52f331516a
commit 4896240914
2 changed files with 79 additions and 69 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
cert_test.go Normal file
View File

@ -0,0 +1,79 @@
// Copyright (c) 2018 The Decred developers
// Use of this source code is governed by an ISC
// license that can be found in the LICENSE file.
package main
import (
"crypto/x509"
"encoding/pem"
"io/ioutil"
"os"
"testing"
)
// TestCertCreationWithHosts creates a certificate pair with extra hosts and
// ensures the extra hosts are present in the generated files.
func TestCertCreationWithHosts(t *testing.T) {
certFile, err := ioutil.TempFile("", "certfile")
if err != nil {
t.Fatalf("Unable to create temp certfile: %s", err)
}
certFile.Close()
defer os.Remove(certFile.Name())
keyFile, err := ioutil.TempFile("", "keyfile")
if err != nil {
t.Fatalf("Unable to create temp keyfile: %s", err)
}
keyFile.Close()
defer os.Remove(keyFile.Name())
// Generate cert pair with extra hosts.
hostnames := []string{"hostname1", "hostname2"}
err = genCertPair(certFile.Name(), keyFile.Name(), hostnames)
if err != nil {
t.Fatalf("Certificate was not created correctly: %s", err)
}
certBytes, err := ioutil.ReadFile(certFile.Name())
if err != nil {
t.Fatalf("Unable to read the certfile: %s", err)
}
pemCert, _ := pem.Decode(certBytes)
x509Cert, err := x509.ParseCertificate(pemCert.Bytes)
if err != nil {
t.Fatalf("Unable to parse the certificate: %s", err)
}
// Ensure the specified extra hosts are present.
for _, host := range hostnames {
err := x509Cert.VerifyHostname(host)
if err != nil {
t.Fatalf("failed to verify extra host '%s'", host)
}
}
}
// TestCertCreationWithOutHosts ensures the creating a certificate pair without
// any hosts works as intended.
func TestCertCreationWithOutHosts(t *testing.T) {
certFile, err := ioutil.TempFile("", "certfile")
if err != nil {
t.Fatalf("Unable to create temp certfile: %s", err)
}
certFile.Close()
defer os.Remove(certFile.Name())
keyFile, err := ioutil.TempFile("", "keyfile")
if err != nil {
t.Fatalf("Unable to create temp keyfile: %s", err)
}
keyFile.Close()
defer os.Remove(keyFile.Name())
// Generate cert pair with no extra hosts.
err = genCertPair(certFile.Name(), keyFile.Name(), nil)
if err != nil {
t.Fatalf("Certificate was not created correctly: %s", err)
}
}

69
rpcserver_test.go
View File

@ -10,10 +10,7 @@ package main
import (
"bytes"
"crypto/x509"
"encoding/pem"
"fmt"
"io/ioutil"
"os"
"runtime/debug"
"testing"
@ -167,69 +164,3 @@ func TestRpcServer(t *testing.T) {
currentTestNum++
}
}
// TestCertCreationWithHosts creates a certificate pair with extra hosts and
// ensures the extra hosts are present in the generated files.
func TestCertCreationWithHosts(t *testing.T) {
certFile, err := ioutil.TempFile("", "certfile")
if err != nil {
t.Fatalf("Unable to create temp certfile: %s", err)
}
certFile.Close()
defer os.Remove(certFile.Name())
keyFile, err := ioutil.TempFile("", "keyfile")
if err != nil {
t.Fatalf("Unable to create temp keyfile: %s", err)
}
keyFile.Close()
defer os.Remove(keyFile.Name())
// Generate cert pair with extra hosts.
hostnames := []string{"hostname1", "hostname2"}
err = genCertPair(certFile.Name(), keyFile.Name(), hostnames)
if err != nil {
t.Fatalf("Certificate was not created correctly: %s", err)
}
certBytes, err := ioutil.ReadFile(certFile.Name())
if err != nil {
t.Fatalf("Unable to read the certfile: %s", err)
}
pemCert, _ := pem.Decode(certBytes)
x509Cert, err := x509.ParseCertificate(pemCert.Bytes)
if err != nil {
t.Fatalf("Unable to parse the certificate: %s", err)
}
// Ensure the specified extra hosts are present.
for _, host := range hostnames {
err := x509Cert.VerifyHostname(host)
if err != nil {
t.Fatalf("failed to verify extra host '%s'", host)
}
}
}
// TestCertCreationWithOutHosts ensures the creating a certificate pair without
// any hosts works as intended.
func TestCertCreationWithOutHosts(t *testing.T) {
certFile, err := ioutil.TempFile("", "certfile")
if err != nil {
t.Fatalf("Unable to create temp certfile: %s", err)
}
certFile.Close()
defer os.Remove(certFile.Name())
keyFile, err := ioutil.TempFile("", "keyfile")
if err != nil {
t.Fatalf("Unable to create temp keyfile: %s", err)
}
keyFile.Close()
defer os.Remove(keyFile.Name())
// Generate cert pair with no extra hosts.
err = genCertPair(certFile.Name(), keyFile.Name(), nil)
if err != nil {
t.Fatalf("Certificate was not created correctly: %s", err)
}
}