From 3e9a51a78b9028a91e2201b5d241a884ebbe84a1 Mon Sep 17 00:00:00 2001
From: Michael Hoang <enzime@users.noreply.github.com>
Date: Mon, 14 Mar 2022 00:26:49 +1100
Subject: [PATCH] nixos/tests: Make remote-ssh test work with flakes

---
 nixos/tests/vscode-remote-ssh.nix | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/nixos/tests/vscode-remote-ssh.nix b/nixos/tests/vscode-remote-ssh.nix
index 3d0468fa5f4..871eeab0cdb 100644
--- a/nixos/tests/vscode-remote-ssh.nix
+++ b/nixos/tests/vscode-remote-ssh.nix
@@ -1,9 +1,13 @@
-import ./make-test-python.nix ({ lib, ... }: let
-  pkgs = (import ../..) {
-    config.allowUnfreePredicate = pkg: builtins.elem (lib.getName pkg) [
-      "vscode" "vscode-with-extensions" "vscode-extension-ms-vscode-remote-remote-ssh"
-    ];
-  };
+import ./make-test-python.nix ({ lib, ... }@args: let
+  pkgs = args.pkgs.extend (self: super: {
+    stdenv = super.stdenv.override {
+      config = super.config // {
+        allowUnfreePredicate = pkg: builtins.elem (lib.getName pkg) [
+          "vscode" "vscode-with-extensions" "vscode-extension-ms-vscode-remote-remote-ssh"
+        ];
+      };
+    };
+  });
 
   inherit (import ./ssh-keys.nix pkgs) snakeOilPrivateKey snakeOilPublicKey;