[functions] Add tests for listTextureChannels, listTextureSlots (donm…

…ccurdy#545)

packages/functions/test/list-texture-channels.test.ts

@@ -1,13 +1,54 @@
 require('source-map-support').install();
 
 import test from 'tape';
+import { Document, TextureChannel } from '@gltf-transform/core';
+import { listTextureChannels, getTextureChannelMask } from '@gltf-transform/functions';
+import { MaterialsSheen } from '@gltf-transform/extensions';
+
+const { R, G, B, A } = TextureChannel;
 
 test('@gltf-transform/functions::listTextureChannels', (t) => {
-	// TODO(test)
+	const document = new Document();
+	const textureA = document.createTexture();
+	const textureB = document.createTexture();
+	const sheenExtension = document.createExtension(MaterialsSheen);
+	const sheen = sheenExtension.createSheen().setSheenRoughnessTexture(textureB);
+	const material = document
+		.createMaterial()
+		.setAlphaMode('BLEND')
+		.setBaseColorTexture(textureA)
+		.setExtension('KHR_materials_sheen', sheen);
+
+	t.deepEquals(listTextureChannels(document, textureA), [R, G, B, A], 'baseColorTexture RGBA');
+	t.deepEquals(listTextureChannels(document, textureB), [A], 'sheenColorTexture A');
+
+	material.setAlphaMode('OPAQUE');
+	t.deepEquals(listTextureChannels(document, textureA), [R, G, B], 'baseColorTexture RGB');
+
+	sheen.setSheenColorTexture(textureB);
+	t.deepEquals(listTextureChannels(document, textureB), [R, G, B, A], 'sheenColorTexture RGBA');
 	t.end();
 });
 
 test('@gltf-transform/functions::getTextureChannelMask', (t) => {
-	// TODO(test)
+	const document = new Document();
+	const textureA = document.createTexture();
+	const textureB = document.createTexture();
+	const sheenExtension = document.createExtension(MaterialsSheen);
+	const sheen = sheenExtension.createSheen().setSheenRoughnessTexture(textureB);
+	const material = document
+		.createMaterial()
+		.setAlphaMode('BLEND')
+		.setBaseColorTexture(textureA)
+		.setExtension('KHR_materials_sheen', sheen);
+
+	t.equals(getTextureChannelMask(document, textureA), R | G | B | A, 'baseColorTexture RGBA');
+	t.equals(getTextureChannelMask(document, textureB), A, 'sheenColorTexture A');
+
+	material.setAlphaMode('OPAQUE');
+	t.equals(getTextureChannelMask(document, textureA), R | G | B, 'baseColorTexture RGB');
+
+	sheen.setSheenColorTexture(textureB);
+	t.equals(getTextureChannelMask(document, textureB), R | G | B | A, 'sheenColorTexture RGBA');
 	t.end();
 });

packages/functions/test/list-texture-slots.test.ts

@@ -1,8 +1,18 @@
 require('source-map-support').install();
 
 import test from 'tape';
+import { Document } from '@gltf-transform/core';
+import { listTextureSlots } from '@gltf-transform/functions';
+import { MaterialsSheen } from '@gltf-transform/extensions';
 
 test('@gltf-transform/functions::listTextureSlots', (t) => {
-	// TODO(test)
+	const document = new Document();
+	const textureA = document.createTexture();
+	const textureB = document.createTexture();
+	const sheenExtension = document.createExtension(MaterialsSheen);
+	const sheen = sheenExtension.createSheen().setSheenColorTexture(textureB);
+	document.createMaterial().setBaseColorTexture(textureA).setExtension('KHR_materials_sheen', sheen);
+	t.deepEquals(listTextureSlots(document, textureA), ['baseColorTexture'], 'baseColorTexture');
+	t.deepEquals(listTextureSlots(document, textureB), ['sheenColorTexture'], 'sheenColorTexture');
 	t.end();
 });