Skip to content
Snippets Groups Projects
Commit fac246a3 authored by Devinzenzhang's avatar Devinzenzhang
Browse files

test: test member

parent ab21030b
Branches
Tags
1 merge request!3Resolve "member"
......@@ -37,7 +37,7 @@ class MemberController:
cursor = self.connection.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS Member(
ID INT PRIMARY KEY NOT NULL autoincrement,
ID INT PRIMARY KEY NOT NULL,
name VARCHAR(255) NOT NULL,
address VARCHAR(255) NOT NULL,
email VARCHAR(255) NOT NULL,
......@@ -51,17 +51,17 @@ class MemberController:
def addMember(self, member : Member):
cursor = self.connection.cursor()
cursor.execute('''
INSERT INTO Member (name, address, email, phone, status, booksBorrowed)
VALUES (?, ?, ?, ?, ?, ?)
''', (member.getName(), member.getAddress(), member.getEmail(), member.getPhone(), member.getStatus(), member.getBooksBorrowed()))
INSERT INTO Member (ID, name, address, email, phone, status, booksBorrowed)
VALUES (?, ?, ?, ?, ?, ?, ?)
''', (member.getID(), member.getName(), member.getAddress(), member.getEmail(), member.getPhone(), member.getStatus(), member.getBooksBorrowed()))
self.connection.commit()
def removeMember(self, ID : Member):
cursor = self.connection.cursor()
cursor.execute('''
DELETE FROM Book WHERE ID = ?
''', (ID))
DELETE FROM Member WHERE ID = ?
''', (ID,))
self.connection.commit()
......@@ -83,5 +83,5 @@ class MemberController:
cursor = self.connection.cursor()
cursor.execute('''
SELECT * FROM Member WHERE ID = ?
''', (ID))
''', (ID,))
return cursor.fetchone()
\ No newline at end of file
File added
import sys
import os
# Get the directory containing the src directory
src_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../src'))
# Add src directory to the system path
sys.path.append(src_path)
from Member import Member, MemberController
from datetime import date
import sqlite3
def test():
connection = sqlite3.connect('library.db')
member = Member(1, 'a', 'b', 'c', 'd', 'e', 1)
member2 = Member(2, 'g', 'h', 'i', 'j', 'k', 2)
member3 = Member(3, 'l', 'm', 'n', 'o', 'p', 3)
member_controller = MemberController(connection)
member_controller.addMember(member)
member_controller.addMember(member2)
member_controller.addMember(member3)
mem = member_controller.getMember(2)
print("Member Details:", mem)
member_controller.removeMember(2)
members = member_controller.displayMember()
for i in members:
print(i)
member_controller.editMember(3, '2', '3', '4', '5', '6', 7)
members = member_controller.displayMember()
for i in members:
print(i)
connection.close()
return
test()
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment