diff --git a/frontend/three_d_garden/__tests__/bot_test.tsx b/frontend/three_d_garden/__tests__/bot_test.tsx
index c08108b45..9bdcc195d 100644
--- a/frontend/three_d_garden/__tests__/bot_test.tsx
+++ b/frontend/three_d_garden/__tests__/bot_test.tsx
@@ -34,6 +34,20 @@ describe("", () => {
.toEqual([-1344, 110, 75]);
});
+ it("renders: v1.7", () => {
+ const p = fakeProps();
+ p.config.kitVersion = "v1.7";
+ const wrapper = mount();
+ expect(wrapper.find({ name: "button-group" }).length).toEqual(10);
+ });
+
+ it("renders: v1.8", () => {
+ const p = fakeProps();
+ p.config.kitVersion = "v1.8";
+ const wrapper = mount();
+ expect(wrapper.find({ name: "button-group" }).length).toEqual(6);
+ });
+
it("loads shapes", () => {
const p = fakeProps();
mount();
diff --git a/frontend/three_d_garden/__tests__/packaging_test.tsx b/frontend/three_d_garden/__tests__/packaging_test.tsx
index 1dcdefb97..523440ace 100644
--- a/frontend/three_d_garden/__tests__/packaging_test.tsx
+++ b/frontend/three_d_garden/__tests__/packaging_test.tsx
@@ -10,7 +10,9 @@ describe("", () => {
});
it("renders", () => {
- const wrapper = mount();
+ const p = fakeProps();
+ p.config.kitVersion = "v1.7";
+ const wrapper = mount();
expect(wrapper.html()).toContain("packaging");
expect(wrapper.html()).toContain("100");
expect(wrapper.html()).not.toContain("170");
@@ -19,6 +21,7 @@ describe("", () => {
it("renders: XL", () => {
const p = fakeProps();
p.config.sizePreset = "Genesis XL";
+ p.config.kitVersion = "v1.8";
const wrapper = mount();
expect(wrapper.html()).toContain("170");
expect(wrapper.html()).not.toContain("100");