testing.go

package lobster

import "github.com/gorilla/mux"

import "github.com/LunaNode/lobster/utils"

const TEST_BANDWIDTH = 1000

var testTables []string = []string{"users", "region_bandwidth", "vms", "plans", "charges", "sessions", "form_tokens", "antiflood"}

func TestReset() {
	cfg = &Config{
		Default: ConfigDefault{
			Debug: true,
		},
		Billing: ConfigBilling{
			BandwidthOverageFee: 0.003,
		},
		Database: ConfigDatabase{
			Host:     "localhost",
			Username: "lobstertest",
			Password: "",
			Name:     "lobstertest",
		},
		Novnc: ConfigNovnc{
			Listen: "127.0.0.1:6080",
			Url:    "TOKEN",
		},
	}
	db = MakeDatabase()

	// clear all tables
	for _, table := range testTables {
		db.Exec("DELETE FROM " + table)
	}
}

func TestSetup() {
	router = mux.NewRouter()
	db = MakeDatabase()
}

// Creates user and returns user id.
func TestUser() int {
	result := db.Exec("INSERT INTO users (username, password, credit) VALUES (?, '', 1000000)", utils.Uid(8))
	return result.LastInsertId()
}

func TestVm(userId int) int {
	result := db.Exec("INSERT INTO plans (name, price, ram, cpu, storage, bandwidth) VALUES ('', 6000, 512, 1, 15, ?)", TEST_BANDWIDTH)
	planId := result.LastInsertId()
	result = db.Exec("INSERT INTO vms (user_id, region, plan_id, name, status) VALUES (?, 'test', ?, '', 'active')", userId, planId)
	vmId := result.LastInsertId()
	return vmId
}