Added tests for batch

Author: Alpha1337k

tests/integration/batch.test.ts

@@ -1,11 +1,13 @@
 import makeServer from './drivers/default/server';
 import Post from '../stubs/models/post';
 import { Batch } from '../../src/batch';
+import { Orion } from '../../src/orion';
 
 let server: any;
 
 beforeEach(() => {
 	server = makeServer();
+	Orion.setBaseUrl("https://api-mock.test/api")
 });
 
 afterEach(() => {
@@ -14,25 +16,98 @@ afterEach(() => {
 
 describe('Batch tests', () => {
 	test('saving a couple of models', async () => {
-		await Batch.store(
-			[
-				new Post({title: "item1", id: 10, created_at: "", updated_at: ""}),
-				new Post({title: "item2", id: 12, created_at: "", updated_at: ""}),
-			]
-		)
+		const posts = [
+			new Post(),
+			new Post(),
+		]
+		posts[0].$attributes.title = "First";
+		posts[1].$attributes.title = "Second";
+
+		const res = await Batch.store(posts);
+
+		expect(server.schema.posts.all()).toHaveLength(2);
+		expect(server.schema.posts.find('1').attrs.title).toBe("First")
+		expect(server.schema.posts.find('2').attrs.title).toBe("Second")
+		expect(server.schema.posts.find('1').attrs.title).toEqual(res[0].$attributes.title)
+		expect(server.schema.posts.find('1').attrs.created_at).toEqual(res[0].$attributes.created_at)
+		expect(server.schema.posts.find('2').attrs.title).toEqual(res[1].$attributes.title)
+		expect(server.schema.posts.find('2').attrs.created_at).toEqual(res[1].$attributes.created_at)
 	});
 
 	test('updating a couple of models', async () => {
-		const post1 = await Post.$query().find(1);
-		const post2 = await Post.$query().find(2);
+		const posts = [
+			new Post(),
+			new Post(),
+			new Post(),
+		]
+		posts[0].$attributes.title = "First";
+		posts[1].$attributes.title = "Second";
+		posts[2].$attributes.title = "Third";
+
+		let res = await Batch.store(posts);
+
+		res[0].$attributes.title = "NewFirst";
+		res[1].$attributes.title = "NewSecond";
+
+		res = await Batch.update([res[0],res[1]]);
+
+		expect(res).toHaveLength(2);
+		expect(server.schema.posts.find('1').attrs.title).toBe("NewFirst")
+		expect(server.schema.posts.find('2').attrs.title).toBe("NewSecond")
+		expect(server.schema.posts.find('1').attrs.title).toEqual(res[0].$attributes.title)
+		expect(server.schema.posts.find('2').attrs.title).toEqual(res[1].$attributes.title)
+		expect(server.schema.posts.find('3').attrs.title).toEqual("Third");
+
+	});
+
+	test('deleting a couple of models', async () => {
+		const posts = [
+			new Post(),
+			new Post(),
+			new Post(),
+		]
+		posts[0].$attributes.title = "First";
+		posts[1].$attributes.title = "Second";
+		posts[2].$attributes.title = "Third";
+
+		let res = await Batch.store(posts);
+
+		let ModelDelete = await Batch.delete([res[1]]);
+		let idDelete = await Batch.delete([3], new Post);
+
+		expect(server.schema.posts.find('1').attrs.deleted_at).toBeUndefined();
+		expect(server.schema.posts.find('2').attrs.deleted_at).toBeDefined();
+		expect(server.schema.posts.find('3').attrs.deleted_at).toBeDefined();
+		expect(server.schema.posts.find('2').attrs.title).toEqual(ModelDelete[0].$attributes.title)
+		expect(server.schema.posts.find('3').attrs.title).toEqual(idDelete[0].$attributes.title)
+
+
+	});
+
+	test('restoring a couple of models', async () => {
+		const posts = [
+			new Post(),
+			new Post(),
+			new Post(),
+		]
+		posts[0].$attributes.title = "First";
+		posts[1].$attributes.title = "Second";
+		posts[2].$attributes.title = "Third";
+
+		let res = await Batch.store(posts);
+
+		// delete ID 2 & 3
+		let ModelDelete = await Batch.delete([res[1]]);
+		let idDelete = await Batch.delete([3], new Post);
+
+		res = await Batch.restore([...ModelDelete, ...idDelete]);
+
+		expect(server.schema.posts.find('1').attrs.deleted_at).toBeFalsy();
+		expect(server.schema.posts.find('2').attrs.deleted_at).toBeFalsy();
+		expect(server.schema.posts.find('3').attrs.deleted_at).toBeFalsy();
+		expect(server.schema.posts.find('2').attrs.title).toEqual(res[0].$attributes.title);
+		expect(server.schema.posts.find('3').attrs.title).toEqual(res[1].$attributes.title);
 
-		post1.$attributes.title = "new title";
-		post2.$attributes.title = "newer title";
 
-		await Batch.update(
-			[
-				post1, post2
-			]
-		)
 	});
 });

tests/integration/drivers/default/server.ts

@@ -129,6 +129,74 @@ export default function makeServer() {
 					updated: [request.params.tag_id],
 				};
 			});
+
+			this.post('/api/posts/batch', (schema: any, request) => {
+				const body: {
+					resources: any[]
+				} = JSON.parse(request.requestBody);
+
+				const rval: any[] = [];
+				for (let i = 0; i < body.resources.length; i++) {
+					rval.push(schema.posts.create(body.resources[i]));
+				}
+
+				return {data: rval};
+			})
+
+			this.patch('/api/posts/batch', (schema: any, request) => {
+				const body: {
+					resources: object
+				} = JSON.parse(request.requestBody);
+
+				const rval: any[] = [];
+				for (const key in body.resources) {
+					const attrs = body.resources[key];
+	
+					const post = schema.posts.find(key);
+
+	
+					rval.push(post.update(attrs));
+				}
+
+				return {data: rval};
+			})
+
+			this.delete('/api/posts/batch', (schema: any, request) => {
+				const body: {
+					resources: number[]
+				} = JSON.parse(request.requestBody);
+
+				const rval: any[] = [];
+				for (let i = 0; i < body.resources.length; i++) {
+					const id = body.resources[i];
+					const post = schema.posts.find(id);
+
+					post.update({ deleted_at: '2021-01-01' });
+
+					rval.push(post);
+				}
+
+				return {data: rval};
+			})
+
+			this.post('/api/posts/batch/restore', (schema: any, request) => {
+				const body: {
+					resources: number[]
+				} = JSON.parse(request.requestBody);
+
+				const rval: any[] = [];
+
+				for (let i = 0; i < body.resources.length; i++) {
+					const id = body.resources[i];
+					const post = schema.posts.find(id);
+
+					post.update({ deleted_at: null });
+
+					rval.push(post);
+				}
+
+				return {data: rval};
+			})
 		},
 	});
 }