database.mod/sqlite.mod/tests/auth_test_01.bmx

SuperStrict

Framework Database.SQLite
Import BRL.filesystem
Import BRL.StandardIO

DeleteFile("autotest.db")

Local db:TDBConnection = LoadDatabase("SQLITE", "autotest.db")

If Not db Then
	Print "Problem initialising database environment."
	End
End If

If db.hasError() Then
	errorAndClose(db)
End If

If db.isOpen() Then

	' create a table
	Local s:String = "CREATE TABLE person (id integer primary key AUTOINCREMENT, " + ..
	  " forename varchar(30)," + ..
	  " surname varchar(30) )"

	db.executeQuery(s)
	
	
	' insert some data
	Local names:String[][] = [ ..
	[ "Alfred", "Aho" ],   ..
	[ "Brian", "Kernighan" ], ..
	[ "Peter", "Weinberger" ] ]

	For Local i:Int = 0 Until names.length
		db.executeQuery("INSERT INTO person values (NULL, '" + names[i][0] + "', '" + names[i][1] + "')")
		If db.hasError() Then
			errorAndClose(db)
		End If
	Next


	' add a user - this will enable authentication
	' cast to SQLite to access SQLite-only functionality
	Local sqliteDB:TDBSQLite = TDBSQLite(db)

	sqliteDB.addUser("user", "password", True)

	If db.hasError() Then
		errorAndClose(db)
	End If


	' close
	db.close()
	
	' open and attempt to use without authenticating...
	db = LoadDatabase("SQLITE", "autotest.db")
	
	Local query:TDatabaseQuery = db.executeQuery("SELECT * from person")

	' we should get an error
	If db.hasError() Then
		Print "EXPECTED : " + db.error().toString()
	End If
	
	db.close()
	
	' connect with username and password...
	db = LoadDatabase("SQLITE", "autotest.db", , , "user", "password")
	
	Print "Trying again!..."
	
	query = db.executeQuery("SELECT * from person")

	' we should get an error
	If db.hasError() Then
		errorAndClose(db)
	End If

	While query.nextRow()
		Local record:TQueryRecord = query.rowRecord()
		
		Print "Name = " + TDBString(record.value(1)).value + " " + TDBString(record.value(2)).value
	Wend
		
End If

Function errorAndClose(db:TDBConnection)
	Print db.error().toString()
	db.close()
	End
End Function