Add more tests

bryanjos · bryanjos · commit 41c27094ec4a · 2018-07-29T11:58:22.000-05:00
diff --git a/src/processes/process_queue.ts b/src/processes/process_queue.ts
@@ -0,0 +1,25 @@
+import {PID} from 'erlang-types'
+
+class ProcessQueue {
+  pid: PID
+  tasks: Function[]
+
+  constructor(pid: PID) {
+    this.pid = pid
+    this.tasks = []
+  }
+
+  empty() {
+    return this.tasks.length === 0
+  }
+
+  add(task: any) {
+    this.tasks.push(task)
+  }
+
+  next() {
+    return this.tasks.shift()
+  }
+}
+
+export default ProcessQueue
diff --git a/src/processes/scheduler.ts b/src/processes/scheduler.ts
@@ -1,26 +1,5 @@
 import {PID} from 'erlang-types'
-
-class ProcessQueue {
-  pid: PID
-  tasks: Function[]
-
-  constructor(pid: PID) {
-    this.pid = pid
-    this.tasks = []
-  }
-
-  empty() {
-    return this.tasks.length === 0
-  }
-
-  add(task: any) {
-    this.tasks.push(task)
-  }
-
-  next() {
-    return this.tasks.shift()
-  }
-}
+import ProcessQueue from './process_queue'
 
 class Scheduler {
   isRunning: boolean
diff --git a/test/mailbox_test.js b/test/mailbox_test.js
@@ -0,0 +1,39 @@
+import test from 'ava'
+import Mailbox from '../src/processes/mailbox'
+
+test('constructor', function(t) {
+  const mailbox = new Mailbox()
+
+  t.is(mailbox.messages.length, 0)
+})
+
+test('get', function(t) {
+  const mailbox = new Mailbox()
+
+  t.is(mailbox.get().length, 0)
+
+  mailbox.deliver('Hi')
+
+  t.is(mailbox.get().length, 1)
+})
+
+test('isEmpty', function(t) {
+  const mailbox = new Mailbox()
+
+  t.true(mailbox.isEmpty())
+
+  mailbox.deliver('Hi')
+
+  t.false(mailbox.isEmpty())
+})
+
+test('removeAt', function(t) {
+  const mailbox = new Mailbox()
+
+  mailbox.deliver('Hi')
+  mailbox.deliver('Goodbye')
+
+  mailbox.removeAt(0)
+
+  t.is(mailbox.get()[0], 'Goodbye')
+})
diff --git a/test/process_queue_test.js b/test/process_queue_test.js
@@ -0,0 +1,30 @@
+import test from 'ava'
+import ProcessQueue from '../src/processes/process_queue'
+import {PID} from 'erlang-types'
+
+test('constructor', function(t) {
+  const pid = new PID()
+  const queue = new ProcessQueue(pid)
+
+  t.is(queue.pid.id, pid.id)
+  t.is(queue.tasks.length, 0)
+  t.true(queue.empty())
+})
+
+test('add', function(t) {
+  const pid = new PID()
+  const queue = new ProcessQueue(pid)
+
+  queue.add(() => 'Hi')
+  t.false(queue.empty())
+})
+
+test('next', function(t) {
+  const pid = new PID()
+  const queue = new ProcessQueue(pid)
+  const func = () => 'Hi'
+
+  queue.add(func)
+
+  t.is(queue.next(), func)
+})
