scsusers/main_test.go

53 lines
1.3 KiB
Go

package scsusers
import (
"flag"
"fmt"
"os"
"testing"
"github.com/jmoiron/sqlx"
_ "github.com/mattn/go-sqlite3"
)
func TestUsers(t *testing.T) {
var email string
generatePassword(16)
flag.StringVar(&email, "email", "", "Email address to use for registration test")
flag.Parse()
db, err := sqlx.Open("sqlite3", ":memory:")
if err != nil {
fmt.Println("Couldn't open sqlite3 in-memory db:" + err.Error())
os.Exit(1)
}
err = db.Ping()
if err != nil {
fmt.Println("Couldn't ping sqlite3 in-memory db:" + err.Error())
os.Exit(1)
}
schema := `CREATE TABLE test_users (
id integer primary key autoincrement,
username text NOT NULL unique,
password text NOT NULL);`
_, err = db.Exec(schema)
if err != nil {
fmt.Println("Schema creation failed: " + err.Error())
os.Exit(1)
}
Init(db, "test", "Example Test", "nobody@nowhere.com", "localhost:25", true)
a := UsernameAvailable("testuser")
fmt.Printf("Initial test of username available: %v\n", a)
a = Register("testuser", email, "127.0.0.1")
fmt.Printf("Register returned %v\n", a)
fmt.Printf("Attempt to log in with invalid username returned %v\n", Login("baduser", "badpass"))
fmt.Printf("Enter code from email:")
var code string
fmt.Scan(&code)
a = Login("testuser", code)
fmt.Printf("Login returned %v\n", a)
}