mirror of
https://github.com/balena-os/balena-supervisor.git
synced 2024-12-24 15:56:40 +00:00
652b596c80
We add a bunch of additional unit tests, and also a coverage report using istanbul. The tests are not meant to cover everything, but they're a first attempt at having *some* unit testing on the supervisor. There's much to improve but hopefully it helps catch obvious errors. Change-Type: patch Signed-off-by: Pablo Carranza Velez <pablo@resin.io>
43 lines
2.3 KiB
CoffeeScript
43 lines
2.3 KiB
CoffeeScript
Promise = require 'bluebird'
|
|
iptables = require '../src/lib/iptables'
|
|
|
|
childProcess = require('child_process')
|
|
|
|
m = require 'mochainon'
|
|
{ stub } = m.sinon
|
|
{ expect } = m.chai
|
|
|
|
describe 'iptables', ->
|
|
it 'calls iptables to delete and recreate rules to block a port', ->
|
|
stub(childProcess, 'execAsync').returns(Promise.resolve())
|
|
iptables.rejectOnAllInterfacesExcept(['foo', 'bar'], 42)
|
|
.then ->
|
|
expect(childProcess.execAsync.callCount).to.equal(6)
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -j REJECT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -A INPUT -p tcp --dport 42 -j REJECT')
|
|
.then ->
|
|
childProcess.execAsync.restore()
|
|
|
|
it "falls back to blocking the port with DROP if there's no REJECT support", ->
|
|
stub(childProcess, 'execAsync').callsFake (cmd) ->
|
|
if /REJECT$/.test(cmd)
|
|
Promise.reject()
|
|
else
|
|
Promise.resolve()
|
|
iptables.rejectOnAllInterfacesExcept(['foo', 'bar'], 42)
|
|
.then ->
|
|
expect(childProcess.execAsync.callCount).to.equal(8)
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -j REJECT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -A INPUT -p tcp --dport 42 -j REJECT')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -j DROP')
|
|
expect(childProcess.execAsync).to.be.calledWith('iptables -A INPUT -p tcp --dport 42 -j DROP')
|
|
.then ->
|
|
childProcess.execAsync.restore() |