infrastructure/testhelper.go

38 lines
794 B
Go

package main;
import (
"net/http"
"io/ioutil"
"database/sql"
_ "github.com/lib/pq"
"os"
"log"
"fmt"
)
func crash(err error) {
if err != nil {
panic(err)
}
}
func main() {
db, err := sql.Open("postgres", os.Getenv("DB"))
crash(err)
http.HandleFunc("/helper/verification-code", func(w http.ResponseWriter, r *http.Request) {
account := r.PostFormValue("account")
var code string
err := db.QueryRow("SELECT verification_code FROM pending_accounts WHERE number = $1", account).Scan(&code)
crash(err)
fmt.Fprintf(w, "%s\n", code)
})
http.HandleFunc("/v1/directory/reconcile", func(w http.ResponseWriter, r *http.Request) {
b, err := ioutil.ReadAll(r.Body)
defer r.Body.Close()
crash(err)
log.Printf("%s", b)
})
log.Fatal(http.ListenAndServe(":8082", nil))
}