[tests] add tests

particle-iot · Jul 26, 2024 · 851d12e · 851d12e
1 parent dc1511c
commit 851d12e
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git a/src/dfu.js b/src/dfu.js
@@ -238,7 +238,7 @@ class Dfu {
 	async getProtectionState() {
 		try {
 			const res = await this._getStringDescriptor(0xfe);
-			const state = res.split(';').find(kv => kv.startsWith('s='))?.split('=')[1];
+			const state = res.split(';').find(kv => kv.startsWith('sm='))?.split('=')[1];
 
 			switch (state) {
 				case 'o': return { protected: false, overridden: false };

diff --git a/src/dfu.test.js b/src/dfu.test.js
@@ -365,8 +365,35 @@ describe('dfu', () => {
 	});
 
 	describe('getProtectionState', () => {
+		it ('detects an Open Device from string desc for Device-OS >= 6.1.2', async () => {
+			const dfu = new Dfu();
+			sinon.stub(dfu, '_getStringDescriptor').resolves("sm=o");
+
+			const res = await dfu.getProtectionState();
+			expect(res.protected).to.eql(false);
+			expect(res.overridden).to.eql(false);
+		});
+
+		it ('detects a Protected Device from string desc for Device-OS >= 6.1.2', async () => {
+			const dfu = new Dfu();
+			sinon.stub(dfu, '_getStringDescriptor').resolves("sm=p");
+
+			const res = await dfu.getProtectionState();
+			expect(res.protected).to.eql(true);
+		});
+
+		it ('detects a Protected Device in Service Mode from string desc for Device-OS >= 6.1.2', async () => {
+			const dfu = new Dfu();
+			sinon.stub(dfu, '_getStringDescriptor').resolves("sm=s");
+
+			const res = await dfu.getProtectionState();
+			expect(res.protected).to.eql(false);
+			expect(res.overridden).to.eql(true);
+		});
+
 		it('returns that all segments are protected', async () => {
 			const dfu = new Dfu();
+			sinon.stub(dfu, '_getStringDescriptor').rejects('random error');
 			sinon.stub(dfu, 'setAltSetting').resolves();
 			const internalFlashDesc = {
 				'name': 'Internal Flash',
@@ -409,11 +436,13 @@ describe('dfu', () => {
 
 			const res = await dfu.getProtectionState();
 
+			expect(dfu.setAltSetting).to.have.been.calledOnce;
 			expect(res.protected).to.eql(true);
 		});
 
 		it('returns that all segments are not protected', async () => {
 			const dfu = new Dfu();
+			sinon.stub(dfu, '_getStringDescriptor').rejects('random error');
 			sinon.stub(dfu, 'setAltSetting').resolves();
 			const internalFlashDesc = {
 				'name': 'Internal Flash',
@@ -456,6 +485,7 @@ describe('dfu', () => {
 
 			const res = await dfu.getProtectionState();
 
+			expect(dfu.setAltSetting).to.have.been.calledOnce;
 			expect(res.protected).to.eql(false);
 		});
 	});